def testPreferencesResponseShouldBeAsExpectedFromConnection(self): # Given... class StubConnection (object): def request(self, url, method, data, headers): import StringIO response = StringIO.StringIO('MyBody') response.getheaders = lambda: {'h1':1} return response StubConnection = Stub(PcsConnection)(StubConnection) conn = StubConnection() source = LocationsScreenscrapeSource() # When... response_body, response_headers = \ source.get_preferences_response(conn=conn, sessionid='123abc') # Then... self.assertEqual(response_body, 'MyBody') self.assertEqual(response_headers, {'h1':1})