def test_check(self): provider = Reference() @account_check def test(self, account): pass provider.test = test provider.test(provider, PaymentAccount(provider=PROVIDER_REFERENCE)) with self.assertRaises(ValueError): provider.test(provider, PaymentAccount(provider=PROVIDER_BOKU))
def get_or_create_payment_account(email='*****@*****.**', name='Fake App Developer'): user, _ = UserProfile.objects.get_or_create( email=email, source=mkt.LOGIN_SOURCE_UNKNOWN, display_name=name) try: acct = PaymentAccount.objects.get(user=user) except PaymentAccount.DoesNotExist: acct = Reference().account_create( user, {'account_name': name, 'name': name, 'email': email}) return acct
def setUp(self, *args, **kw): super(TestReference, self).setUp(*args, **kw) self.user = UserProfile.objects.get(pk=999) self.ref = Reference()
def setUp(self): self.user = UserProfile.objects.get(pk=999) self.ref = Reference()
def generate_app_from_spec(name, categories, type, status, num_previews=1, num_ratings=1, locale_names=('en-US', 'es-ES'), preview_files=(), video_files=(), developer_name='Fake App Developer', developer_email='*****@*****.**', privacy_policy='Fake privacy policy', premium_type='free', description=None, default_locale='en-US', rereview=False, special_regions={}, inapp_id=None, inapp_secret=None, popularity=0, tarako=False, **spec): status = STATUS_CHOICES_API_LOOKUP[status] names = generate_localized_names(name, locale_names) if type == 'hosted': app = generate_hosted_app( names[default_locale], categories, developer_name, status=status, default_locale=default_locale, **spec) else: app = generate_packaged_app( names, type, categories, developer_name, default_locale=default_locale, status=status, **spec) generate_icon(app) if not preview_files: generate_previews(app, num_previews) if video_files: for i, f in enumerate(video_files): t = tempfile.mktemp() copy_stored_file(f, t, src_storage=local_storage, dst_storage=private_storage) p = Preview.objects.create(addon=app, filetype="video/webm", caption="video " + str(i), position=i) resize_video(t, p.pk) if preview_files: for i, f in enumerate(preview_files): p = Preview.objects.create(addon=app, filetype="image/png", caption="screenshot " + str(i), position=i + len(video_files)) t = tempfile.mktemp() copy_stored_file(f, t, src_storage=local_storage, dst_storage=private_storage) resize_preview(t, p.pk) generate_ratings(app, num_ratings) app.name = names if not description: description = GENERIC_DESCRIPTION app.description = description app.privacy_policy = privacy_policy app.support_email = developer_email premium_type = mkt.ADDON_PREMIUM_API_LOOKUP[premium_type] app.premium_type = premium_type app.default_locale = default_locale if popularity: app.popularity.create(value=popularity) if premium_type != mkt.ADDON_FREE and status != mkt.STATUS_NULL: acct = get_or_create_payment_account(developer_email, developer_name) product_uri = Reference().product_create(acct, app) AddonPaymentAccount.objects.create(addon=app, payment_account=acct, account_uri=acct.uri, product_uri=product_uri) if premium_type in (mkt.ADDON_PREMIUM, mkt.ADDON_PREMIUM_INAPP): price = get_or_create_price(spec.get('price', '0.99')) AddonPremium.objects.create(addon=app, price=price) if premium_type in (mkt.ADDON_FREE_INAPP, mkt.ADDON_PREMIUM_INAPP): UserInappKey.create(acct.user, secret=inapp_secret, public_id=inapp_id, access_type=ACCESS_PURCHASE) for optField in ('support_url', 'homepage', 'is_offline'): if optField in spec: setattr(app, optField, spec[optField]) # Status has to be updated at the end because STATUS_DELETED apps can't # be saved. app.status = status app.save() for (region, region_status) in special_regions.iteritems(): app.geodata.update(**{'region_%s_nominated' % (region,): datetime.datetime.now(), 'region_%s_status' % (region,): STATUS_CHOICES_API_LOOKUP[region_status]}) if tarako: AdditionalReview.objects.create(app=app, queue='tarako') addon_review_aggregates(app.pk) if rereview: RereviewQueue.objects.get_or_create(addon=app) try: u = UserProfile.objects.get(email=developer_email) except UserProfile.DoesNotExist: u = create_user(developer_email) u.display_name = developer_name u.save() AddonUser.objects.create(user=u, addon=app) return app