def testSqlite3(self):
import sqlite3
builder = QueryBuilder()
conn = sqlite3.connect('tests/Chinook_Sqlite_AutoIncrementPKs.sqlite')
builder.add_table_with_fields('album', 'a', conn)
query = builder.select(['albumid', 'artistid'])
self.assertQuery(query, 'select a.albumid albumid, a.artistid artistid from album a')
def testPostgresql(self):
import psycopg2
builder = QueryBuilder()
conn = psycopg2.connect("dbname=test host=%s port=%s user=%s password=%s" % ( \
self.config.get('postgresql', 'host'),
self.config.get('postgresql', 'port'),
self.config.get('postgresql', 'user'),
self.config.get('postgresql', 'passwd')))
c = conn.cursor()
c.execute('drop table if exists album')
c.execute('''create table album
(AlbumId int, Title char(100), ArtistId int)''')
c.close()
builder.add_table_with_fields('album', 'a', conn)
query = builder.select(['albumid', 'artistid'])
self.assertQuery(query, 'select a.albumid albumid, a.artistid artistid from album a')
def testMySql(self):
import mysql.connector as mdb
builder = QueryBuilder()
conn = mdb.connect(
host=self.config.get('mysql', 'host'),
port=self.config.get('mysql', 'port'),
user=self.config.get('mysql', 'user'),
passwd=self.config.get('mysql', 'passwd'))
c = conn.cursor()
c.execute('drop table if exists test.album')
c.execute('''create table if not exists test.album
(AlbumId int, Title varchar(100), ArtistId int)''')
c.close()
builder.add_table_with_fields('test.album', 'a', conn)
query = builder.select(['albumid', 'artistid'])
self.assertQuery(query, 'select a.albumid albumid, a.artistid artistid from test.album a')
