def test_has_flat_member(self): john, paul, george, ringo, freddy = DummyUser().create_dummy_beatles() flat_1, flat_2 = DummyUser().set_flats_to_beatles(john, paul, george, ringo, freddy) flat_3 = Flat() DummyUser().create_dummy_group(group_name='members') DummyUser().add_dummy_group(john, group_name='members') self.assertEqual(has_flat_member(flat_1), True) self.assertEqual(has_flat_member(flat_2), False) self.assertEqual(has_flat_member(flat_3), False) self.assertEqual(has_flat_member(None), False) self.assertEqual(has_flat_member(User()), False)
def test_has_flat_member(self): john, paul, george, ringo, freddy = DummyUser().create_dummy_beatles() flat_1, flat_2 = DummyUser().set_flats_to_beatles( john, paul, george, ringo, freddy) flat_3 = Flat() DummyUser().create_dummy_group(group_name='members') DummyUser().add_dummy_group(john, group_name='members') self.assertEqual(has_flat_member(flat_1), True) self.assertEqual(has_flat_member(flat_2), False) self.assertEqual(has_flat_member(flat_3), False) self.assertEqual(has_flat_member(None), False) self.assertEqual(has_flat_member(User()), False)
def flat_user_CSS_class(flat): # Фільтр визначає CSS-клас для комірки з flat # залежно від користувачів, повязаних з цією квартирою css_class = "" users = get_flat_users(flat) if users: css_class = "flat-has-users" if has_flat_member(flat): css_class = "flat-has-member" return css_class