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)