def fetch_amo_user(email):
log.debug('#'*80)
amo = AMOOAuth(domain=settings.AMOOAUTH_DOMAIN,
port=settings.AMOOAUTH_PORT,
protocol=settings.AMOOAUTH_PROTOCOL,
prefix=settings.AMOOAUTH_PREFIX)
amo.set_consumer(consumer_key=settings.AMOOAUTH_CONSUMERKEY,
consumer_secret=settings.AMOOAUTH_CONSUMERSECRET)
return amo.get_user_by_email(email) or None
def setUp(self):
self.author = User.objects.get(username='******')
self.addonrev = Package.objects.get(name='test-addon',
author__username='******').latest
self.hashtag = hashtag()
self.amo = AMOOAuth(domain=settings.AMOOAUTH_DOMAIN,
port=settings.AMOOAUTH_PORT,
protocol=settings.AMOOAUTH_PROTOCOL,
prefix=settings.AMOOAUTH_PREFIX)
def fetch_amo_user(email):
log.debug('#' * 80)
amo = AMOOAuth(domain=settings.AMOOAUTH_DOMAIN,
port=settings.AMOOAUTH_PORT,
protocol=settings.AMOOAUTH_PROTOCOL,
prefix=settings.AMOOAUTH_PREFIX)
amo.set_consumer(consumer_key=settings.AMOOAUTH_CONSUMERKEY,
consumer_secret=settings.AMOOAUTH_CONSUMERSECRET)
return amo.get_user_by_email(email) or None
class UploadTest(TestCase):
fixtures = ["mozilla_user", "users", "core_sdk", "packages"]
ADDON_AMO_ID = 1
AMO_FILE_ID = 20
def setUp(self):
self.author = User.objects.get(username="******")
self.addonrev = Package.objects.get(name="test-addon", author__username="******").latest
self.hashtag = hashtag()
self.amo = AMOOAuth(
domain=settings.AMOOAUTH_DOMAIN,
port=settings.AMOOAUTH_PORT,
protocol=settings.AMOOAUTH_PROTOCOL,
prefix=settings.AMOOAUTH_PREFIX,
)
def test_create_new_amo_addon(self):
AMOOAuth._send = Mock(
return_value={"status": STATUS_NULL, "id": self.ADDON_AMO_ID} # current result from AMO API
)
upload_to_amo(self.addonrev.pk, self.hashtag)
# checking status and other attributes
addonrev = Package.objects.get(name="test-addon", author__username="******").latest
eq_(addonrev.package.amo_id, self.ADDON_AMO_ID)
eq_(addonrev.amo_version_name, "initial")
eq_(addonrev.amo_status, STATUS_NULL)
# check if right API was called
assert "POST" in AMOOAuth._send.call_args[0]
assert self.amo.url("addon") in AMOOAuth._send.call_args[0]
def test_update_amo_addon(self):
send_backup = AMOOAuth._send
AMOOAuth._send = Mock(return_value={"id": self.AMO_FILE_ID})
# set add-on as uploaded
self.addonrev.amo_status = STATUS_PUBLIC
self.addonrev.amo_version_name = self.addonrev.get_version_name()
self.addonrev.package.amo_id = self.ADDON_AMO_ID
self.addonrev.package.latest_uploaded = self.addonrev
self.addonrev.package.save()
# create a new "clean" revision
self.addonrev.save()
assert not self.addonrev.amo_version_name
assert not self.addonrev.amo_status
# upload it to AMO
upload_to_amo(self.addonrev.pk, self.hashtag)
# test status and other attributes
addonrev = Package.objects.get(name="test-addon", author__username="******").latest
eq_(addonrev.pk, addonrev.package.latest_uploaded.pk)
eq_(addonrev.amo_version_name, "initial.rev%d" % addonrev.revision_number)
eq_(addonrev.amo_file_id, self.AMO_FILE_ID)
eq_(addonrev.amo_status, STATUS_UNREVIEWED) # hardcoded
# check if right API was called
assert "POST" in AMOOAuth._send.call_args[0]
assert self.amo.url("version") % self.ADDON_AMO_ID in AMOOAuth._send.call_args[0]
AMOOAuth._send = send_backup
def test_get_details_from_amo(self):
get_addon_details_backup = helpers.get_addon_details
# create a fake uploaded revision
self.addonrev.amo_status = STATUS_NULL
self.addonrev.amo_version_name = self.addonrev.get_version_name()
self.addonrev.package.amo_id = self.ADDON_AMO_ID
self.addonrev.package.latest_uploaded = self.addonrev
self.addonrev.package.save()
# create a new "clean" revision
self.addonrev.save()
assert not self.addonrev.package.get_view_on_amo_url()
helpers.get_addon_details = Mock(
return_value={
"status": STATUS_NAMES[STATUS_PUBLIC],
"status_code": STATUS_PUBLIC,
"rating": 0,
"version": self.addonrev.amo_version_name,
"slug": "some-slug",
}
)
# check the AMO status of the addon
r = self.client.get(reverse("amo_get_addon_details", args=[self.addonrev.pk]))
eq_(r.status_code, 200)
addonrev = Package.objects.get(name="test-addon", author__username="******").latest
eq_(addonrev.pk, self.addonrev.pk)
eq_(addonrev.amo_status, STATUS_PUBLIC)
eq_(addonrev.package.amo_slug, "some-slug")
assert addonrev.package.get_view_on_amo_url()
helpers.get_addon_details = get_addon_details_backup
class UploadTest(TestCase):
fixtures = ['mozilla_user', 'users', 'core_sdk', 'packages']
ADDON_AMO_ID = 1
AMO_FILE_ID = 20
def setUp(self):
self.author = User.objects.get(username='******')
self.addonrev = Package.objects.get(name='test-addon',
author__username='******').latest
self.hashtag = hashtag()
self.amo = AMOOAuth(domain=settings.AMOOAUTH_DOMAIN,
port=settings.AMOOAUTH_PORT,
protocol=settings.AMOOAUTH_PROTOCOL,
prefix=settings.AMOOAUTH_PREFIX)
def tearDown(self):
AMOOAuth._send = OLD_AMOOAUTH_SEND
def test_create_new_amo_addon(self):
AMOOAuth._send = Mock(return_value={
'status': STATUS_NULL, # current result from AMO API
'id': self.ADDON_AMO_ID})
upload_to_amo(self.addonrev.pk, self.hashtag)
# checking status and other attributes
addonrev = Package.objects.get(name='test-addon',
author__username='******').latest
eq_(addonrev.package.amo_id, self.ADDON_AMO_ID)
eq_(addonrev.amo_version_name, 'initial')
eq_(addonrev.amo_status, STATUS_NULL)
# check if right API was called
assert 'POST' in AMOOAuth._send.call_args[0]
assert self.amo.url('addon') in AMOOAuth._send.call_args[0]
def test_update_amo_addon(self):
send_backup = AMOOAuth._send
AMOOAuth._send = Mock(return_value={'id': self.AMO_FILE_ID})
# set add-on as uploaded
self.addonrev.amo_status = STATUS_PUBLIC
self.addonrev.amo_version_name = self.addonrev.get_version_name()
self.addonrev.package.amo_id = self.ADDON_AMO_ID
self.addonrev.package.latest_uploaded = self.addonrev
self.addonrev.package.save()
# create a new "clean" revision
self.addonrev.save()
assert not self.addonrev.amo_version_name
assert not self.addonrev.amo_status
# upload it to AMO
upload_to_amo(self.addonrev.pk, self.hashtag)
# test status and other attributes
addonrev = Package.objects.get(name='test-addon',
author__username='******').latest
eq_(addonrev.pk, addonrev.package.latest_uploaded.pk)
eq_(addonrev.amo_version_name,
'initial.rev%d' % addonrev.revision_number)
eq_(addonrev.amo_file_id, self.AMO_FILE_ID)
eq_(addonrev.amo_status, STATUS_UNREVIEWED) # hardcoded
# check if right API was called
assert 'POST' in AMOOAuth._send.call_args[0]
assert self.amo.url('version') % self.ADDON_AMO_ID in AMOOAuth._send.call_args[0]
AMOOAuth._send = send_backup
def test_get_details_from_amo(self):
get_addon_details_backup = helpers.get_addon_details
# create a fake uploaded revision
self.addonrev.amo_status = STATUS_NULL
self.addonrev.amo_version_name = self.addonrev.get_version_name()
self.addonrev.package.amo_id = self.ADDON_AMO_ID
self.addonrev.package.latest_uploaded = self.addonrev
self.addonrev.package.save()
# create a new "clean" revision
self.addonrev.save()
assert not self.addonrev.package.get_view_on_amo_url()
helpers.get_addon_details = Mock(
return_value={'status': STATUS_NAMES[STATUS_PUBLIC],
'status_code': STATUS_PUBLIC,
'rating': 0,
'version': self.addonrev.amo_version_name,
'slug': 'some-slug'})
# check the AMO status of the addon
r = self.client.get(reverse('amo_get_addon_details',
args=[self.addonrev.pk]))
eq_(r.status_code, 200)
addonrev = Package.objects.get(name='test-addon',
author__username='******').latest
eq_(addonrev.pk, self.addonrev.pk)
eq_(addonrev.amo_status, STATUS_PUBLIC)
eq_(addonrev.package.amo_slug, 'some-slug')
assert addonrev.package.get_view_on_amo_url()
helpers.get_addon_details = get_addon_details_backup
