Added tests for webmodel

paramecio2/tests/conftest.py

@@ -0,0 +1,68 @@
+import pytest
+from paramecio2.libraries.db.webmodel import WebModel
+from paramecio2.libraries.db import corefields
+
+
+def pytest_addoption(parser):
+    parser.addoption("--mysql_host", action="store", default="host", help="MySQL host: default localhost")
+    parser.addoption("--mysql_user", action="store", default="root", help="Mysql User for make the test: default root")
+    parser.addoption("--mysql_password", action="store", default="", help="Mysql password for make the test: default empty value")
+    parser.addoption("--mysql_db", action="store", default="test_paramecio_db", help="Mysql Database for execute the test: default test_paramecio_db")
+    parser.addoption("--mysql_type", action="store", default="pymysql", help="MySQL Python library used: options pymysql mysqldb")  
+    
+@pytest.fixture
+def webmodel_conn(request):
+    options={'mysql_host': '', 'mysql_user': '', 'mysql_password': '', 'mysql_db': '', 'mysql_type': ''}
+    
+    for k in options:
+        options[k]=request.config.getoption(k)
+    
+    # Get connection 
+    #'db': options['mysql_db'],
+    
+    WebModel.connections={'default': {'name': 'default', 'host': options['mysql_host'], 'user': options['mysql_user'], 'password': options['mysql_password'], 'db': options['mysql_db'], 'charset': 'utf8mb4', 'set_connection': False, 'db_type': options['mysql_type']} }
+    
+    #conn=WebModel.connection()
+    
+    # Create database using raw mysql methods
+    
+    if options['mysql_type']=='pymysql':
+
+        import pymysql.cursors
+        pymysql.install_as_MySQLdb
+
+        conn=pymysql.connect(options['mysql_host'],
+                    user=options['mysql_user'],
+                    passwd=options['mysql_password'],
+                    charset='utf8mb4',
+                    cursorclass=pymysql.cursors.DictCursor)
+    else:
+        import MySQLdb.cursors
+        
+        conn=MySQLdb.connect(options['mysql_host'],
+            user=options['mysql_user'],
+            passwd=options['mysql_password'],
+            charset='utf8mb4',
+            cursorclass=MySQLdb.cursors.DictCursor)
+            
+    conn.query(
+            '''CREATE DATABASE {name}
+            DEFAULT CHARACTER SET {charset}'''
+            .format(
+                name=options['mysql_db'], charset='utf8mb4'
+            )
+        )
+    
+    #conn.close()
+    
+    def drop_database():
+        print('Finish him')
+        conn.query('DROP DATABASE IF EXISTS %s' % options['mysql_db'])
+        conn.close()
+        
+    request.addfinalizer(drop_database)
+    
+    final_conn=WebModel.connection()
+    
+    return final_conn
+