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})
