def test_archive(self):
original_list = Product.by_domain(self.domain.name, wrap=False)
self.products[0].archive()
new_list = Product.by_domain(self.domain.name, wrap=False)
self.assertTrue(
self.products[0]._id not in [p['_id'] for p in new_list],
"Archived product still returned by Product.by_domain()")
self.assertEqual(len(new_list), len(original_list) - 1)
self.assertEqual(
len(
Product.by_domain(self.domain.name,
wrap=False,
include_archived=True)), len(original_list))
self.assertEqual(
len(
Product.archived_by_domain(self.domain.name,
wrap=False,
include_archived=True)), 1)
self.products[0].unarchive()
self.assertEqual(len(original_list),
len(Product.by_domain(self.domain.name, wrap=False)))
def product_data(self):
data = []
if self.show_inactive:
products = Product.archived_by_domain(
domain=self.domain,
limit=self.limit,
skip=self.skip(),
)
else:
products = Product.by_domain(
domain=self.domain,
limit=self.limit,
skip=self.skip(),
)
for p in products:
if p.program_id:
program = Program.get(p.program_id)
else:
program = get_or_create_default_program(self.domain)
p.program_id = program.get_id
p.save()
info = p._doc
info['program'] = program.name
info['edit_url'] = reverse('commtrack_product_edit', kwargs={'domain': self.domain, 'prod_id': p._id})
info['archive_action_desc'] = self.get_archive_text(self.show_inactive)
info['archive_action_text'] = _("Un-Archive") if self.show_inactive else _("Archive")
info['archive_url'] = reverse(
'unarchive_product' if self.show_inactive else 'archive_product',
kwargs={'domain': self.domain, 'prod_id': p._id}
)
data.append(info)
return data
def test_archive(self):
original_list = Product.by_domain(self.domain.name, wrap=False)
self.products[0].archive()
new_list = Product.by_domain(self.domain.name, wrap=False)
self.assertTrue(
self.products[0]._id not in [p['_id'] for p in new_list],
"Archived product still returned by Product.by_domain()"
)
self.assertEqual(
len(new_list),
len(original_list) - 1
)
self.assertEqual(
len(Product.by_domain(self.domain.name, wrap=False, include_archived=True)),
len(original_list)
)
self.assertEqual(
len(Product.archived_by_domain(self.domain.name, wrap=False, include_archived=True)),
1
)
self.products[0].unarchive()
self.assertEqual(
len(original_list),
len(Product.by_domain(self.domain.name, wrap=False))
)
def product_data(self):
data = []
if self.show_inactive:
products = Product.archived_by_domain(
domain=self.domain,
limit=self.limit,
skip=self.skip(),
)
else:
products = Product.by_domain(
domain=self.domain,
limit=self.limit,
skip=self.skip(),
)
for p in products:
if p.program_id:
program = Program.get(p.program_id)
else:
program = get_or_make_def_program(self.domain)
p.program_id = program.get_id
p.save()
info = p._doc
info['program'] = program.name
info['edit_url'] = reverse('commtrack_product_edit', kwargs={'domain': self.domain, 'prod_id': p._id})
info['archive_action_desc'] = self.get_archive_text(self.show_inactive)
info['archive_action_text'] = _("Un-Archive") if self.show_inactive else _("Archive")
info['archive_url'] = reverse(
'unarchive_product' if self.show_inactive else 'archive_product',
kwargs={'domain': self.domain, 'prod_id': p._id}
)
data.append(info)
return data
