def setUp(self): self.router_mock = Mock(spec=DatapointRouter) carbon_client.settings = TestSettings() # reset to defaults factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a'), self.router_mock) self.protocol = factory.buildProtocol(('127.0.0.1', 2003)) self.transport = StringTransport() self.protocol.makeConnection(self.transport)
def setUp(self): self.protocol_mock = Mock(spec=CarbonPickleClientProtocol) self.protocol_patch = patch( 'carbon.client.CarbonPickleClientProtocol', new=Mock(return_value=self.protocol_mock)) self.protocol_patch.start() carbon_client.settings = TestSettings() self.factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory.buildProtocol(None) self.connected_factory.started = True
class CarbonClientFactoryTest(TestCase): def setUp(self): self.protocol_mock = Mock(spec=CarbonPickleClientProtocol) self.protocol_patch = patch( 'carbon.client.CarbonPickleClientProtocol', new=Mock(return_value=self.protocol_mock)) self.protocol_patch.start() carbon_client.settings = TestSettings() self.factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory.buildProtocol(None) self.connected_factory.started = True def tearDown(self): if self.factory.deferSendPending and self.factory.deferSendPending.active(): self.factory.deferSendPending.cancel() self.protocol_patch.stop() def test_schedule_send_schedules_call_to_send_queued(self): self.factory.scheduleSend() self.assertIsInstance(self.factory.deferSendPending, DelayedCall) self.assertTrue(self.factory.deferSendPending.active()) def test_schedule_send_ignores_already_scheduled(self): self.factory.scheduleSend() expected_fire_time = self.factory.deferSendPending.getTime() self.factory.scheduleSend() self.assertTrue(expected_fire_time, self.factory.deferSendPending.getTime()) def test_send_queued_should_noop_if_not_connected(self): self.factory.scheduleSend() self.assertFalse(self.protocol_mock.sendQueued.called) def test_send_queued_should_call_protocol_send_queued(self): self.connected_factory.sendQueued() self.protocol_mock.sendQueued.assert_called_once_with()
def setUp(self): carbon_client.settings = TestSettings() # reset to defaults factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a')) self.protocol = factory.buildProtocol(('127.0.0.1', 2003)) self.transport = StringTransport() self.protocol.makeConnection(self.transport)
class CarbonClientFactoryTest(TestCase): def setUp(self): self.router_mock = Mock(spec=DatapointRouter) self.protocol_mock = Mock(spec=CarbonPickleClientProtocol) self.protocol_patch = patch( 'carbon.client.CarbonPickleClientProtocol', new=Mock(return_value=self.protocol_mock)) self.protocol_patch.start() carbon_client.settings = TestSettings() self.factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a'), self.router_mock) self.connected_factory = CarbonPickleClientFactory(('127.0.0.1', 2003, 'a'), self.router_mock) self.connected_factory.buildProtocol(None) self.connected_factory.started = True def tearDown(self): if self.factory.deferSendPending and self.factory.deferSendPending.active(): self.factory.deferSendPending.cancel() self.protocol_patch.stop() def test_schedule_send_schedules_call_to_send_queued(self): self.factory.scheduleSend() self.assertIsInstance(self.factory.deferSendPending, DelayedCall) self.assertTrue(self.factory.deferSendPending.active()) def test_schedule_send_ignores_already_scheduled(self): self.factory.scheduleSend() expected_fire_time = self.factory.deferSendPending.getTime() self.factory.scheduleSend() self.assertTrue(expected_fire_time, self.factory.deferSendPending.getTime()) def test_send_queued_should_noop_if_not_connected(self): self.factory.scheduleSend() self.assertFalse(self.protocol_mock.sendQueued.called) def test_send_queued_should_call_protocol_send_queued(self): self.connected_factory.sendQueued() self.protocol_mock.sendQueued.assert_called_once_with()