def test_context_data(self):
# Імітуємо будинок з одної квартири:
floors = ['0']
entrances = ['1']
flat = Flat(floor_No=0, entrance_No=1)
flat.save()
block_scheme = {
'0': {
'1': [
flat,
]
}
}
# {0: {1: [flat, ], 2: [flat, ]}, 1: {1: [flat, ], 2: [flat, ]},}
block_length = {'1': 1}
kwargs = {
'block_scheme': block_scheme,
'block_length': block_length,
'floors': floors,
'entrances': entrances,
}
# Setup request and view.
request = RequestFactory().get('/flats/scheme/')
view = FlatScheme()
view = setup_view(view, request, kwargs)
# Run.
context = view.get_context_data()
# Check.
self.assertEqual(context['block_scheme'], block_scheme)
self.assertEqual(context['block_length'], block_length)
self.assertEqual(context['floors'], floors)
self.assertEqual(context['entrances'], entrances)
def test_context_data_2(self):
# Імітуємо будинок з кількох квартир:
flat1 = Flat(floor_No=1, entrance_No=1)
flat1.save()
flat2 = Flat(floor_No=2, entrance_No=2)
flat2.save()
flat3 = Flat(floor_No=2, entrance_No=2)
flat3.save()
floors = ["2", "1"]
entrances = ["1", "2"]
block_scheme = {"1": {"1": [flat1]}, "2": {"2": [flat2, flat3]}}
# {0: {1: [flat, ], 2: [flat, ]}, 1: {1: [flat, ], 2: [flat, ]},}
block_length = {"1": 1, "2": 2}
kwargs = {"block_scheme": block_scheme, "block_length": block_length, "floors": floors, "entrances": entrances}
# Setup request and view.
request = RequestFactory().get("/flats/scheme/")
view = FlatScheme()
view = setup_view(view, request, kwargs)
# Run.
context = view.get_context_data()
# Check.
self.assertEqual(context["block_scheme"], block_scheme)
self.assertEqual(context["block_length"], block_length)
self.assertEqual(context["floors"], floors)
self.assertEqual(context["entrances"], entrances)
def test_get_value_list_gives_proper_values_for_Flat(self):
cls = AllFieldsMixin()
cls.model = Flat
flat = Flat(id=5, flat_No='5', floor_No=1, entrance_No=2)
flat.save()
key_list = ("flat_No", "floor_No", "entrance_No")
value_list = cls.get_value_list(flat, key_list)
# Списки мають мати довжину
self.assertEqual(len(value_list), 3)
# Перевірка значень:
self.assertEqual(value_list[0], '5')
self.assertEqual(value_list[1], 1)
self.assertEqual(value_list[2], 2)
def test_get_value_list_gives_proper_values_for_Flat(self):
cls = AllFieldsMixin()
cls.model = Flat
flat = Flat(id=5, flat_No='5', floor_No=1, entrance_No=2)
flat.save()
key_list = ("flat_No", "floor_No", "entrance_No")
value_list = cls.get_value_list(flat, key_list)
# Списки мають мати довжину
self.assertEqual(len(value_list), 3)
# Перевірка значень:
self.assertEqual(value_list[0], '5')
self.assertEqual(value_list[1], 1)
self.assertEqual(value_list[2], 2)
def create_dummy_flat(self,
id=1,
flat_No="25а",
floor_No=2,
entrance_No=3,
flat_99=25):
# створюємо квартиру:
flat = Flat(id=id,
flat_No=flat_No,
floor_No=floor_No,
entrance_No=entrance_No,
flat_99=flat_99)
flat.save()
# print('created flat:', flat)
return flat
def test_flat_user_CSS_class(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(flat_user_CSS_class(flat_1), "flat-has-member")
self.assertEqual(flat_user_CSS_class(flat_2), "flat-has-users")
self.assertEqual(flat_user_CSS_class(flat_3), "")
self.assertEqual(flat_user_CSS_class(None), "")
self.assertEqual(flat_user_CSS_class(User()), "")
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_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_context_data(self):
# Імітуємо будинок з одної квартири:
floors = ["0"]
entrances = ["1"]
flat = Flat(floor_No=0, entrance_No=1)
flat.save()
block_scheme = {"0": {"1": [flat]}}
# {0: {1: [flat, ], 2: [flat, ]}, 1: {1: [flat, ], 2: [flat, ]},}
block_length = {"1": 1}
kwargs = {"block_scheme": block_scheme, "block_length": block_length, "floors": floors, "entrances": entrances}
# Setup request and view.
request = RequestFactory().get("/flats/scheme/")
view = FlatScheme()
view = setup_view(view, request, kwargs)
# Run.
context = view.get_context_data()
# Check.
self.assertEqual(context["block_scheme"], block_scheme)
self.assertEqual(context["block_length"], block_length)
self.assertEqual(context["floors"], floors)
self.assertEqual(context["entrances"], entrances)
def create_dummy_building(self,
floors=(
0,
1,
2,
), entrances=(
1,
2,
3,
)):
for f in floors:
for e in entrances:
for i in range(f + e):
no = f * 100 + e * 10 + i + 1
flat_No = str(no)
# створюємо квартиру:
flat = Flat(flat_No=flat_No,
floor_No=f,
entrance_No=e,
flat_99=no)
flat.save()
def setUp(self):
self.cls_view = FlatDetailHorizontal
self.path = '/flats/5/h/'
self.template = 'flats/flat_detail_h.html'
flat = Flat(id=5, flat_No='5')
flat.save()
def test_context_data_2(self):
# Імітуємо будинок з кількох квартир:
flat1 = Flat(floor_No=1, entrance_No=1)
flat1.save()
flat2 = Flat(floor_No=2, entrance_No=2)
flat2.save()
flat3 = Flat(floor_No=2, entrance_No=2)
flat3.save()
floors = ['2', '1']
entrances = ['1', '2']
block_scheme = {'1': {'1': [flat1]}, '2': {'2': [flat2, flat3]}}
# {0: {1: [flat, ], 2: [flat, ]}, 1: {1: [flat, ], 2: [flat, ]},}
block_length = {'1': 1, '2': 2}
kwargs = {
'block_scheme': block_scheme,
'block_length': block_length,
'floors': floors,
'entrances': entrances,
}
# Setup request and view.
request = RequestFactory().get('/flats/scheme/')
view = FlatScheme()
view = setup_view(view, request, kwargs)
# Run.
context = view.get_context_data()
# Check.
self.assertEqual(context['block_scheme'], block_scheme)
self.assertEqual(context['block_length'], block_length)
self.assertEqual(context['floors'], floors)
self.assertEqual(context['entrances'], entrances)
def setUp(self):
self.cls_view = FlatDetailHorizontal
self.path = "/flats/5/h/"
self.template = "flats/flat_detail_h.html"
flat = Flat(id=5, flat_No="5")
flat.save()
user2.last_name = 'Ciccone'
user2.save()
profile2 = UserProfile(user=user2)
profile2.picture = "Sally.png"
profile2.save()
user3 = User.objects.create_user('rihanna', '*****@*****.**', 'rihanna')
user3.first_name = 'Rihanna'
user3.last_name = 'Fenty'
user3.save()
profile3 = UserProfile(user=user3)
profile3.save()
flat1 = Flat(name='26 Winton Drive', description='The ultimate home')
flat1.save()
flat2 = Flat(name='28 Winton Drive', description='A flat for the less lucky ones')
flat2.save()
flat3 = Flat(name='30 Winton Drive', description='Study flat')
flat3.save()
flat4 = Flat(name='Inactive flat', description='Long days, long nights', active=False)
flat4.save()
flatmember1 = Flat_Member(user=user2, flat=flat1, join_date=datetime.datetime.now())
flatmember1.save()