mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
139 lines
3.2 KiB
Python
139 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.sqlite3 as sqlite3
|
|
|
|
|
|
class MockSqlite3(object):
|
|
'''
|
|
Mock sqlite3 class
|
|
'''
|
|
version = '2.6.0'
|
|
sqlite_version = '3.8.2'
|
|
|
|
def __init__(self):
|
|
self.dbase = None
|
|
self.isolation_level = None
|
|
|
|
def connect(self, dbase, isolation_level=None):
|
|
'''
|
|
Mock connect method
|
|
'''
|
|
self.dbase = dbase
|
|
self.isolation_level = isolation_level
|
|
return MockSqlite3()
|
|
|
|
@staticmethod
|
|
def cursor():
|
|
'''
|
|
Mock connect method
|
|
'''
|
|
return MockSqlite3()
|
|
|
|
@staticmethod
|
|
def execute(sql):
|
|
'''
|
|
Mock connect method
|
|
'''
|
|
return sql
|
|
|
|
@staticmethod
|
|
def fetchall():
|
|
'''
|
|
Mock connect method
|
|
'''
|
|
return True
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class Sqlite3TestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
TestCase for salt.modules.sqlite3
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {sqlite3: {'sqlite3': MockSqlite3()}}
|
|
|
|
# 'version' function tests: 1
|
|
|
|
def test_version(self):
|
|
'''
|
|
Tests if it return version of pysqlite.
|
|
'''
|
|
self.assertEqual(sqlite3.version(), '2.6.0')
|
|
|
|
# 'sqlite_version' function tests: 1
|
|
|
|
def test_sqlite_version(self):
|
|
'''
|
|
Tests if it return version of sqlite.
|
|
'''
|
|
self.assertEqual(sqlite3.sqlite_version(), '3.8.2')
|
|
|
|
# 'modify' function tests: 1
|
|
|
|
def test_modify(self):
|
|
'''
|
|
Tests if it issue an SQL query to sqlite3 (with no return data).
|
|
'''
|
|
self.assertFalse(sqlite3.modify())
|
|
|
|
self.assertTrue(sqlite3.modify
|
|
('/root/test.db',
|
|
'CREATE TABLE test(id INT, testdata TEXT);'))
|
|
|
|
# 'fetch' function tests: 1
|
|
|
|
def test_fetch(self):
|
|
'''
|
|
Tests if it retrieve data from an sqlite3 db
|
|
(returns all rows, be careful!)
|
|
'''
|
|
self.assertFalse(sqlite3.fetch())
|
|
|
|
self.assertTrue(sqlite3.fetch
|
|
('/root/test.db',
|
|
'CREATE TABLE test(id INT, testdata TEXT);'))
|
|
|
|
# 'tables' function tests: 1
|
|
|
|
def test_tables(self):
|
|
'''
|
|
Tests if it show all tables in the database.
|
|
'''
|
|
self.assertFalse(sqlite3.tables())
|
|
|
|
self.assertTrue(sqlite3.tables('/root/test.db'))
|
|
|
|
# 'indices' function tests: 1
|
|
|
|
def test_indices(self):
|
|
'''
|
|
Tests if it show all indices in the database.
|
|
'''
|
|
self.assertFalse(sqlite3.indices())
|
|
|
|
self.assertTrue(sqlite3.indices('/root/test.db'))
|
|
|
|
# 'indexes' function tests: 1
|
|
|
|
def test_indexes(self):
|
|
'''
|
|
Tests if it show all indices in the database,
|
|
for people with poor spelling skills
|
|
'''
|
|
self.assertTrue(sqlite3.indexes('/root/test.db'))
|