def test_process__has_only_new_active_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), created_at=datetime.datetime.now()) self.request.process() self.check_not_processed()
def test_process__only_premium_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), premium_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.request.process() self.check_not_processed(premiums=2)
def setUp(self): super(RandomPremiumRequestPrototypeTests, self).setUp() create_test_map() self.account_1 = self.accounts_factory.create_account() self.account_2 = self.accounts_factory.create_account() AccountPrototype._db_all().update(created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) self.request = RandomPremiumRequestPrototype.create(self.account_1.id, days=30)
def test_run_random_premium_requests_processing__has_requests_can_process(self): account_2 = self.accounts_factory.create_account() AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) request = RandomPremiumRequestPrototype.create(self.account.id, days=30) self.worker.run_random_premium_requests_processing() request.reload() self.assertTrue(request.state.is_PROCESSED) self.assertEqual(request.receiver_id, account_2.id)
def setUp(self): super(RandomPremiumRequestPrototypeTests, self).setUp() create_test_map() result, account_id, bundle_id = register_user('test_user_1', '*****@*****.**', '111111') self.account_1 = AccountPrototype.get_by_id(account_id) result, account_id, bundle_id = register_user('test_user_2', '*****@*****.**', '111111') self.account_2 = AccountPrototype.get_by_id(account_id) AccountPrototype._db_all().update(created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) self.request = RandomPremiumRequestPrototype.create(self.account_1.id, days=30)
def setUp(self): super(RandomPremiumRequestPrototypeTests, self).setUp() create_test_map() self.account_1 = self.accounts_factory.create_account() self.account_2 = self.accounts_factory.create_account() AccountPrototype._db_all().update( created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) self.request = RandomPremiumRequestPrototype.create(self.account_1.id, days=30)
def test_run_random_premium_requests_processing__has_requests_can_process(self): result, account_id, bundle_id = register_user('test_user_2', '*****@*****.**', '111111') account_2 = AccountPrototype.get_by_id(account_id) AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) request = RandomPremiumRequestPrototype.create(self.account.id, days=30) self.worker.run_random_premium_requests_processing() request.reload() self.assertTrue(request.state.is_PROCESSED) self.assertEqual(request.receiver_id, account_2.id)
def test_run_random_premium_requests_processing__has_requests_can_process( self): account_2 = self.accounts_factory.create_account() AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) request = RandomPremiumRequestPrototype.create(self.account.id, days=30) self.worker.run_random_premium_requests_processing() request.reload() self.assertTrue(request.state.is_PROCESSED) self.assertEqual(request.receiver_id, account_2.id)
def test_process__has_active_accounts(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.request.process() self.assertEqual(MessagePrototype._db_count(), 1) message = MessagePrototype._db_get_object(0) self.assertEqual(message.recipient_id, self.account_2.id) self.assertEqual(message.sender_id, get_system_user().id) self.assertEqual(list(AccountPrototype._db_filter(premium_end_at__gt=datetime.datetime.now()).values_list('id', flat=True)), [self.account_2.id]) self.request.reload() self.assertTrue(self.request.state.is_PROCESSED) self.assertEqual(self.request.receiver_id, self.account_2.id)
def test_process__has_active_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) with self.check_new_message(self.account_2.id, [logic.get_system_user_id()]): self.request.process() self.assertEqual( list( AccountPrototype._db_filter( premium_end_at__gt=datetime.datetime.now()).values_list( 'id', flat=True)), [self.account_2.id]) self.request.reload() self.assertTrue(self.request.state.is_PROCESSED) self.assertEqual(self.request.receiver_id, self.account_2.id)
def get_achievements_source_iterator(self, achievement): from the_tale.accounts.prototypes import AccountPrototype if achievement.type.source.is_ACCOUNT: return (AccountPrototype(model=account_model) for account_model in AccountPrototype._db_all()) if achievement.type.source.is_GAME_OBJECT: return (heroes_logic.load_hero(hero_model=hero_model) for hero_model in heroes_models.Hero.objects.all().iterator())
def get_achievements_source_iterator(self, achievement): from the_tale.accounts.prototypes import AccountPrototype from the_tale.game.heroes.prototypes import HeroPrototype if achievement.type.source.is_ACCOUNT: return (AccountPrototype(model=account_model) for account_model in AccountPrototype._db_all()) if achievement.type.source.is_GAME_OBJECT: return (HeroPrototype(model=hero_model) for hero_model in HeroPrototype._db_all())
def get_achievements_source_iterator(self, achievement): from the_tale.accounts.prototypes import AccountPrototype if achievement.type.source.is_ACCOUNT: return (AccountPrototype(model=account_model) for account_model in AccountPrototype._db_all()) if achievement.type.source.is_GAME_OBJECT: return ( heroes_logic.load_hero(hero_model=hero_model) for hero_model in heroes_models.Hero.objects.all().iterator())
def setUp(self): super(RandomPremiumRequestPrototypeTests, self).setUp() create_test_map() result, account_id, bundle_id = register_user('test_user_1', '*****@*****.**', '111111') self.account_1 = AccountPrototype.get_by_id(account_id) result, account_id, bundle_id = register_user('test_user_2', '*****@*****.**', '111111') self.account_2 = AccountPrototype.get_by_id(account_id) AccountPrototype._db_all().update( created_at=datetime.datetime.now() - accounts_settings.RANDOM_PREMIUM_CREATED_AT_BARRIER) self.request = RandomPremiumRequestPrototype.create(self.account_1.id, days=30)
def test_process__has_active_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.request.process() self.assertEqual(MessagePrototype._db_count(), 1) message = MessagePrototype._db_get_object(0) self.assertEqual(message.recipient_id, self.account_2.id) self.assertEqual(message.sender_id, get_system_user().id) self.assertEqual( list( AccountPrototype._db_filter( premium_end_at__gt=datetime.datetime.now()).values_list( 'id', flat=True)), [self.account_2.id]) self.request.reload() self.assertTrue(self.request.state.is_PROCESSED) self.assertEqual(self.request.receiver_id, self.account_2.id)
def test_process__has_only_new_active_accounts(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), created_at=datetime.datetime.now()) self.request.process() self.check_not_processed()
def test_process__only_active_initiator(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.account_2.remove() self.request.process() self.check_not_processed()
def test_process__only_premium_accounts(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), premium_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.request.process() self.check_not_processed(premiums=2)
def test_process__only_fast_accounts(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), is_fast=True) self.request.process() self.check_not_processed()
def test_process__no_active_accounts(self): AccountPrototype._db_all().update(active_end_at=datetime.datetime.now()) self.request.process() self.check_not_processed()
def test_process__only_active_initiator(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1)) self.account_2.remove() self.request.process() self.check_not_processed()
def test_process__only_fast_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now() + datetime.timedelta(days=1), is_fast=True) self.request.process() self.check_not_processed()
def test_process__no_active_accounts(self): AccountPrototype._db_all().update( active_end_at=datetime.datetime.now()) self.request.process() self.check_not_processed()