def test_retrieve_rows(self, mock_uri, mock_get_from_uri, mock_presto_config): client = PrestoClient('any_host', 'any_user') dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/resources/valid_rest_response_level1.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = [ "http://localhost:8080/v1/statement/2015_harih/2", "" ] self.assertEqual(client.get_rows(), []) self.assertEqual(client.next_uri, "http://localhost:8080/v1/statement/2015_harih/2") with open(dir + '/resources/valid_rest_response_level2.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_uri.side_effect = [ "http://localhost:8080/v1/statement/2015_harih/2", "" ] expected_row = [[ "uuid1", "http://localhost:8080", "presto-main:0.97", True ], ["uuid2", "http://worker:8080", "presto-main:0.97", False]] self.assertEqual(client.get_rows(), expected_row) self.assertEqual(client.next_uri, "")
def test_retrieve_rows(self, mock_uri, mock_get_from_uri, mock_port): mock_port.return_value = 8080 client = PrestoClient('any_host', 'any_user') dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/resources/valid_rest_response_level1.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = [ "http://localhost:8080/v1/statement/2015_harih/2", "" ] self.assertEqual(client.get_rows(), []) self.assertEqual(client.next_uri, "http://localhost:8080/v1/statement/2015_harih/2") with open(dir + '/resources/valid_rest_response_level2.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_uri.side_effect = [ "http://localhost:8080/v1/statement/2015_harih/2", "" ] expected_row = [["uuid1", "http://localhost:8080", "presto-main:0.97", True], ["uuid2", "http://worker:8080", "presto-main:0.97", False]] self.assertEqual(client.get_rows(), expected_row) self.assertEqual(client.next_uri, "")
def test_limit_rows(self, mock_uri, mock_get_from_uri): client = PrestoClient('any_host', 'any_user', 8080) dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/files/valid_rest_response_level2.txt') as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = ["any_next_uri", ""] self.assertEqual(client.get_rows(0), [])
def test_limit_rows(self, mock_uri, mock_get_from_uri, mock_presto_config): client = PrestoClient('any_host', 'any_user') dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/resources/valid_rest_response_level2.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = ["any_next_uri", ""] self.assertEqual(client.get_rows(0), [])
def test_execute_query_get_port(self, run_mock, conn_mock): client = PrestoClient('any_host', 'any_user') client.rows = ['hello'] client.next_uri = 'hello' client.response_from_server = {'hello': 'hello'} run_mock.return_value = _AttributeString('http-server.http.port=8080') run_mock.return_value.failed = False client.execute_query('select * from nation') self.assertEqual(client.port, 8080) self.assertEqual(client.rows, []) self.assertEqual(client.next_uri, '') self.assertEqual(client.response_from_server, {})
def testrun_sql_get_port(self, sudo_mock, conn_mock, mock_presto_config): client = PrestoClient('any_host', 'any_user') client.rows = ['hello'] client.next_uri = 'hello' client.response_from_server = {'hello': 'hello'} sudo_mock.return_value = _AttributeString('http-server.http.port=8080') sudo_mock.return_value.failed = False sudo_mock.return_value.return_code = 0 client.run_sql('select * from nation') self.assertEqual(client.port, 8080) self.assertEqual(client.rows, []) self.assertEqual(client.next_uri, '') self.assertEqual(client.response_from_server, {})
def testrun_sql_get_port(self, sudo_mock, conn_mock, mock_presto_config): client = PrestoClient('any_host', 'any_user') client.rows = ['hello'] client.next_uri = 'hello' client.response_from_server = {'hello': 'hello'} sudo_mock.return_value = _AttributeString('http-server.http.port=8080') sudo_mock.return_value.failed = False sudo_mock.return_value.return_code = 0 client.run_sql('select * from nation') self.assertEqual(client.port, 8080) self.assertEqual(client.rows, []) self.assertEqual(client.next_uri, '') self.assertEqual(client.response_from_server, {})
def test_append_rows(self, mock_uri, mock_get_from_uri, mock_port): mock_port.return_value = 8080 client = PrestoClient('any_host', 'any_user') dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/resources/valid_rest_response_level2.txt') \ as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = ["any_next_uri", "any_next_next_uri", "", ""] expected_row = [ ["uuid1", "http://localhost:8080", "presto-main:0.97", True], ["uuid2", "http://worker:8080", "presto-main:0.97", False], ["uuid1", "http://localhost:8080", "presto-main:0.97", True], ["uuid2", "http://worker:8080", "presto-main:0.97", False] ] self.assertEqual(client.get_rows(), expected_row)
def test_append_rows(self, mock_uri, mock_get_from_uri): client = PrestoClient('any_host', 'any_user', 8080) dir = os.path.abspath(os.path.dirname(__file__)) with open(dir + '/files/valid_rest_response_level2.txt') as json_file: client.response_from_server = json.load(json_file) mock_get_from_uri.return_value = True mock_uri.side_effect = ["any_next_uri", "any_next_next_uri", "", ""] expected_row = [["uuid1", "http://localhost:8080", "presto-main:0.97", True], ["uuid2", "http://worker:8080", "presto-main:0.97", False], ["uuid1", "http://localhost:8080", "presto-main:0.97", True], ["uuid2", "http://worker:8080", "presto-main:0.97", False]] self.assertEqual(client.get_rows(), expected_row)