def test_collect_stats_functions(self): i1 = Item.objects.get(id='item_1_1') i1.stats = i1.collect_stats() i1.save() self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1'])) i2 = Item.objects.get(id='item_1_2') i2.stats = i2.collect_stats() i2.save() self.assertTrue(utils.compare_dicts(i2.stats, self.expected['items']['i2'])) i3 = Item.objects.get(id='item_1_3') i3.stats = i3.collect_stats() i3.save() self.assertTrue(utils.compare_dicts(i3.stats, self.expected['items']['i3'])) p1 = Project.objects.get(id='project_1_1') p1.stats = p1.collect_stats() p1.save() self.assertTrue(utils.compare_dicts(p1.stats, self.expected['projects']['p1'])) c1 = Collection.objects.get(id='collection_1_1') c1.stats = c1.collect_stats() c1.save() self.assertTrue(utils.compare_dicts(c1.stats, self.expected['collections']['c1']))
def test_update_object_stats(self): # test by passing object directly i1 = Item.objects.get(id='item_1_1') utils.update_object_stats(i1) self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1'])) # test by passing model and id utils.update_object_stats(model=Item, id='item_1_2') i2 = Item.objects.get(id='item_1_2') self.assertTrue(utils.compare_dicts(i2.stats, self.expected['items']['i2']))
def test_update_model_stats(self): errors = utils.update_model_stats(Item) self.assertEqual(errors, []) self.assertTrue(not errors) i1 = Item.objects.get(id='item_1_1') self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1'])) i2 = Item.objects.get(id='item_1_2') self.assertTrue(utils.compare_dicts(i2.stats, self.expected['items']['i2'])) i3 = Item.objects.get(id='item_1_3') self.assertTrue(utils.compare_dicts(i3.stats, self.expected['items']['i3']))
def test_stats_on_empty_objects(self): expected = self.empty_stats c1 = Collection(id='nobagscollection', name='test-collection-2', created=now()) c1.save() self.assertTrue(utils.compare_dicts(c1.stats, expected)) p1 = Project(id='nobagsproject', name='test-project-2', collection=c1, created=now()) p1.save() self.assertTrue(utils.compare_dicts(p1.stats, expected)) i1 = Item(id='nobagsitem', title='test-item-1', project=p1, collection=c1, created=now(), original_item_type='1') i1.stats = i1.collect_stats() i1.save() p1.stats = p1.collect_stats() c1.stats = c1.collect_stats() self.assertTrue(utils.compare_dicts(i1.stats, expected)) self.assertTrue(utils.compare_dicts(p1.stats, expected)) self.assertTrue(utils.compare_dicts(c1.stats, expected))
def test_mgmt_cmd_all(self): call_command('update_stats', All=True) i1 = Item.objects.get(id='item_1_1') self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1'])) i2 = Item.objects.get(id='item_1_2') self.assertTrue(utils.compare_dicts(i2.stats, self.expected['items']['i2'])) i3 = Item.objects.get(id='item_1_3') self.assertTrue(utils.compare_dicts(i3.stats, self.expected['items']['i3'])) p1 = Project.objects.get(id='project_1_1') self.assertTrue(utils.compare_dicts(p1.stats, self.expected['projects']['p1'])) c1 = Collection.objects.get(id='collection_1_1') self.assertTrue(utils.compare_dicts(c1.stats, self.expected['collections']['c1']))
def test_update_all_stats(self): errors = utils.update_all_stats() self.assertEqual(errors, []) self.assertTrue(not errors) i1 = Item.objects.get(id='item_1_1') self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1'])) i2 = Item.objects.get(id='item_1_2') self.assertTrue(utils.compare_dicts(i2.stats, self.expected['items']['i2'])) i3 = Item.objects.get(id='item_1_3') self.assertTrue(utils.compare_dicts(i3.stats, self.expected['items']['i3'])) p1 = Project.objects.get(id='project_1_1') self.assertTrue(utils.compare_dicts(p1.stats, self.expected['projects']['p1'])) c1 = Collection.objects.get(id='collection_1_1') self.assertTrue(utils.compare_dicts(c1.stats, self.expected['collections']['c1']))
def test_mgmt_cmd_single_item(self): call_command('update_stats', item='item_1_1') i1 = Item.objects.get(id='item_1_1') self.assertTrue(utils.compare_dicts(i1.stats, self.expected['items']['i1']))