Skip to content

Commit

Permalink
Add test_vega_crud_view_no_list_action
Browse files Browse the repository at this point in the history 
Test that VegaCRUD view with no list action works.
  • Loading branch information
@moshthepitt
moshthepitt committed Feb 19, 2020
1 parent c48dc40 commit 12eb2c4
Showing 1 changed file with 96 additions and 106 deletions.
202 changes: 96 additions & 106 deletions tests/test_views.py
View file
Original file line number Diff line number Diff line change
@@ -1,141 +1,140 @@
"""
vega-admin module to test views
"""
"""vega-admin module to test views."""
from django.conf import settings
from django.contrib.auth.models import Permission, User
from django.contrib.contenttypes.models import ContentType
from django.test import TestCase, override_settings

from model_mommy import mommy

from vega_admin.views import (VegaCreateView, VegaCRUDView, VegaDeleteView,
VegaDetailView, VegaListView, VegaUpdateView)
from vega_admin.views import (
VegaCreateView,
VegaCRUDView,
VegaDeleteView,
VegaDetailView,
VegaListView,
VegaUpdateView,
)

from .artist_app.forms import ArtistForm
from .artist_app.models import Artist, Song
from .artist_app.views import (ArtistCreate, ArtistDelete, ArtistListView,
ArtistRead, ArtistUpdate)
from .artist_app.views import (
ArtistCreate,
ArtistDelete,
ArtistListView,
ArtistRead,
ArtistUpdate,
)


class TestViewsBase(TestCase):
"""
Base test class for views
"""

def _song_permissions(self):
"""
Create permissions
"""
"""Base test class for views."""

maxDiff = None

def _song_permissions(self): # pylint: disable=no-self-use
"""Create permissions."""
content_type = ContentType.objects.get_for_model(Song)
list_permission, _ = Permission.objects.get_or_create(
codename='list_song',
codename="list_song",
content_type=content_type,
defaults=dict(name='Can List Songs'),
defaults=dict(name="Can List Songs"),
)
create_permission, _ = Permission.objects.get_or_create(
codename='create_song',
codename="create_song",
content_type=content_type,
defaults=dict(name='Can Create Songs'),
defaults=dict(name="Can Create Songs"),
)
view_permission, _ = Permission.objects.get_or_create(
codename='view_song',
codename="view_song",
content_type=content_type,
defaults=dict(name='Can View Songs'),
defaults=dict(name="Can View Songs"),
)
update_permission, _ = Permission.objects.get_or_create(
codename='update_song',
codename="update_song",
content_type=content_type,
defaults=dict(name='Can Update Songs'),
defaults=dict(name="Can Update Songs"),
)
delete_permission, _ = Permission.objects.get_or_create(
codename='delete_song',
codename="delete_song",
content_type=content_type,
defaults=dict(name='Can Delete Songs'),
defaults=dict(name="Can Delete Songs"),
)
artists_permission, _ = Permission.objects.get_or_create(
codename='artists_song',
codename="artists_song",
content_type=content_type,
defaults=dict(name='Can List Song Artists'),
defaults=dict(name="Can List Song Artists"),
)
return [list_permission, create_permission, update_permission,
delete_permission, artists_permission, view_permission, ]
return [
list_permission,
create_permission,
update_permission,
delete_permission,
artists_permission,
view_permission,
]

def _artist_permissions(self):
"""
Create permissions
"""
def _artist_permissions(self): # pylint: disable=no-self-use
"""Create permissions."""
content_type = ContentType.objects.get_for_model(Artist)
list_permission, _ = Permission.objects.get_or_create(
codename='list_artist',
codename="list_artist",
content_type=content_type,
defaults=dict(name='Can List Artists'),
defaults=dict(name="Can List Artists"),
)
other_permission, _ = Permission.objects.get_or_create(
codename='other_artist',
codename="other_artist",
content_type=content_type,
defaults=dict(name='Can `Other` Artists'),
defaults=dict(name="Can `Other` Artists"),
)
return [list_permission, other_permission, ]
return [list_permission, other_permission]

def setUp(self):
"""setUp"""
"""Set up."""
super().setUp()
self.maxDiff = None
self.user = User.objects.create_user(
username='mosh',
email='[email protected]',
password='hunter2',
username="mosh", email="[email protected]", password="hunter2"
)

