move connection-level test to the spec that tests the class that implements the logic