예제 #1
0
    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')
예제 #2
0
 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)