def tearDown(self):
"""tearDown"""
"""Tear down."""
super().tearDown()
Song.objects.all().delete()
Artist.objects.all().delete()
Permission.objects.filter(
content_type=ContentType.objects.get_for_model(Song)).delete()
content_type=ContentType.objects.get_for_model(Song)
).delete()
Permission.objects.filter(
content_type=ContentType.objects.get_for_model(Artist)).delete()
content_type=ContentType.objects.get_for_model(Artist)
).delete()
User.objects.all().delete()


@override_settings(
ROOT_URLCONF="tests.artist_app.urls",
VEGA_TEMPLATE="basic"
)
@override_settings(ROOT_URLCONF="tests.artist_app.urls", VEGA_TEMPLATE="basic")
class TestViews(TestViewsBase):
"""
Test class for views
"""
"""Test class for views."""

def test_vega_crud_view(self):
"""
Test VegaCRUDView
"""
"""Test VegaCRUDView."""
default_actions = ["create", "view", "update", "list", "delete"]

class ArtistCrud(VegaCRUDView):
"""ArtistCrud class."""

model = Artist

view = ArtistCrud()

self.assertEqual(Artist, view.model)
self.assertEqual(
list(set(default_actions)), list(set(view.get_actions())))
self.assertEqual(list(set(default_actions)), list(set(view.get_actions())))
self.assertEqual("artist_app.artist", view.crud_path)
self.assertEqual("artist_app", view.app_label)
self.assertEqual("artist", view.model_name)

self.assertEqual(
Artist, view.get_view_class_for_action("create")().model)
self.assertEqual(
Artist, view.get_view_class_for_action("update")().model)
self.assertEqual(
Artist, view.get_view_class_for_action("delete")().model)
self.assertEqual(
Artist, view.get_view_class_for_action("list")().model)
self.assertEqual(
Artist, view.get_view_class_for_action("view")().model)
self.assertEqual(Artist, view.get_view_class_for_action("create")().model)
self.assertEqual(Artist, view.get_view_class_for_action("update")().model)
self.assertEqual(Artist, view.get_view_class_for_action("delete")().model)
self.assertEqual(Artist, view.get_view_class_for_action("list")().model)
self.assertEqual(Artist, view.get_view_class_for_action("view")().model)

