def test_get_func_for_method(self): """ tests the get_func_for_method """ mr = MethodRouter(None, None) def fake(): pass mr.add_route(endpoint_func=fake, methods=['GET', 'post']) def fake2(): pass mr.add_route(endpoint_func=fake2, methods=['PATCH']) self.assertEqual(mr.get_func_for_method('GET'), fake) self.assertEqual(mr.get_func_for_method('get'), fake) self.assertEqual(mr.get_func_for_method('POST'), fake) self.assertEqual(mr.get_func_for_method('post'), fake) self.assertEqual(mr.get_func_for_method('PATCH'), fake2) self.assertEqual(mr.get_func_for_method('patch'), fake2) self.assertRaises(MethodNotAllowed, mr.get_func_for_method, 'PUT') self.assertRaises(MethodNotAllowed, mr.get_func_for_method, 'put')
def test_get_func_for_method_available(self): """ Tests that the get_func_for_method works appropriately """ mr = MethodRouter(None, None) mr.add_route(endpoint_func=1, methods=['GET']) self.assertEqual(mr.get_func_for_method('GET'), 1)