def test_adu_flag(self): addon = Addon.objects.get(pk=3615) now = datetime.datetime.now() counter = UpdateCount.objects.create(addon=addon, date=now, count=1234) counter.save() self.assertTrue( addon.average_daily_users > addon.total_downloads + 10000, 'Unexpected ADU count. ADU of %d not greater than %d' % ( addon.average_daily_users, addon.total_downloads + 10000)) adu = cron.update_addon_average_daily_users flag_reindexing_amo('new', 'old', 'alias') try: # Should fail. self.assertRaises(CommandError, adu) # Should work with the environ flag. os.environ['FORCE_INDEXING'] = '1' adu() finally: unflag_reindexing_amo() del os.environ['FORCE_INDEXING'] addon = Addon.objects.get(pk=3615) eq_(addon.average_daily_users, 1234)
def flag_database(new_index, old_index, alias, stdout=sys.stdout): """Flags the database to indicate that the reindexing has started.""" log('Flagging the database to start the reindexation', stdout=stdout) flag_reindexing_amo(new_index=new_index, old_index=old_index, alias=alias)