def setUp(self):
user_details = {
'username': '******',
'password': '******',
'email': '*****@*****.**'
}
self.user = User.objects.create_user(**user_details)
self.user.generate_key()
self.org = Organization.objects.create()
OrganizationUser.objects.create(user=self.user, organization=self.org)
cb = CanonicalBuilding(active=True)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb,
property_name='ADMIN BUILDING',
address_line_1='100 Admin St')
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.org
b.save()
self.building = b
self.api_url = reverse_lazy('api:v1:get_building')
self.params = {
'building_id': cb.pk,
'organization_id': self.org.pk,
}
auth_string = base64.urlsafe_b64encode('{}:{}'.format(
self.user.username, self.user.api_key))
self.auth_string = 'Basic {}'.format(auth_string)
def setUp(self):
self.snapshots = []
for x in range(50):
cb = CanonicalBuilding()
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb)
b.extra_data = {'my new field': 'something extra'}
b.save()
self.snapshots.append(b)
def setUp(self):
user_details = {
'username': '******',
'password': '******',
'email': '*****@*****.**',
'first_name': 'Johnny',
'last_name': 'Energy',
}
self.user = User.objects.create_user(**user_details)
self.org = Organization.objects.create(name='my org')
self.org.add_member(self.user)
self.client.login(**user_details)
self.fake_request = FakeRequest(user=self.user)
# arrange
self.NUMBER_ACTIVE = 50
NUMBER_INACTIVE = 25
NUMBER_WITHOUT_CANONICAL = 5
for i in range(self.NUMBER_ACTIVE):
cb = CanonicalBuilding(active=True)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb)
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.org
b.save()
for i in range(NUMBER_INACTIVE):
cb = CanonicalBuilding(active=False)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb)
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.org
b.save()
for i in range(NUMBER_WITHOUT_CANONICAL):
b = SEEDFactory.building_snapshot()
b.super_organization = self.org
b.save()
def _create_buildings(self):
"""
Create 10 buildings in each child org.
Also set one shared and one unshared field to a known value.
"""
for _ in range(10):
cb = CanonicalBuilding(active=True)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb,
property_name='ADMIN BUILDING',
address_line_1='100 Admin St')
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.parent_org
b.save()
for _ in range(10):
cb = CanonicalBuilding(active=True)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb,
property_name='ENG BUILDING',
address_line_1='100 Eng St')
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.eng_org
b.save()
for _ in range(10):
cb = CanonicalBuilding(active=True)
cb.save()
b = SEEDFactory.building_snapshot(canonical_building=cb,
property_name='DES BUILDING',
address_line_1='100 Des St')
cb.canonical_snapshot = b
cb.save()
b.super_organization = self.des_org
b.save()
