class TestChainReporters(unittest.TestCase): def setUp(self): self.rpt1 = mock.Mock() self.rpt2 = mock.Mock() self.chain = Repeater(self.rpt1, self.rpt2) def test_start_called(self): self.chain.start() self.rpt1.start.assert_called_with() self.rpt2.start.assert_called_with() def test_feed_called(self): self.chain.feed("some text") self.rpt1.feed.assert_called_with("some text") self.rpt2.feed.assert_called_with("some text") def test_stop_called(self): self.chain.stop(123) self.rpt1.stop.assert_called_with(123) self.rpt2.stop.assert_called_with(123)
def setUp(self): self.rpt1 = mock.Mock() self.rpt2 = mock.Mock() self.chain = Repeater(self.rpt1, self.rpt2)