def test_routetype_count2(self): b_counter = Counter() b1_counter = Counter() counter = Counter('count2') counter1 = Counter('count1') p = Pipe( Loop([A(), B(), A()]), Branch(self.only_b, counter1, route_type=B, join=True, share=True), counter, Branch(self.only_a, self.a_to_b, self.only_b, b1_counter, route_type=A, share=False, join=True), self.only_b, b_counter) self.assertFalse(p.finished()) BotFrame.run() self.assertEqual(counter1.count, 1) self.assertEqual(counter.count, 4) self.assertEqual(b1_counter.count, 2) self.assertEqual(b_counter.count, 4) self.assertTrue(p.finished())
def test_routetype_count2(self): self.a_count = 0 self.b_count = 0 p = Pipe( Loop([A(), B(), A()]), Branch(self.only_b, route_type=B, join=True), Branch(self.only_a, self.a_to_b, route_type=A, share=False, join=True), self.only_b) self.assertFalse(p.finished()) BotFrame.run() self.assertEqual(self.b_count, 5) self.assertEqual(self.a_count, 2) self.assertTrue(p.finished())