def test_quiet_mod(self):
"""Modify a view without print."""
view = create_view()
argv = ['', 'view', 'mod', '-i', str(view.id), '-n', 'NEW NAME', '-q']
Command().run_from_argv(argv)
# Test stdout
out = self.stdout.getvalue()
self.assertFalse(out, "Output is printed:\n"+out)
def test_mod(self):
"""Modify a view."""
view = create_view()
argv = ['', 'view', 'mod', '-i', str(view.id), '-n', 'NEW NAME']
Command().run_from_argv(argv)
# Test mod
view = View.objects.get(id=view.id)
self.assertEqual(view.name, 'NEW NAME', "View's name wasn't modified.")
def test_delete_several(self):
"""Delete a several views."""
create_view() ; create_view()
IDS = ','.join([ str(s.id) for s in View.objects.all() ])
argv = ['', 'view', 'del', '-i', IDS]
Command().run_from_argv(argv)
# Test deletion
new_count = View.objects.count()
self.assertEqual(new_count, 0, "All views wasn't deleted.")
def test_add(self):
"""Add a view."""
source_ids = ','.join([ str(s.id) for s in Source.objects.all()[:2] ])
argv = ['', 'view', 'add', '-n', 'TEST VIEW', '-s', source_ids]
Command().run_from_argv(argv)
# Test creation
new_count = View.objects.count()
out = self.stdout.getvalue()
self.assertGreater(new_count, 0, "View wasn't created.")
def test_list(self):
"""Get listing."""
self.view = create_view()
argv = ['', 'view', 'list']
Command().run_from_argv(argv)
# Test stdout
out = self.stdout.getvalue()
self.assertTrue(out, "No output.")
self.assertIn("Count: 1", out, "Output isn't showing total count.")
def test_rm_sources(self):
"""Remove sources from a view."""
view = create_view()
source_ids = ','.join([ str(v.id) for v in view.sources.all()[:1] ])
default_count = view.sources.count()
argv = ['', 'view', 'mod', '-i', str(view.id), '--rm-sources', source_ids ]
Command().run_from_argv(argv)
# Test mod
view = View.objects.get(id=view.id)
new_count = view.sources.count()
self.assertGreater(default_count, new_count, "Sources wasn't removed.")
def test_add_sources(self):
"""Add sources to a view."""
view = create_view()
view_source_ids = [ v.id for v in view.sources.all() ]
default_count = len(view_source_ids)
# Set new sources
sources = Source.objects.exclude(id__in=view_source_ids)[:2]
source_ids = ','.join([ str(s.id) for s in sources ])
argv = ['', 'view', 'mod', '-i', str(view.id), '--add-sources', source_ids ]
Command().run_from_argv(argv)
# Test mod
view = View.objects.get(id=view.id)
new_count = view.sources.count()
self.assertLess(default_count, new_count, "Sources wasn't added.")