class TestPrtocolFactory(unittest.TestCase): def setUp(self): self.comaster = COMasterMock() self.factory = MaraPorotocolFactory(self.comaster) def test_protocol_holds_reference_to_factory(self): protocol = self.factory.buildProtocol(('127.0.0.1', 0)) self.assertEqual(protocol.factory, self.factory) def test_protocol_holds_reference_to_comaster(self): protocol = self.factory.buildProtocol(('127.0.0.1', 0)) self.assertEqual(protocol.comaster, self.comaster)
def setUp(self): self.comaster = COMasterMock() self.factory = MaraPorotocolFactory(self.comaster)