salt/tests/unit/modules/rbenv_test.py

137 lines
4.6 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 salttesting import skipIf, TestCase
from salttesting.mock import (
NO_MOCK,
NO_MOCK_REASON,
MagicMock,
patch)
from salttesting.helpers import ensure_in_syspath
ensure_in_syspath('../../')
# Import Salt Libs
from salt.modules import rbenv
import os
# Globals
rbenv.__grains__ = {}
rbenv.__salt__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
class RbenvTestCase(TestCase):
'''
Test cases for salt.modules.rbenv
'''
def test_install(self):
'''
Test for install Rbenv systemwide
'''
with patch.object(rbenv, '_rbenv_path', return_value=True):
with patch.object(rbenv, '_install_rbenv', return_value=True):
with patch.object(rbenv, '_install_ruby_build',
return_value=True):
with patch.object(os.path, 'expanduser', return_value='A'):
self.assertTrue(rbenv.install())
def test_update(self):
'''
Test for updates the current versions of Rbenv and Ruby-Build
'''
with patch.object(rbenv, '_rbenv_path', return_value=True):
with patch.object(rbenv, '_update_rbenv', return_value=True):
with patch.object(rbenv, '_update_ruby_build',
return_value=True):
with patch.object(os.path, 'expanduser', return_value='A'):
self.assertTrue(rbenv.update())
def test_is_installed(self):
'''
Test for check if Rbenv is installed.
'''
with patch.object(rbenv, '_rbenv_bin', return_value='A'):
with patch.dict(rbenv.__salt__,
{'cmd.has_exec': MagicMock(return_value=True)}):
self.assertTrue(rbenv.is_installed())
def test_install_ruby(self):
'''
Test for install a ruby implementation.
'''
with patch.dict(rbenv.__grains__, {'os': 'FreeBSD'}):
with patch.dict(rbenv.__salt__,
{'config.get': MagicMock(return_value='True')}):
with patch.object(rbenv, '_rbenv_exec',
return_value={'retcode': 0,
'stderr': 'stderr'}):
with patch.object(rbenv, 'rehash', return_value=None):
self.assertEqual(rbenv.install_ruby('ruby'), 'stderr')
with patch.object(rbenv, '_rbenv_exec',
return_value={'retcode': 1,
'stderr': 'stderr'}):
with patch.object(rbenv, 'uninstall_ruby',
return_value=None):
self.assertFalse(rbenv.install_ruby('ruby'))
def test_uninstall_ruby(self):
'''
Test for uninstall a ruby implementation.
'''
with patch.object(rbenv, '_rbenv_exec', return_value=None):
self.assertTrue(rbenv.uninstall_ruby('ruby', 'runas'))
def test_versions(self):
'''
Test for list the installed versions of ruby.
'''
with patch.object(rbenv, '_rbenv_exec', return_value='A\nBC\nD'):
self.assertListEqual(rbenv.versions(), ['A', 'BC', 'D'])
def test_default(self):
'''
Test for returns or sets the currently defined default ruby.
'''
with patch.object(rbenv, '_rbenv_exec',
MagicMock(side_effect=[None, False])):
self.assertTrue(rbenv.default('ruby', 'runas'))
self.assertEqual(rbenv.default(), '')
def test_list_(self):
'''
Test for list the installable versions of ruby.
'''
with patch.object(rbenv, '_rbenv_exec', return_value='A\nB\nCD\n'):
self.assertListEqual(rbenv.list_(), ['A', 'B', 'CD'])
def test_rehash(self):
'''
Test for run rbenv rehash to update the installed shims.
'''
with patch.object(rbenv, '_rbenv_exec', return_value=None):
self.assertTrue(rbenv.rehash())
def test_do_with_ruby(self):
'''
Test for execute a ruby command with rbenv's shims using a
specific ruby version.
'''
with patch.object(rbenv, 'do', return_value='A'):
self.assertEqual(rbenv.do_with_ruby('ruby', 'cmdline'), 'A')
if __name__ == '__main__':
from integration import run_tests
run_tests(RbenvTestCase, needs_daemon=False)