def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
class DashboardModel(object):
"""
The class controlling the database
"""
def __init__(self, verbose=0, dbpath=None):
"""
Class constructor
"""
self.db = SQLiteAPI(verbose=verbose, dbpath=dbpath)
def set_verbose(self, verbose=0):
"""
Verbose mode setter
"""
self.db.set_verbose(verbose=verbose)
def set_dbpath(self, dbpath="./sqlite.db"):
"""
Database path setter
"""
self.db.set_dbpath(dbpath=dbpath)
def exit(self):
"""
Close sqlite3 database
Initialize the db3 object
"""
self.db.exit()
def dump(self):
"""
Dump the database to store it
"""
return self.db.dump()
class TestDump(unittest.TestCase):
"""
Testsuite 1 to test map function
"""
def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
def tearDown(self):
"""
tearDown function launched after a testcase
"""
os.system("rm -fr " + self.dbpath)
def test_dump(self):
"""
Simple test to be sure write function alive
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
item = {}
item["status"] = "OK"
self.unit.write(table=table["name"], item=item)
self.unit.write(table=table["name"], item=item)
self.unit.write(table=table["name"], item=item)
self.unit.write(table=table["name"], item=item)
self.assertEqual(str, type(self.unit.dump()))
class TestCreateTable(unittest.TestCase):
"""
Testsuite 1 to test create table function
"""
def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
def tearDown(self):
"""
tearDown function launched after a testcase
"""
os.system("rm -fr " + self.dbpath)
def test_CreateDefaultTable_WrongType_Failure(self):
"""
Try to test passing a table not supporting
"""
table = None
self.assertEqual(1, self.unit.create_table(table))
def test_CreateDefaultTable_TableExists_Failure(self):
"""
Try to test passing a table name, but it already exists
"""
table = "MyTable"
self.unit.create_table(table)
self.assertEqual(1, self.unit.create_table(table))
def test_CreateDefaultTable_TableNotExists_Success(self):
"""
Try to test passing a table name, and it doesn't exist
"""
table = "MyTable"
self.assertEqual(0, self.unit.create_table(table))
def __init__(self, verbose=0, dbpath=None):
"""
Class constructor
"""
self.db = SQLiteAPI(verbose=verbose, dbpath=dbpath)
# pylint: disable=E1120
# line too long
# pylint: disable=C0301
# too many lines in module
# pylint: disable=C0302
import os
import sys
sys.path.append("src")
from sqlite_api import SQLiteAPI
if __name__ == '__main__':
os.system("rm -fr test/db.db3")
API = SQLiteAPI(verbose=1, dbpath="test/db.db3")
API.open()
PAGE = {}
PAGE["name"] = "Pagees"
PAGE["column"] = []
PAGE["column"].append({"name" : 'name', "type" : "String"})
PAGE["column"].append({"name" : 'comment', "type" : "str"})
PAGE["column"].append({"name" : 'status', "type" : "int"})
API.create_table(PAGE)
ITEM = {}
ITEM["name"] = "aParameter1"
ITEM["comment"] = "ThisIsAnUpdate"
ITEM["status"] = "OK"
API.write(table=PAGE["name"], item=ITEM)
#API.write(table=PAGE["name"], item=ITEM)
#ITEM = {}
class TestColumnName(unittest.TestCase):
"""
Testsuite 1 to ensure proper column name
"""
def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
def tearDown(self):
"""
tearDown function launched after a testcase
"""
os.system("rm -fr " + self.dbpath)
def test_DefaultTable_ColumnName(self):
"""
Create a default table check the column name
"""
table = "MyTable"
self.unit.create_table(table)
column = self.unit.read_column_info("MyTable")
ipp = 0
for col in column:
if ipp == 0:
self.assertEqual("id", col["name"].lower())
self.assertEqual("integer", col["type"].lower())
if ipp == 1:
self.assertEqual("item", col["name"].lower())
self.assertEqual("char(100)", col["type"].lower())
if ipp == 2:
self.assertEqual("value", col["name"].lower())
self.assertEqual("blob", col["type"].lower())
if ipp == 3:
self.assertEqual("datetime", col["name"].lower())
self.assertEqual("char(100)", col["type"].lower())
ipp = ipp + 1
def test_Table_ColumnName(self):
"""
Create a default table check the column name
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
table_column_read = self.unit.read_column_info(table["name"])
# Now rebuilt the table by adding the ID column naturaly
# added by the API
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'id', "type" : "integer"})
table["column"].append({"name" : 'name', "type" : "char(100)"})
table["column"].append({"name" : 'comment', "type" : "char(100)"})
table["column"].append({"name" : 'status', "type" : "integer"})
table["column"].append({"name" : 'datetime', "type" : "char(100)"})
ipp = 0
columns = zip(table["column"], table_column_read)
for col in columns:
if ipp == 0:
self.assertEqual(col[0]["name"], col[1]["name"].lower())
self.assertEqual(col[0]["type"], col[1]["type"].lower())
if ipp == 1:
self.assertEqual(col[0]["name"], col[1]["name"].lower())
self.assertEqual(col[0]["type"], col[1]["type"].lower())
if ipp == 2:
self.assertEqual(col[0]["name"], col[1]["name"].lower())
self.assertEqual(col[0]["type"], col[1]["type"].lower())
if ipp == 3:
self.assertEqual(col[0]["name"], col[1]["name"].lower())
self.assertEqual(col[0]["type"], col[1]["type"].lower())
if ipp == 4:
self.assertEqual(col[0]["name"], col[1]["name"].lower())
self.assertEqual(col[0]["type"], col[1]["type"].lower())
ipp = ipp + 1
class TestRead(unittest.TestCase):
"""
Testsuite 1 to test map function
"""
def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
def tearDown(self):
"""
tearDown function launched after a testcase
"""
os.system("rm -fr " + self.dbpath)
def test_read_single_filter_success(self):
"""
Simple test to be sure write function alive
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
item = {}
item["status"] = "OK"
self.unit.write(table=table["name"], item=item)
ret = self.unit.read(table=table["name"], filters=item)
self.assertEqual(list, type(ret))
def test_read_multi_filters_success(self):
"""
Simple test to be sure write function alive
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
item = {}
item["name"] = "aParameter1"
item["comment"] = "a status"
item["status"] = "OK"
self.unit.write(table=table["name"], item=item)
self.unit.write(table=table["name"], item=item)
self.unit.write(table=table["name"], item=item)
ret = self.unit.read(table=table["name"], filters=item)
self.assertEqual(list, type(ret))
for lines in ret:
self.assertEqual(list, type(lines))
class TestUpdate(unittest.TestCase):
"""
Testsuite 1 to test map function
"""
def setUp(self):
"""
Setup function launched before a testcase
"""
self.dbpath = 'test/db.db3'
self.unit = SQLiteAPI(verbose=1, dbpath=self.dbpath)
self.unit.open()
def tearDown(self):
"""
tearDown function launched after a testcase
"""
os.system("rm -fr " + self.dbpath)
def test_update_single_field_success(self):
"""
Simple test to be sure write function alive
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
item = {}
item["name"] = "aParameter1"
item["comment"] = "ThisIsAnUpdate"
item["status"] = "OK"
self.unit.write(table=table["name"], item=item)
item = {}
item["name"] = "aParameter1"
new_item = {}
new_item["status"] = "KO"
self.assertEqual(0, self.unit.update(table=table["name"], item=item, new_item=new_item))
def test_update_multi_field_success(self):
"""
Simple test to be sure write function alive
"""
# Build a table to store into the database
table = {}
table["name"] = "Pagees"
table["column"] = []
table["column"].append({"name" : 'name', "type" : "String"})
table["column"].append({"name" : 'comment', "type" : "str"})
table["column"].append({"name" : 'status', "type" : "int"})
self.unit.create_table(table)
item = {}
item["name"] = "aParameter1"
item["comment"] = "ThisIsAnUpdate"
item["status"] = "OK"
self.unit.write(table=table["name"], item=item)
item = {}
item["name"] = "aParameter1"
new_item = {}
new_item["comment"] = "ThisIsABadUpdate"
new_item["status"] = "KO"
self.assertEqual(0, self.unit.update(table=table["name"], item=item, new_item=new_item))