Exemplo n.º 1
0
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('*****@*****.**'))
Exemplo n.º 2
0
 def setUp(self):
     GaeTestCase.setUp(self)
     self.base_view = BaseView()