def setUp(self): self.protocol_mock = Mock(spec=CarbonClientProtocol) self.protocol_patch = patch('carbon.client.CarbonClientProtocol', new=Mock(return_value=self.protocol_mock)) self.protocol_patch.start() carbon_client.settings = TestSettings() self.factory = CarbonClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory = CarbonClientFactory(('127.0.0.1', 2003, 'a')) self.connected_factory.buildProtocol(None) self.connected_factory.started = True
def setUp(self): carbon_client.settings = TestSettings() # reset to defaults factory = CarbonClientFactory(('127.0.0.1', 2003, 'a')) self.protocol = factory.buildProtocol(('127.0.0.1', 2003)) self.transport = StringTransport() self.protocol.makeConnection(self.transport)