From b8162073cdb6dd8ceb82e580ce58355fc0073b55 Mon Sep 17 00:00:00 2001 From: Jacob Beck Date: Mon, 21 Oct 2019 08:01:32 -0600 Subject: [PATCH] bump those timeouts... --- test/integration/048_rpc_test/test_rpc.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/test/integration/048_rpc_test/test_rpc.py b/test/integration/048_rpc_test/test_rpc.py index 546915d8e34..b9bb0f81a96 100644 --- a/test/integration/048_rpc_test/test_rpc.py +++ b/test/integration/048_rpc_test/test_rpc.py @@ -199,11 +199,15 @@ def poll_for_result(self, request_token, request_id=1, timeout=60, status='succe .format(delta, status, result) ) - def async_query(self, _method, _sql=None, _test_request_id=1, macros=None, **kwargs): + def async_query(self, _method, _sql=None, _test_request_id=1, _poll_timeout=60, macros=None, **kwargs): response = self.query(_method, _sql, _test_request_id, macros, **kwargs).json() result = self.assertIsResult(response, _test_request_id) self.assertIn('request_token', result) - return self.poll_for_result(result['request_token'], _test_request_id) + return self.poll_for_result( + result['request_token'], + request_id=_test_request_id, + timeout=_poll_timeout, + ) def query(self, _method, _sql=None, _test_request_id=1, macros=None, **kwargs): built = self.build_query(_method, kwargs, _sql, _test_request_id, macros) @@ -1145,7 +1149,7 @@ def test_deps_compilation_postgres(self): status = self._check_start_predeps() # do a dbt deps, wait for the result - self.assertIsResult(self.async_query('deps').json()) + self.assertIsResult(self.async_query('deps', _poll_timeout=120).json()) self._check_deps_ok(status) @@ -1154,7 +1158,7 @@ def test_deps_cli_compilation_postgres(self): status = self._check_start_predeps() # do a dbt deps, wait for the result - self.assertIsResult(self.async_query('cli_args', cli='deps').json()) + self.assertIsResult(self.async_query('cli_args', cli='deps', _poll_timeout=120).json()) self._check_deps_ok(status) @@ -1203,4 +1207,4 @@ def test_postgres_status_error(self): self.assertIn('Invalid test config', str(data['message'])) # deps should work - self.assertIsResult(self.async_query('deps').json()) + self.assertIsResult(self.async_query('deps', _poll_timeout=120).json())