class TestRouterView(TestCase): def setUp(self): self.request = mock.Mock() self.router = MarimoRouter() def tearDown(self): pass def test_get(self): self.assertRaises(Http404, self.router.get, HttpRequest()) @mock.patch('marimo.views.router._marimo_widgets', widgets) @mock.patch('marimo.views.router.HttpResponse') def test_route_success(self, http_response): bulk = [ {'id':'1', 'widget_name':'test', 'args':['one', 'two'], 'kwargs':{}} ] self.router.route(self.request, bulk) response = json.loads(http_response.call_args[0][0]) self.assertEqual(len(response), 1) self.assertEqual(response[0]['status'], 'succeeded') self.assertEqual(response[0]['id'], '1') @mock.patch('marimo.views.router._marimo_widgets', widgets) @mock.patch('marimo.views.router.HttpResponse') def test_route_callable_fails(self, http_response): bulk = [ {'id':'1', 'widget_name':'failure', 'args':['one', 'two'], 'kwargs':{}} ] self.router.route(self.request, bulk) response = json.loads(http_response.call_args[0][0]) self.assertEquals(response[0]['status'], 'failed') @mock.patch('marimo.views.router._marimo_widgets', widgets) @mock.patch('marimo.views.router.HttpResponse') def test_route_no_such_widget(self, http_response): bulk = [ {'id':'1', 'widget_name':'nopechucktesta', 'args':['one', 'two'], 'kwargs':{}} ] self.router.route(self.request, bulk) response = json.loads(http_response.call_args[0][0]) self.assertEquals(response[0]['status'], 'WidgetNotFound')
def setUp(self): self.request = mock.Mock() self.router = MarimoRouter()