import smart_imports smart_imports.all() class ArtifactFormsTests(utils_testcase.TestCase): def setUp(self): super(ArtifactFormsTests, self).setUp() game_logic.create_test_map() def get_form_data(self): artifact = logic.create_random_artifact_record(uuid='sword') data = linguistics_helpers.get_word_post_data(artifact.utg_name, prefix='name') data.update({ 'level': '1', 'type': 'ARTIFACT_TYPE.RING', 'power_type': 'ARTIFACT_POWER_TYPE.NEUTRAL', 'rare_effect': 'ARTIFACT_EFFECT.POISON', 'epic_effect': 'ARTIFACT_EFFECT.GREAT_PHYSICAL_DAMAGE', 'special_effect': 'ARTIFACT_EFFECT.NO_EFFECT', 'description': 'artifact description', 'weapon_type': 'WEAPON_TYPE.TYPE_1', 'material': 'MATERIAL.MATERIAL_3', 'uuid': 'some-uuid', 'mob': ''}) return data
import smart_imports smart_imports.all() def collect_data(account_id): data = [] for third_party in models.AccessToken.objects.filter(account=account_id): data.append(('third_party_token', { 'created_at': third_party.created_at, 'updated_at': third_party.updated_at, 'uid': third_party.uid, 'application_name': third_party.application_name, 'application_info': third_party.application_info, 'application_description': third_party.application_description, 'state': third_party.state })) return data def remove_data(account_id): models.AccessToken.objects.filter(account=account_id).delete()