class BaseViewTest(GaeTestCase): def setUp(self): GaeTestCase.setUp(self) self.base_view = BaseView() def test_is_public(self): def is_public(password, approvals): text = Text(password=password, approvals=approvals) return self.base_view._is_public(text) self.assertTrue(is_public('', [])) self.assertFalse(is_public('password', [])) self.assertFalse(is_public('', ['example.com'])) self.assertFalse(is_public('password', ['example.com'])) def test_to_login(self): self.assertEqual( self.base_view._to_login('%2F')._headers['location'], ('Location', '/auth/login/%2F') ) encoded_url = 'https%3A%2F%2Ftext-n.appspot.com%2F' location = self.base_view._to_login(encoded_url)._headers['location'] self.assertEqual(location, ('Location', '/auth/login/' + encoded_url)) def test_get_current_user_email(self): self.assertIsNone(self.base_view._get_current_user_email()) email = '*****@*****.**' self.login(email) self.assertEqual(self.base_view._get_current_user_email(), email) def test_read_permission(self): self.login('*****@*****.**') text = Text( text='Text', approvals=['*****@*****.**', 'example3.com'] ) text.put() def has_read_permission(email): self.login(email) return self.base_view._has_read_permission(text) self.assertTrue(has_read_permission('*****@*****.**')) self.assertTrue(has_read_permission('*****@*****.**')) self.assertTrue(has_read_permission('*****@*****.**')) self.assertFalse(has_read_permission('*****@*****.**')) self.assertFalse(has_read_permission('*****@*****.**'))
def setUp(self): GaeTestCase.setUp(self) self.base_view = BaseView()