def test_users_mediation(self):
# simulation inherit MediatorMixin
simulation = Simulation(users_count=2, max_books_per_user=2, exchange_points_count=1)
simulation.generate_items()
# each user inherit ColleagueMixin
for user in simulation.all_users:
user.set_mediator(simulation)
simulation.broadcast_message_to_colleagues('Broadcasted message')
user_colleague = simulation.all_users[0]
user_colleague.send_message_to_mediator('new message')
self.assertEqual(user_colleague.mediator, simulation)
self.assertTrue(user_colleague in simulation.colleagues)
def test_users_mediation(self):
# simulation inherit MediatorMixin
simulation = Simulation(users_count=2,
max_books_per_user=2,
exchange_points_count=1)
simulation.generate_items()
# each user inherit ColleagueMixin
for user in simulation.all_users:
user.set_mediator(simulation)
simulation.broadcast_message_to_colleagues('Broadcasted message')
user_colleague = simulation.all_users[0]
user_colleague.send_message_to_mediator('new message')
self.assertEqual(user_colleague.mediator, simulation)
self.assertTrue(user_colleague in simulation.colleagues)
