示例#1
0
    def setUp(self):
        self.configs = configs.copy()
        self.service = MockSelectRows(**self.configs)
        self.expected_kwargs = {
            'expected_args': [self.service.get_server_url()]
            , 'data': {"schemaName": schema, "query.queryName": query}
            , 'headers': None
            , 'timeout': 30
        }

        self.args = [
            server_context, schema, query
        ]
示例#2
0
class TestSelectRows(unittest.TestCase):

    def setUp(self):
        self.configs = configs.copy()
        self.service = MockSelectRows(**self.configs)
        self.expected_kwargs = {
            'expected_args': [self.service.get_server_url()]
            , 'data': {"schemaName": schema, "query.queryName": query}
            , 'headers': None
            , 'timeout': 30
        }

        self.args = [
            server_context, schema, query
        ]

    def test_success(self):
        test = self
        success_test(test, self.service.get_successful_response(), select_rows, *self.args, **self.expected_kwargs)

    def test_unauthorized(self):
        test = self
        throws_error_test(test, RequestAuthorizationError, self.service.get_unauthorized_response()
                          , select_rows, *self.args, **self.expected_kwargs)

    def test_query_not_found(self):
        test = self
        throws_error_test(test, QueryNotFoundError,  self.service.get_query_not_found_response()
                          , select_rows, *self.args, **self.expected_kwargs)

    def test_server_not_found(self):
        test = self
        throws_error_test(test, ServerNotFoundError, self.service.get_server_not_found_response()
                          , select_rows, *self.args, **self.expected_kwargs)

    def test_general_error(self):
        test = self
        throws_error_test(test, RequestError, self.service.get_general_error_response()
                          , select_rows, *self.args, **self.expected_kwargs)