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 setUp(self): super(TestIndexCommand, self).setUp() if is_reindexing_amo(): unflag_reindexing_amo() self.url = reverse('search.search') # Any index created during the test will be deleted. self.indices = call_es('_status').json()['indices'].keys()
def unflag_database(stdout=sys.stdout): """Unflag the database to indicate that the reindexing is over.""" log('Unflagging the database', stdout=stdout) unflag_reindexing_amo()