Refractored Test to use a FakeSocket implementation