34 lines
956 B
Python
34 lines
956 B
Python
|
from django.test import TestCase
|
||
|
|
from django.http.request import HttpRequest
|
||
|
|
from resume import set_resume
|
||
|
|
from resume.middleware import ResumeStorage
|
||
|
|
|
||
|
|
# Create your tests here.
|
||
|
|
class ResumeTests(TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
self.request = HttpRequest()
|
||
|
|
self.request.path = 'test/foo'
|
||
|
|
self.request.session = {}
|
||
|
|
self.request._resume_points = ResumeStorage(self.request)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super().tearDown()
|
||
|
|
del self.request
|
||
|
|
|
||
|
|
def test_resume_point_storage(self):
|
||
|
|
|
||
|
|
assert(len(self.request._resume_points) == 0)
|
||
|
|
|
||
|
|
no_resume_point = self.request._resume_points.pop()
|
||
|
|
assert(no_resume_point is None)
|
||
|
|
|
||
|
|
set_resume(self.request)
|
||
|
|
|
||
|
|
assert(len(self.request._resume_points) == 1)
|
||
|
|
|
||
|
|
one_resume_point = self.request._resume_points.pop()
|
||
|
|
assert(one_resume_point == 'test/foo')
|
||
|
|
assert(len(self.request._resume_points) == 0)
|