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']))
