예제 #1
0
class ContextMiddlewareTestCase(ValidatorTestCase):
    """ Tests for class ContextMiddleware """

    def setUp(self):
        """ Create a ContextMiddleware instance """
        super(ContextMiddlewareTestCase, self).setUp()
        self.item = ContextMiddleware()


    def test_make_context(self):
        """ Tests for method make_context """
        self.item.external = mock.MagicMock()
        input = "MyInput"
        expected = "OK"
        self.item.external.return_value = "OK"
        observed = self.item.make_context(input)
        self.assertEqual(expected, observed)

    def test_process_request(self):
        """ Tests for method process_request """
        self.item.external = mock.MagicMock()
        input = "MyInput"
        expected = "OK"
        self.item.external.return_value = "OK"
        observed = self.item.process_request(input)
        self.assertEqual(expected, observed)

    def tearDown(self):
        """ Cleanup the ContextMiddleware instance """
        super(ContextMiddlewareTestCase, self).tearDown()
        self.m.UnsetStubs()
        self.m.ResetAll()
예제 #2
0
 def setUp(self):
     """ Create a ContextMiddleware instance """
     super(ContextMiddlewareTestCase, self).setUp()
     self.item = ContextMiddleware()