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())
