class MiddlewareTest(unittest.TestCase): def setUp(self) -> None: self.middleware = Middleware() def test_call(self): self.middleware(None) def test_add_market(self): mock_market = mock.Mock() self.assertIsNone(self.middleware.add_market(mock_market)) def test_remove_market(self): mock_market = mock.Mock() self.assertIsNone(self.middleware.remove_market(mock_market))
def setUp(self) -> None: self.middleware = Middleware()