def get_chest_intervals_count(self, date): starts = RandomPremiumRequestPrototype._db_all().values_list( 'created_at', flat=True) return len([ True for created_at in starts if created_at.date() <= date < (created_at + datetime.timedelta( days=shop_conf.payments_settings.RANDOM_PREMIUM_DAYS)).date() ])
def test_get_unprocessed__no_waiting(self): RandomPremiumRequestPrototype._db_all().update(state=relations.RANDOM_PREMIUM_REQUEST_STATE.PROCESSED) self.assertEqual(RandomPremiumRequestPrototype.get_unprocessed(), None)
def test_get_unprocessed__not_exist(self): RandomPremiumRequestPrototype._db_all().delete() self.assertEqual(RandomPremiumRequestPrototype.get_unprocessed(), None)
def test_get_unprocessed__no_waiting(self): RandomPremiumRequestPrototype._db_all().update( state=relations.RANDOM_PREMIUM_REQUEST_STATE.PROCESSED) self.assertEqual(RandomPremiumRequestPrototype.get_unprocessed(), None)
def test_get_unprocessed__not_exist(self): RandomPremiumRequestPrototype._db_all().delete() self.assertEqual(RandomPremiumRequestPrototype.get_unprocessed(), None)
def get_chest_intervals_count(self, date): starts = RandomPremiumRequestPrototype._db_all().values_list('created_at', flat=True) return len([True for created_at in starts if created_at.date() <= date < (created_at + datetime.timedelta(days=shop_conf.payments_settings.RANDOM_PREMIUM_DAYS)).date()] )