def test_presto_endpoint(self):
presto = PrestoQueryEngine("1/XXX", "api.treasuredata.com",
"sample_datasets", True)
self.assertEqual(presto.endpoint, "https://api.treasuredata.com")
presto = PrestoQueryEngine("1/XXX", "http://api.treasuredata.com",
"sample_datasets", True)
self.assertEqual(presto.endpoint, "http://api.treasuredata.com")
presto = PrestoQueryEngine("1/XXX", "https://api.treasuredata.com",
"sample_datasets", True)
self.assertEqual(presto.endpoint, "https://api.treasuredata.com")
def test_create_header(self):
presto_no_header = PrestoQueryEngine('1/XXX', 'https://api.treasuredata.com/', 'sample_datasets', False)
self.assertEquals(presto_no_header.create_header('foo'), '')
ua = self.presto.user_agent
header = self.presto.create_header()
self.assertEquals(header, "-- client: {0}\n".format(ua))
header = self.presto.create_header('foo')
self.assertEquals(header, "-- client: {0}\n-- foo\n".format(ua))
header = self.presto.create_header(['foo', 'bar'])
self.assertEquals(header, "-- client: {0}\n-- foo\n-- bar\n".format(ua))
def test_create_header(self):
presto_no_header = PrestoQueryEngine(
"1/XXX", "https://api.treasuredata.com/", "sample_datasets", False
)
self.assertEqual(presto_no_header.create_header("foo"), "")
ua = self.presto.user_agent
header = self.presto.create_header()
self.assertEqual(header, f"-- client: {ua}\n")
header = self.presto.create_header("foo")
self.assertEqual(header, f"-- client: {ua}\n-- foo\n")
header = self.presto.create_header(["foo", "bar"])
self.assertEqual(header, f"-- client: {ua}\n-- foo\n-- bar\n")
def setUp(self, connect):
self.presto = PrestoQueryEngine(
"1/XXX", "https://api.treasuredata.com/", "sample_datasets", True
)
self.assertTrue(connect.called)
self.assertEqual(self.presto.executed, None)
def setUp(self, connect):
self.presto = PrestoQueryEngine('1/XXX', 'https://api.treasuredata.com/', 'sample_datasets', True)
self.assertTrue(connect.called)
