def test_search_tools_within_a_category(self):
# Pretend Foxy is the only bookmarks related search add-on
AddonCategory.objects.all().delete()
foxy = Addon.objects.get(name__localized_string='FoxyProxy Standard')
foxy.type = amo.ADDON_SEARCH
foxy.save()
bookmarks = Category.objects.get(slug='bookmarks')
bookmarks.addoncategory_set.add(
AddonCategory(addon=foxy, feature=False))
bookmarks.save()
url = reverse('browse.search-tools.rss',
args=('bookmarks', )) + '?sort=popular'
r = self.client.get(url)
assert r.status_code == 200
doc = pq(r.content)
assert doc('rss channel title')[0].text == (
'Bookmarks :: Search Tools :: Add-ons for Firefox')
link = doc('rss channel link')[0].text
rel_link = reverse('browse.search-tools.rss',
args=('bookmarks', )) + '?sort=popular'
assert link.endswith(rel_link), ('Unexpected link: %r' % link)
assert doc('rss channel description')[0].text == (
"Search tools relating to Bookmarks")
assert [e.text for e in doc('rss channel item title')
] == (['FoxyProxy Standard 2.17'])
def save(self):
category = self.cleaned_data['category']
# Clear any old categor[y|ies]
self.addon.categories.all().delete()
# Add new category
AddonCategory(addon=self.addon, category_id=category).save()
# Remove old, outdated categories cache on the model.
del self.addon.all_categories
def save(self):
category_slug = self.cleaned_data['category']
# Clear any old categor[y|ies]
AddonCategory.objects.filter(addon=self.addon).delete()
# Add new categor[y|ies]
for app in CATEGORIES.keys():
category = CATEGORIES[app].get(self.addon.type,
{}).get(category_slug, None)
if category:
AddonCategory(addon=self.addon, category_id=category.id).save()
# Remove old, outdated categories cache on the model.
del self.addon.all_categories
def save(self, addon):
application = self.cleaned_data.get('application')
categories_new = self.cleaned_data['categories']
categories_old = [cats for app, cats in addon.app_categories if
(app and application and app.id == application)
or (not app and not application)]
if categories_old:
categories_old = categories_old[0]
# Add new categories.
for c in set(categories_new) - set(categories_old):
AddonCategory(addon=addon, category=c).save()
# Remove old categories.
for c in set(categories_old) - set(categories_new):
AddonCategory.objects.filter(addon=addon, category=c).delete()
def save(self, addon):
application = self.cleaned_data.get('application')
categories_new = [c.id for c in self.cleaned_data['categories']]
categories_old = [
c.id for c in
addon.app_categories.get(amo.APP_IDS[application], [])]
# Add new categories.
for c_id in set(categories_new) - set(categories_old):
AddonCategory(addon=addon, category_id=c_id).save()
# Remove old categories.
for c_id in set(categories_old) - set(categories_new):
AddonCategory.objects.filter(
addon=addon, category_id=c_id).delete()
# Remove old, outdated categories cache on the model.
del addon.all_categories
def save(self, commit=False):
data = self.cleaned_data
addon = Addon.objects.create(slug=data.get('slug'),
status=amo.STATUS_PENDING,
type=amo.ADDON_PERSONA)
addon.name = {'en-US': data['name']}
if data.get('description'):
addon.description = data['description']
addon._current_version = Version.objects.create(addon=addon,
version='0')
addon.save()
# Create Persona instance.
p = Persona()
p.persona_id = 0
p.addon = addon
p.header = 'header.png'
if data['footer_hash']:
p.footer = 'footer.png'
if data['accentcolor']:
p.accentcolor = data['accentcolor'].lstrip('#')
if data['textcolor']:
p.textcolor = data['textcolor'].lstrip('#')
p.license = data['license']
p.submit = datetime.now()
user = self.request.user
p.author = user.username
p.display_username = user.name
p.save()
# Save header, footer, and preview images.
save_theme.delay(data['header_hash'], data['footer_hash'], addon)
# Save user info.
addon.addonuser_set.create(user=user, role=amo.AUTHOR_ROLE_OWNER)
# Save tags.
for t in data['tags']:
Tag(tag_text=t).save_tag(addon)
# Save categories.
AddonCategory(addon=addon, category=data['category']).save()
return addon
def save(self, addon):
application = self.cleaned_data.get('application')
categories_new = [c.id for c in self.cleaned_data['categories']]
categories_old = [
c.id for c in
addon.app_categories.get(amo.APP_IDS[application].short, [])]
# Add new categories.
for c_id in set(categories_new) - set(categories_old):
AddonCategory(addon=addon, category_id=c_id).save()
# Remove old categories.
for c_id in set(categories_old) - set(categories_new):
AddonCategory.objects.filter(
addon=addon, category_id=c_id).delete()
# Remove old, outdated categories cache on the model.
del addon.all_categories
# Make sure the add-on is properly re-indexed
addons_tasks.index_addons.delay([addon.id])