self.assertIsInstance(
view.get_view_class_for_action("create")(), VegaCreateView
Expand All @@ -146,10 +145,8 @@ class ArtistCrud(VegaCRUDView):
self.assertIsInstance(
view.get_view_class_for_action("delete")(), VegaDeleteView
)
self.assertIsInstance(
view.get_view_class_for_action("list")(), VegaListView)
self.assertIsInstance(
view.get_view_class_for_action("view")(), VegaDetailView)
self.assertIsInstance(view.get_view_class_for_action("list")(), VegaListView)
self.assertIsInstance(view.get_view_class_for_action("view")(), VegaDetailView)

self.assertEqual(
f"{view.crud_path}/create/",
Expand Down Expand Up @@ -184,15 +181,12 @@ class ArtistCrud(VegaCRUDView):

for action in default_actions:
self.assertEqual(
f"{view.crud_path}-{action}",
view.get_url_name_for_action(action)
f"{view.crud_path}-{action}", view.get_url_name_for_action(action)
)

@override_settings(VEGA_FORCE_ORDERING=True, VEGA_ORDERING_FIELD=["pk"])
def test_vega_list_view(self):
"""
Test VegaListView
"""
"""Test VegaListView."""
Artist.objects.all().delete()

artist = mommy.make("artist_app.Artist", name="Bob")
Expand All @@ -207,21 +201,18 @@ def test_vega_list_view(self):
self.assertTemplateUsed(res, "vega_admin/basic/list.html")

res = self.client.get("/list/artists/?q=Bob")
self.assertDictEqual({
"q": "Bob"
}, res.context["vega_listview_search_form"].initial)
self.assertDictEqual(
{"q": "Bob"}, res.context["vega_listview_search_form"].initial
)
self.assertEqual(
set(["q", ]),
set(res.context["vega_listview_search_form"].fields.keys())
set(["q"]), set(res.context["vega_listview_search_form"].fields.keys())
)
self.assertEqual(res.context["object_list"].count(), 1)
self.assertTrue(res.context["object_list"].ordered)
self.assertEqual(res.context["object_list"].first(), artist)

def test_vega_view_view(self):
"""
Test VegaReadView
"""
"""Test VegaReadView."""
artist = mommy.make("artist_app.Artist", name="Bob")
res = self.client.get(f"/view/artists/view/{artist.id}")
self.assertEqual(res.status_code, 200)
Expand All @@ -230,9 +221,7 @@ def test_vega_view_view(self):
self.assertTemplateUsed(res, "vega_admin/basic/read.html")

def test_vega_create_view(self):
"""
Test VegaCreateView
"""
"""Test VegaCreateView."""
res = self.client.get("/edit/artists/create/")
self.assertEqual(res.status_code, 200)
self.assertIsInstance(res.context["form"], ArtistForm)
Expand All @@ -244,52 +233,43 @@ def test_vega_create_view(self):
self.assertRedirects(res, "/edit/artists/create/")
self.assertQuerysetEqual(Artist.objects.all(), ["<Artist: Mosh>"])
self.assertTrue(
settings.VEGA_FORM_VALID_CREATE_TXT in
res.cookies["messages"].value
settings.VEGA_FORM_VALID_CREATE_TXT in res.cookies["messages"].value
)
self.assertFalse(
settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value
)

res = self.client.post("/edit/artists/create/", {})
self.assertEqual(res.status_code, 200)
self.assertTrue(
settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value)
self.assertTrue(settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value)

def test_vega_update_view(self):
"""
Test VegaUpdateView
"""
"""Test VegaUpdateView."""
artist = mommy.make("artist_app.Artist", name="Bob")
res = self.client.get(f"/edit/artists/edit/{artist.id}")
self.assertEqual(res.status_code, 200)
self.assertIsInstance(res.context["form"], ArtistForm)
self.assertIsInstance(res.context["view"], ArtistUpdate)
self.assertIsInstance(res.context["view"], VegaUpdateView)
self.assertTemplateUsed(res, "vega_admin/basic/update.html")
res = self.client.post(
f"/edit/artists/edit/{artist.id}", {"name": "Mosh"})
res = self.client.post(f"/edit/artists/edit/{artist.id}", {"name": "Mosh"})
self.assertEqual(res.status_code, 302)
self.assertRedirects(res, f"/edit/artists/edit/{artist.id}")
artist.refresh_from_db()
self.assertEqual("Mosh", artist.name)
self.assertTrue(
settings.VEGA_FORM_VALID_UPDATE_TXT in
res.cookies["messages"].value
settings.VEGA_FORM_VALID_UPDATE_TXT in res.cookies["messages"].value
)
self.assertFalse(
settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value
)

res = self.client.post(f"/edit/artists/edit/{artist.id}", {})
self.assertEqual(res.status_code, 200)
self.assertTrue(
settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value)
self.assertTrue(settings.VEGA_FORM_INVALID_TXT in res.cookies["messages"].value)

def test_vega_delete_view(self):
"""
Test ArtistDelete
"""
"""Test ArtistDelete."""
artist = mommy.make("artist_app.Artist", name="Bob")
res = self.client.get(f"/edit/artists/delete/{artist.id}")
self.assertEqual(res.status_code, 200)
Expand All @@ -307,7 +287,17 @@ def test_vega_delete_view(self):
mommy.make("artist_app.Song", name="Nuts", artist=artist2)
res = self.client.post(f"/edit/artists/delete/{artist2.id}")
self.assertTrue(
settings.VEGA_DELETE_PROTECTED_ERROR_TXT in
res.cookies["messages"].value
settings.VEGA_DELETE_PROTECTED_ERROR_TXT in res.cookies["messages"].value
)
self.assertTrue(Artist.objects.filter(id=artist2.id).exists())

def test_vega_crud_view_no_list_action(self):
"""Test Vega CRUD view with no list action."""
res = self.client.get("/create-artist-only/create/")
# we are checking that the cancel button is rendered as expected
self.assertContains(
res,
"""<a href="/" class="btn btn-default btn-block vega-cancel">Cancel</a>""",
status_code=200,
html=True,
)

0 comments on commit 12eb2c4

Please sign in to comment.