class TestProvideSizeSubscriber(unittest.TestCase): def setUp(self): self.transfer_future = mock.Mock(spec=TransferFuture) self.transfer_meta = TransferMeta() self.transfer_future.meta = self.transfer_meta def test_size_set(self): self.transfer_meta.provide_transfer_size(5) subscriber = ProvideSizeSubscriber(10) subscriber.on_queued(self.transfer_future) self.assertEqual(self.transfer_meta.size, 10)
class TestTransferMeta(unittest.TestCase): def setUp(self): self.transfer_meta = TransferMeta() def test_size(self): self.assertEqual(self.transfer_meta.size, None) self.transfer_meta.provide_transfer_size(5) self.assertEqual(self.transfer_meta.size, 5) def test_call_args(self): call_args = object() transfer_meta = TransferMeta(call_args) # Assert the that call args provided is the same as is returned self.assertIs(transfer_meta.call_args, call_args) def test_transfer_id(self): transfer_meta = TransferMeta(transfer_id=1) self.assertEqual(transfer_meta.transfer_id, 1) def test_user_context(self): self.transfer_meta.user_context['foo'] = 'bar' self.assertEqual(self.transfer_meta.user_context, {'foo': 'bar'})