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()
