def setUp(self):
"""
Wraps treq so that actual calls are mostly made, but that certain
results can be stubbed out
"""
self.client = mock.Mock(Client)
self.shipper = Shipper(
client_builder=lambda *args, **kwargs: self.client)
# this just runs the call and returns the result of the deferred
def _fake_blocking_call_from_thread(reactor, call, *args, **kwargs):
d = maybeDeferred(call, *args, **kwargs)
return self.successResultOf(d)
self.blocking_call = mock.patch(
'shipper.shipper.threads.blockingCallFromThread',
side_effect=_fake_blocking_call_from_thread).start()
self.addCleanup(mock.patch.stopall)
class ShipperCommands(TestCase):
"""
Tests commands (methods on Shipper)
"""
def setUp(self):
"""
Wraps treq so that actual calls are mostly made, but that certain
results can be stubbed out
"""
self.client = mock.Mock(Client)
self.shipper = Shipper(
client_builder=lambda *args, **kwargs: self.client)
# this just runs the call and returns the result of the deferred
def _fake_blocking_call_from_thread(reactor, call, *args, **kwargs):
d = maybeDeferred(call, *args, **kwargs)
return self.successResultOf(d)
self.blocking_call = mock.patch(
'shipper.shipper.threads.blockingCallFromThread',
side_effect=_fake_blocking_call_from_thread).start()
self.addCleanup(mock.patch.stopall)
def test_wait(self):
"""
Client.wait is called for every for every container passed to
Shipper.wait. The result is a list tuples of container: results
for all the containers.
"""
self.client.wait.side_effect = (
lambda *args, **kwargs: succeed('wait_success'))
containers = [
Container('localhost:1234', {'Id': '1'}),
Container('localhost:2345', {'Id': '2'})
]
result = self.shipper.wait(*containers)
self.client.wait.assert_has_calls([
mock.call('localhost:1234', container=containers[0]),
mock.call('localhost:2345', container=containers[1])
])
self.assertEqual([(container, 'wait_success')
for container in containers], result)
def setUp(self):
"""
Wraps treq so that actual calls are mostly made, but that certain
results can be stubbed out
"""
self.client = mock.Mock(Client)
self.shipper = Shipper(
client_builder=lambda *args, **kwargs: self.client)
# this just runs the call and returns the result of the deferred
def _fake_blocking_call_from_thread(reactor, call, *args, **kwargs):
d = maybeDeferred(call, *args, **kwargs)
return self.successResultOf(d)
self.blocking_call = mock.patch(
'shipper.shipper.threads.blockingCallFromThread',
side_effect = _fake_blocking_call_from_thread).start()
self.addCleanup(mock.patch.stopall)
class ShipperCommands(TestCase):
"""
Tests commands (methods on Shipper)
"""
def setUp(self):
"""
Wraps treq so that actual calls are mostly made, but that certain
results can be stubbed out
"""
self.client = mock.Mock(Client)
self.shipper = Shipper(
client_builder=lambda *args, **kwargs: self.client)
# this just runs the call and returns the result of the deferred
def _fake_blocking_call_from_thread(reactor, call, *args, **kwargs):
d = maybeDeferred(call, *args, **kwargs)
return self.successResultOf(d)
self.blocking_call = mock.patch(
'shipper.shipper.threads.blockingCallFromThread',
side_effect = _fake_blocking_call_from_thread).start()
self.addCleanup(mock.patch.stopall)
def test_wait(self):
"""
Client.wait is called for every for every container passed to
Shipper.wait. The result is a list tuples of container: results
for all the containers.
"""
self.client.wait.side_effect = (
lambda *args, **kwargs: succeed('wait_success'))
containers = [Container('localhost:1234', {'Id': '1'}),
Container('localhost:2345', {'Id': '2'})]
result = self.shipper.wait(*containers)
self.client.wait.assert_has_calls([
mock.call('localhost:1234', container=containers[0]),
mock.call('localhost:2345', container=containers[1])])
self.assertEqual(
[(container, 'wait_success') for container in containers],
result)
