def test_asrsnippet(self):
nightly_snippets = ASRSnippetFactory.create_batch(
2, targets=[TargetFactory(on_release=False, on_nightly=True)])
ASRSnippetFactory.create_batch(2, targets=[TargetFactory(on_release=False, on_beta=True)])
filtr = ChannelFilter(None, {'channel': 'on_nightly'}, ASRSnippet, ASRSnippetAdmin)
result = filtr.queryset(None, ASRSnippet.objects.all())
self.assertTrue(result.count(), 2)
self.assertEqual(set(result.all()), set(nightly_snippets))
def test_asrsnippet(self):
nightly_snippets = ASRSnippetFactory.create_batch(
2, targets=[TargetFactory(on_release=False, on_nightly=True)])
ASRSnippetFactory.create_batch(
2, targets=[TargetFactory(on_release=False, on_beta=True)])
filtr = ChannelFilter(None, {'channel': 'on_nightly'}, ASRSnippet,
ASRSnippetAdmin)
result = filtr.queryset(None, ASRSnippet.objects.all())
self.assertTrue(result.count(), 2)
self.assertEqual(set(result.all()), set(nightly_snippets))
def test_get_queryset(self):
snippets = ASRSnippetFactory.create_batch(2)
qa_snippets = ASRSnippetFactory.create_batch(2, for_qa=True)
request = self.factory.get('/')
admin = ASRSnippetAdmin(ASRSnippet, AdminSite())
# Not Super User
request.user = UserFactory(is_superuser=False)
queryset = admin.get_queryset(request)
self.assertEqual(set(snippets), set(queryset.all()))
# SuperUser
request.user = UserFactory(is_superuser=True)
queryset = admin.get_queryset(request)
self.assertEqual(set(snippets + qa_snippets), set(queryset.all()))
def test_get_queryset(self):
snippets = ASRSnippetFactory.create_batch(2)
qa_snippets = ASRSnippetFactory.create_batch(2, for_qa=True)
request = self.factory.get('/')
admin = ASRSnippetAdmin(ASRSnippet, AdminSite())
# Not Super User
request.user = UserFactory(is_superuser=False)
queryset = admin.get_queryset(request)
self.assertEqual(set(snippets), set(queryset.all()))
# SuperUser
request.user = UserFactory(is_superuser=True)
queryset = admin.get_queryset(request)
self.assertEqual(set(snippets + qa_snippets), set(queryset.all()))
def test_base(self):
snippet1, snippet2 = ASRSnippetFactory.create_batch(2)
snippet3 = ASRSnippetFactory.create(publish_start=datetime(2019, 1, 1))
snippet4 = ASRSnippetFactory.create(publish_end=datetime(2019, 2, 2))
filtr = ASRSnippetFilter(QueryDict(),
queryset=models.ASRSnippet.objects.all())
self.assertEqual(set([snippet1, snippet2, snippet3, snippet4]),
set(filtr.qs))
def test_make_published(self):
to_be_published = ASRSnippetFactory.create_batch(2, status=STATUS_CHOICES['Draft'])
already_published = ASRSnippetFactory(status=STATUS_CHOICES['Published'])
ASRSnippetFactory.create_batch(2, status=STATUS_CHOICES['Draft'])
queryset = ASRSnippet.objects.filter(id__in=[
to_be_published[0].id,
to_be_published[1].id,
already_published.id
])
with patch('snippets.base.admin.adminmodels.messages.warning') as warning:
with patch('snippets.base.admin.adminmodels.messages.success') as success:
ASRSnippetAdmin(ASRSnippet, None).make_published(None, queryset)
self.assertEqual(
set(ASRSnippet.objects.filter(status=STATUS_CHOICES['Published'])),
set(to_be_published + [already_published])
)
self.assertTrue(warning.called)
self.assertTrue(success.called)
def test_make_published(self):
to_be_published = ASRSnippetFactory.create_batch(
2, status=STATUS_CHOICES['Draft'])
already_published = ASRSnippetFactory(
status=STATUS_CHOICES['Published'])
ASRSnippetFactory.create_batch(2, status=STATUS_CHOICES['Draft'])
queryset = ASRSnippet.objects.filter(id__in=[
to_be_published[0].id, to_be_published[1].id, already_published.id
])
with patch(
'snippets.base.admin.adminmodels.messages.warning') as warning:
with patch('snippets.base.admin.adminmodels.messages.success'
) as success:
ASRSnippetAdmin(ASRSnippet,
None).make_published(None, queryset)
self.assertEqual(
set(ASRSnippet.objects.filter(status=STATUS_CHOICES['Published'])),
set(to_be_published + [already_published]))
self.assertTrue(warning.called)
self.assertTrue(success.called)
def test_only_scheduled_all(self):
snippet1, snippet2 = ASRSnippetFactory.create_batch(2)
snippet3 = ASRSnippetFactory.create(publish_end=datetime(2019, 2, 2))
filtr = ASRSnippetFilter(QueryDict(query_string='only_scheduled=all'),
queryset=models.ASRSnippet.objects.all())
self.assertEqual(set([snippet1, snippet2, snippet3]), set(filtr.qs))
def setUp(self):
self.snippet1, self.snippet2 = ASRSnippetFactory.create_batch(2)
def test_only_scheduled_all(self):
snippet1, snippet2 = ASRSnippetFactory.create_batch(2)
snippet3 = ASRSnippetFactory.create(publish_end=datetime(2019, 2, 2))
filtr = ASRSnippetFilter(QueryDict(query_string='only_scheduled=all'),
queryset=models.ASRSnippet.objects.all())
self.assertEqual(set([snippet1, snippet2, snippet3]), set(filtr.qs))
def test_base(self):
snippet1, snippet2 = ASRSnippetFactory.create_batch(2)
snippet3 = ASRSnippetFactory.create(publish_start=datetime(2019, 1, 1))
snippet4 = ASRSnippetFactory.create(publish_end=datetime(2019, 2, 2))
filtr = ASRSnippetFilter(QueryDict(), queryset=models.ASRSnippet.objects.all())
self.assertEqual(set([snippet1, snippet2, snippet3, snippet4]), set(filtr.qs))