Exemple #1
0
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()