def test_successfull_result(self):
self.assertEqual(AccountAchievementsPrototype._db_count(), 0)
self.assertEqual(AccountItemsPrototype._db_count(), 0)
result, account_id, bundle_id = register_user('test_user',
'*****@*****.**',
'111111')
# test result
self.assertEqual(result, REGISTER_USER_RESULT.OK)
self.assertTrue(bundle_id is not None)
#test basic structure
account = AccountPrototype.get_by_id(account_id)
self.assertEqual(account.nick, 'test_user')
self.assertEqual(account.email, '*****@*****.**')
self.assertTrue(not account.is_fast)
hero = HeroPrototype.get_by_account_id(account.id)
# test hero equipment
self.assertEqual(
hero.equipment.get(EQUIPMENT_SLOT.PANTS).id, 'default_pants')
self.assertEqual(
hero.equipment.get(EQUIPMENT_SLOT.BOOTS).id, 'default_boots')
self.assertEqual(
hero.equipment.get(EQUIPMENT_SLOT.PLATE).id, 'default_plate')
self.assertEqual(
hero.equipment.get(EQUIPMENT_SLOT.GLOVES).id, 'default_gloves')
self.assertEqual(
hero.equipment.get(EQUIPMENT_SLOT.HAND_PRIMARY).id,
'default_weapon')
self.assertTrue(
hero.equipment.get(EQUIPMENT_SLOT.HAND_SECONDARY) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.HELMET) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.SHOULDERS) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.CLOAK) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.AMULET) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.RING) is None)
self.assertEqual(HeroPreferencesPrototype._db_count(), 1)
self.assertEqual(
HeroPreferencesPrototype.get_by_hero_id(
hero.id).energy_regeneration_type,
hero.preferences.energy_regeneration_type)
self.assertEqual(account.referer, None)
self.assertEqual(account.referer_domain, None)
self.assertEqual(account.referral_of_id, None)
self.assertEqual(account.action_id, None)
self.assertEqual(account.is_bot, False)
self.assertEqual(AccountAchievementsPrototype._db_count(), 1)
self.assertEqual(AccountItemsPrototype._db_count(), 1)
self.assertEqual(market_models.Goods.objects.count(), 1)
def _set(self, preferences_type, value, change_time=None):
self.updated = True
if change_time is None:
change_time = datetime.datetime.now()
value = self.value_to_set(value)
self.data[preferences_type.base_name] = {'value': value,
'changed_at': time.mktime(change_time.timetuple())}
HeroPreferencesPrototype.update(self.hero.id, preferences_type.base_name, value)
def test_successfull_result(self):
self.assertEqual(AccountAchievementsPrototype._db_count(), 0)
self.assertEqual(AccountItemsPrototype._db_count(), 0)
result, account_id, bundle_id = register_user('test_user', '*****@*****.**', '111111')
# test result
self.assertEqual(result, REGISTER_USER_RESULT.OK)
self.assertTrue(bundle_id is not None)
#test basic structure
account = AccountPrototype.get_by_id(account_id)
self.assertEqual(account.nick, 'test_user')
self.assertEqual(account.email, '*****@*****.**')
self.assertTrue(not account.is_fast)
hero = HeroPrototype.get_by_account_id(account.id)
# test hero equipment
self.assertEqual(hero.equipment.get(EQUIPMENT_SLOT.PANTS).id, 'default_pants')
self.assertEqual(hero.equipment.get(EQUIPMENT_SLOT.BOOTS).id, 'default_boots')
self.assertEqual(hero.equipment.get(EQUIPMENT_SLOT.PLATE).id, 'default_plate')
self.assertEqual(hero.equipment.get(EQUIPMENT_SLOT.GLOVES).id, 'default_gloves')
self.assertEqual(hero.equipment.get(EQUIPMENT_SLOT.HAND_PRIMARY).id, 'default_weapon')
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.HAND_SECONDARY) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.HELMET) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.SHOULDERS) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.CLOAK) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.AMULET) is None)
self.assertTrue(hero.equipment.get(EQUIPMENT_SLOT.RING) is None)
self.assertEqual(HeroPreferencesPrototype._db_count(), 1)
self.assertEqual(HeroPreferencesPrototype.get_by_hero_id(hero.id).energy_regeneration_type,
hero.preferences.energy_regeneration_type)
self.assertEqual(account.referer, None)
self.assertEqual(account.referer_domain, None)
self.assertEqual(account.referral_of_id, None)
self.assertEqual(account.action_id, None)
self.assertEqual(account.is_bot, False)
self.assertEqual(AccountAchievementsPrototype._db_count(), 1)
self.assertEqual(AccountItemsPrototype._db_count(), 1)
self.assertEqual(market_models.Goods.objects.count(), 1)
def _set(self, preferences_type, value, change_time=None):
self.updated = True
if change_time is None:
change_time = datetime.datetime.now()
value = self.value_to_set(value)
self.data[preferences_type.base_name] = {
'value': value,
'changed_at': time.mktime(change_time.timetuple())
}
HeroPreferencesPrototype.update(self.hero.id,
preferences_type.base_name, value)
