class ClientTest(unittest.TestCase):
def setUp(self):
self.client = Client(MY_OAUTH_KEY, MY_OAUTH_SECRET, API_VERSION, API_HOST, API_PORT)
def fake_request(self, *args, **kwargs):
d = defer.Deferred()
reactor.callLater(0, d.callback, mock.sentinel.retval)
return d
@defer.inlineCallbacks
def test_add_record_returns_deferred(self):
self.client._request = mock.Mock(wraps=self.fake_request)
record = Record(layer=TESTING_LAYER, id=69, lat=34.5, lon=-122.8)
deferred = self.client.add_record(record)
self.assertTrue(isinstance(deferred, defer.Deferred))
retval = yield deferred
self.assertEquals(mock.sentinel.retval, retval)
@defer.inlineCallbacks
def test_add_records_returns_deferred(self):
self.client._request = mock.Mock(wraps=self.fake_request)
records = [Record(layer=TESTING_LAYER, id=69, lat=34.5, lon=-122.8), Record(layer=TESTING_LAYER, id=70, lat=34.5, lon=-122.8)]
deferred = self.client.add_records(TESTING_LAYER, records)
self.assertTrue(isinstance(deferred, defer.Deferred))
retval = yield deferred
self.assertEquals(mock.sentinel.retval, retval)
@defer.inlineCallbacks
def test_delete_record_returns_deferred(self):
self.client._request = mock.Mock(wraps=self.fake_request)
deferred = self.client.delete_record(TESTING_LAYER, 4)
self.assertTrue(isinstance(deferred, defer.Deferred))
retval = yield deferred
self.assertEquals(mock.sentinel.retval, retval)
def setUp(self):
self.client = Client(MY_OAUTH_KEY, MY_OAUTH_SECRET, API_VERSION, API_HOST, API_PORT)