63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
|
# -*- coding: utf-8; -*-
|
||
|
|
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
from pyramid import testing
|
||
|
|
from beaker.session import Session as BeakerSession
|
||
|
|
|
||
|
|
from wuttaweb import progress as mod
|
||
|
|
|
||
|
|
|
||
|
|
class TestGetBasicSession(TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.request = testing.DummyRequest()
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
session = mod.get_basic_session(self.request)
|
||
|
|
self.assertIsInstance(session, BeakerSession)
|
||
|
|
self.assertFalse(session.use_cookies)
|
||
|
|
|
||
|
|
|
||
|
|
class TestGetProgressSession(TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.request = testing.DummyRequest()
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
self.request.session.id = 'mockid'
|
||
|
|
session = mod.get_progress_session(self.request, 'foo')
|
||
|
|
self.assertIsInstance(session, BeakerSession)
|
||
|
|
self.assertEqual(session.id, 'mockid.progress.foo')
|
||
|
|
|
||
|
|
|
||
|
|
class TestSessionProgress(TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.request = testing.DummyRequest()
|
||
|
|
self.request.session.id = 'mockid'
|
||
|
|
|
||
|
|
def test_error_url(self):
|
||
|
|
factory = mod.SessionProgress(self.request, 'foo', success_url='/blart')
|
||
|
|
self.assertEqual(factory.error_url, '/blart')
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
|
||
|
|
# sanity / coverage check
|
||
|
|
factory = mod.SessionProgress(self.request, 'foo')
|
||
|
|
prog = factory("doing things", 2)
|
||
|
|
prog.update(1)
|
||
|
|
prog.update(2)
|
||
|
|
prog.handle_success()
|
||
|
|
|
||
|
|
def test_error(self):
|
||
|
|
|
||
|
|
# sanity / coverage check
|
||
|
|
factory = mod.SessionProgress(self.request, 'foo')
|
||
|
|
prog = factory("doing things", 2)
|
||
|
|
prog.update(1)
|
||
|
|
try:
|
||
|
|
raise RuntimeError('omg')
|
||
|
|
except Exception as error:
|
||
|
|
prog.handle_error(error)
|