def test_adding_buildings_with_individual_selection(self): """ Ensure that labels are not actually paginated. """ user = User.objects.create_superuser( email='*****@*****.**', username='******', password='******', ) organization, _, _ = create_organization(user, "test-organization") project = Project.objects.create( name='test-org-1', super_organization=organization, owner=user, ) buildings = self.generate_buildings(organization, 10) self.assertFalse(project.building_snapshots.exists()) selected_buildings = [b.pk for b in buildings if b.pk % 2 == 0] self.assertEqual(len(selected_buildings), 5) params = self.get_filter_params(project) params['selected_buildings'] = selected_buildings add_buildings(project.slug, params, user.pk) self.assertEqual(project.building_snapshots.count(), 5)
def test_adding_buildings_with_select_all(self): """ Ensure that labels are not actually paginated. """ user = User.objects.create_superuser( email='*****@*****.**', username='******', password='******', ) organization, _, _ = create_organization(user, "test-organization") project = Project.objects.create( name='test-org-1', super_organization=organization, owner=user, ) self.generate_buildings(organization, 10) self.assertFalse(project.building_snapshots.exists()) params = self.get_filter_params(project) params['select_all_checkbox'] = True add_buildings(project.slug, params, user.pk) self.assertEqual(project.building_snapshots.count(), 10)