mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Rahul Handay <rahulha@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
import os
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.oracle as oracle
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class OracleTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.oracle
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {oracle: {'cx_Oracle': object()}}
|
|
|
|
def test_run_query(self):
|
|
'''
|
|
Test for Run SQL query and return result
|
|
'''
|
|
with patch.object(oracle, '_connect', MagicMock()) as mock_connect:
|
|
mock_connect.cursor.execute.fetchall.return_value = True
|
|
with patch.object(oracle, 'show_dbs', MagicMock()):
|
|
self.assertTrue(oracle.run_query('db', 'query'))
|
|
|
|
def test_show_dbs(self):
|
|
'''
|
|
Test for Show databases configuration from pillar. Filter by `*args`
|
|
'''
|
|
with patch.dict(oracle.__salt__, {'pillar.get':
|
|
MagicMock(return_value='a')}):
|
|
self.assertDictEqual(oracle.show_dbs('A', 'B'),
|
|
{'A': 'a', 'B': 'a'})
|
|
|
|
self.assertEqual(oracle.show_dbs(), 'a')
|
|
|
|
def test_version(self):
|
|
'''
|
|
Test for Server Version (select banner from v$version)
|
|
'''
|
|
with patch.dict(oracle.__salt__, {'pillar.get':
|
|
MagicMock(return_value='a')}):
|
|
with patch.object(oracle, 'run_query', return_value='A'):
|
|
self.assertDictEqual(oracle.version(), {})
|
|
|
|
def test_client_version(self):
|
|
'''
|
|
Test for Oracle Client Version
|
|
'''
|
|
with patch.object(oracle, 'cx_Oracle',
|
|
MagicMock(side_effect=MagicMock())):
|
|
self.assertEqual(oracle.client_version(), '')
|
|
|
|
def test_show_pillar(self):
|
|
'''
|
|
Test for Show Pillar segment oracle.*
|
|
'''
|
|
with patch.dict(oracle.__salt__, {'pillar.get':
|
|
MagicMock(return_value='a')}):
|
|
self.assertEqual(oracle.show_pillar('item'), 'a')
|
|
|
|
def test_show_env(self):
|
|
'''
|
|
Test for Show Environment used by Oracle Client
|
|
'''
|
|
with patch.object(os, 'environ',
|
|
return_value={'PATH': 'PATH',
|
|
'ORACLE_HOME': 'ORACLE_HOME',
|
|
'TNS_ADMIN': 'TNS_ADMIN',
|
|
'NLS_LANG': 'NLS_LANG'}):
|
|
self.assertDictEqual(oracle.show_env(), {})
|