def test_get_flat_users(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()
self.assertEqual(get_flat_users(flat_1), [john,])
self.assertEqual(get_flat_users(flat_2), [paul, george, ringo])
self.assertEqual(get_flat_users(flat_3), [])
self.assertEqual(get_flat_users(None), [])
self.assertEqual(get_flat_users(User()), [])
def test_get_flat_users(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()
self.assertEqual(get_flat_users(flat_1), [
john,
])
self.assertEqual(get_flat_users(flat_2), [paul, george, ringo])
self.assertEqual(get_flat_users(flat_3), [])
self.assertEqual(get_flat_users(None), [])
self.assertEqual(get_flat_users(User()), [])
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
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
def test_visitor_can_find_link(self):
# Користувач може перейти по лінку потрібні дані
self.browser.get('%s%s' % (self.server_url, self.this_url))
for user in get_flat_users(self.flat2):
link_parent_selector = '#body-table'
link_text = user.username
url_name = 'adm-users-profile'
kwargs = {'pk': user.id}
expected_regex = ""
self.check_go_to_link(self.this_url, link_parent_selector, link_text,
url_name=url_name, kwargs=kwargs, expected_regex=expected_regex)
print('finished: %s' % inspect.stack()[0][3], end=' >> ')
def data_len(self):
flat = Flat.objects.get(id=2)
qs = get_flat_users(flat)
return len(qs)
def get_queryset(self):
# Повертає список всіх користувачів, у профілі яких вказано flat
flat = self.object
qs = get_flat_users(flat)
return qs