salt/tests/unit/modules/test_sqlite3.py

# -*- coding: utf-8 -*-
"""
    :codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
"""

# Import Python Libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Libs
import salt.modules.sqlite3 as sqlite3

# Import Salt Testing Libs
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import TestCase


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


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"))