salt/tests/unit/utils/test_sdb.py

# -*- coding: utf-8 -*-
'''
    :codeauthor: :email:`Vernon Cole <vernondcole@gmail.com>`
'''

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

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

# Import Salt Libs
import salt.utils.sdb as sdb


class SdbTestCase(TestCase, LoaderModuleMockMixin):
    '''
    Test cases for salt.modules.sdb
    '''

    @classmethod
    def setUpClass(cls):
        cls.TEMP_DATABASE_FILE = os.path.join(RUNTIME_VARS.TMP, 'test_sdb.sqlite')
        cls.sdb_opts = {
            'extension_modules': '',
            'optimization_order': [0, 1, 2],
            'test_sdb_data': {
                'driver': 'sqlite3',
                'database': cls.TEMP_DATABASE_FILE,
                'table': 'sdb',
                'create_table': True
                }
        }

    @classmethod
    def tearDownClass(cls):
        try:
            os.unlink(cls.TEMP_DATABASE_FILE)
        except OSError:
            pass

    def setup_loader_modules(self):
        return {sdb: {}}

    # test with SQLite database key not presest

    def test_sqlite_get_not_found(self):
        what = sdb.sdb_get(
                'sdb://test_sdb_data/thisKeyDoesNotExist', self.sdb_opts)
        self.assertEqual(what, None)

    # test with SQLite database write and read

    def test_sqlite_get_found(self):
        expected = {b'name': b'testone', b'number': 46}
        sdb.sdb_set('sdb://test_sdb_data/test1', expected, self.sdb_opts)
        resp = sdb.sdb_get('sdb://test_sdb_data/test1', self.sdb_opts)
        self.assertEqual(resp, expected)