From ef86d4458c21f1d722fb162423cd3dfeea055844 Mon Sep 17 00:00:00 2001 From: sw00 Date: Tue, 10 Nov 2015 00:09:18 -0200 Subject: refactor unit test don't use initialize method of root_resource in test --- service/test/unit/resources/test_root_resource.py | 28 +++++++---------------- 1 file changed, 8 insertions(+), 20 deletions(-) (limited to 'service/test/unit/resources/test_root_resource.py') diff --git a/service/test/unit/resources/test_root_resource.py b/service/test/unit/resources/test_root_resource.py index 068a531f..4cdaf97a 100644 --- a/service/test/unit/resources/test_root_resource.py +++ b/service/test/unit/resources/test_root_resource.py @@ -1,4 +1,5 @@ import unittest +import re from mockito import mock, when from test.unit.resources import DummySite from twisted.web.test.requesthelper import DummyRequest @@ -6,20 +7,13 @@ from pixelated.resources.root_resource import RootResource class TestRootResource(unittest.TestCase): + MAIL_ADDRESS = 'test_user@pixelated-project.org' def setUp(self): - test_email = 'hackerman@pixelated-project.org' - mail_service = mock() - mail_service.account_email = test_email - root_resource = RootResource() - root_resource.initialize(mock(), mock(), mail_service, mock(), mock()) - root_resource._html_template = """ - - - $account_email - - """ + root_resource._mode = root_resource + root_resource.account_email = self.MAIL_ADDRESS + root_resource._html_template = "$account_email" self.web = DummySite(root_resource) def test_render_GET_should_template_account_email(self): @@ -28,15 +22,9 @@ class TestRootResource(unittest.TestCase): d = self.web.get(request) def assert_response(_): - expected = """ - - - hackerman@pixelated.org - - """ - - actual = request.written[0] - self.assertEquals(expected, actual) + expected = "{0}".format(self.MAIL_ADDRESS) + matches = re.findall(expected, request.written[0]) + self.assertEquals(len(matches), 1) d.addCallback(assert_response) return d -- cgit v1.2.3