class TimingOutCQLClientTests(SynchronousTestCase):
"""
Tests for `:py:class:TimingOutCQLClient`
"""
def setUp(self):
"""
Sample client and clock
"""
self.client = mock.Mock(spec=['execute', 'disconnect'])
self.clock = Clock()
self.tclient = TimingOutCQLClient(self.clock, self.client, 10)
def test_execute(self):
"""
Execute is delgated to the client
"""
self.client.execute.return_value = defer.succeed(5)
d = self.tclient.execute(2, 3, a=4)
self.assertEqual(self.successResultOf(d), 5)
self.client.execute.assert_called_once_with(2, 3, a=4)
def test_disconnect(self):
"""
`disconnect()` is delgated to the client
"""
self.client.disconnect.return_value = defer.succeed(5)
d = self.tclient.disconnect()
self.assertEqual(self.successResultOf(d), 5)
self.client.disconnect.assert_called_once_with()
def test_times_out(self):
"""
Execute times out after given time
"""
self.client.execute.return_value = defer.Deferred()
d = self.tclient.execute(2, 3)
self.assertNoResult(d)
self.clock.advance(10)
self.failureResultOf(d, TimedOutError)
class TimingOutCQLClientTests(SynchronousTestCase):
"""
Tests for `:py:class:TimingOutCQLClient`
"""
def setUp(self):
"""
Sample client and clock
"""
self.client = mock.Mock(spec=['execute', 'disconnect'])
self.clock = Clock()
self.tclient = TimingOutCQLClient(self.clock, self.client, 10)
def test_execute(self):
"""
Execute is delgated to the client
"""
self.client.execute.return_value = defer.succeed(5)
d = self.tclient.execute(2, 3, a=4)
self.assertEqual(self.successResultOf(d), 5)
self.client.execute.assert_called_once_with(2, 3, a=4)
def test_disconnect(self):
"""
`disconnect()` is delgated to the client
"""
self.client.disconnect.return_value = defer.succeed(5)
d = self.tclient.disconnect()
self.assertEqual(self.successResultOf(d), 5)
self.client.disconnect.assert_called_once_with()
def test_times_out(self):
"""
Execute times out after given time
"""
self.client.execute.return_value = defer.Deferred()
d = self.tclient.execute(2, 3)
self.assertNoResult(d)
self.clock.advance(10)
self.failureResultOf(d, TimedOutError)
