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()
