def test_proto_table_writer(self):
conn = MySQLConnection(testing.get_datasource())
rs = conn.query("select * from iris.train limit 10;")
self.assertTrue(rs.success())
tw = table_writer.ProtobufWriter(rs)
lines = tw.dump_strings()
self.assertTrue(lines[0].find(
"head { column_names: \"sepal_length\" column_names: \"sepal_width\" column_names: \"petal_length\" column_names: \"petal_width\" column_names: \"class\" }" # noqa: E501
) >= 0)
def test_query(self):
conn = MySQLConnection(testing.get_datasource())
rs = conn.query("select * from notexist limit 1")
self.assertFalse(rs.success())
rs = conn.query("select * from train limit 1")
self.assertTrue(rs.success())
rows = [r for r in rs]
self.assertEqual(1, len(rows))
rs = conn.query("select * from train limit 20")
self.assertTrue(rs.success())
col_info = rs.column_info()
self.assertEqual([('sepal_length', 'FLOAT'), ('sepal_width', 'FLOAT'),
('petal_length', 'FLOAT'), ('petal_width', 'FLOAT'),
('class', 'INT')], col_info)
rows = [r for r in rs]
self.assertTrue(20, len(rows))
def test_exec(self):
conn = MySQLConnection(testing.get_datasource())
rs = conn.execute("create table test_exec(a int)")
self.assertTrue(rs)
rs = conn.execute("insert into test_exec values(1), (2)")
self.assertTrue(rs)
rs = conn.query("select * from test_exec")
self.assertTrue(rs.success())
rows = [r for r in rs]
self.assertTrue(2, len(rows))
rs = conn.execute("drop table test_exec")
self.assertTrue(rs)
with self.assertRaises(Exception):
conn.execute("drop table not_exist")
def test_connecion(self):
try:
conn = MySQLConnection(testing.get_datasource())
conn.close()
except: # noqa: E722
self.fail()
try:
conn_str = testing.get_datasource()
conn_str = conn_str.replace(":3306", "")
conn = MySQLConnection(conn_str)
conn.close()
except: # noqa: E722
self.fail()
def test_connecion(self):
try:
conn = MySQLConnection(testing.get_datasource())
conn.close()
except: # noqa: E722
self.fail()
def test_get_table_schema(self):
conn = MySQLConnection(testing.get_datasource())
col_info = conn.get_table_schema("iris.train")
self.assertEqual([('sepal_length', 'FLOAT'), ('sepal_width', 'FLOAT'),
('petal_length', 'FLOAT'), ('petal_width', 'FLOAT'),
('class', 'INT')], col_info)