def setUp(self):
self.configs = configs.copy()
self.service = MockExecuteSQL(**self.configs)
sql = 'select * from ' + schema + '.' + query
self.expected_kwargs = {
'expected_args': [self.service.get_server_url()]
, 'data': {'sql': sql, "schemaName": schema}
, 'headers': None
, 'timeout': 30
}
self.args = [
server_context, schema, sql
]
class TestExecuteSQL(unittest.TestCase):
def setUp(self):
self.configs = configs.copy()
self.service = MockExecuteSQL(**self.configs)
sql = 'select * from ' + schema + '.' + query
self.expected_kwargs = {
'expected_args': [self.service.get_server_url()]
, 'data': {'sql': sql, "schemaName": schema}
, 'headers': None
, 'timeout': 30
}
self.args = [
server_context, schema, sql
]
def test_success(self):
test = self
success_test(test, self.service.get_successful_response(), execute_sql, *self.args, **self.expected_kwargs)
def test_unauthorized(self):
test = self
throws_error_test(test, RequestAuthorizationError, self.service.get_unauthorized_response()
, execute_sql, *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()
, execute_sql, *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()
, execute_sql, *self.args, **self.expected_kwargs)
def test_general_error(self):
test = self
throws_error_test(test, RequestError, self.service.get_general_error_response()
, execute_sql, *self.args, **self.expected_kwargs)