def test_last_user_deletes_notebook(self):
from augias.views import notebook_delete_post
user = User(email='*****@*****.**')
notebook = Notebook(name='wizzz', users=[user])
DBSession.add_all([user, notebook])
self.assertEqual(Notebook.query().count(), 1)
request = DummyRequest(user=user)
result = notebook_delete_post(notebook, request)
self.assertEqual(Notebook.query().count(), 0)
def test_create_notebook_if_user_has_none_and_no_invites(self):
user = User(email='*****@*****.**')
self.assertEqual(Notebook.query().count(), 0)
result = self._call_view(user=user)
self.assertEqual(Notebook.query().count(), 1)
notebook = Notebook.query().one()
self.assertEqual(notebook.users, [user])
self.assertEqual(result.code, 302)
self.assertEqual(result.location, DummyRequest().resource_url(notebook))
def test_redirect_to_notebook_if_user_has_one_and_no_invites(self):
user = User(email='*****@*****.**')
notebook = Notebook(name='some notebook', users=[user])
DBSession.add_all([user, notebook])
DBSession.flush()
self.assertEqual(Notebook.query().count(), 1)
result = self._call_view(user=user)
self.assertEqual(Notebook.query().count(), 1)
self.assertEqual(result.code, 302)
self.assertEqual(result.location, DummyRequest().resource_url(notebook))
def test_new_notebook(self):
from augias.views import new_notebook
user = User(email='*****@*****.**')
request = DummyRequest({'name': 'First notebook'},
user=user)
result = new_notebook(None, request)
self.assertEqual(Notebook.query().count(), 1)
n = Notebook.query().one()
self.assertEqual(n.name, 'First notebook')
self.assertEqual(n.users, [user])
