Merge "Adding python unittests for apex/virtual/*"
[apex.git] / apex / tests / test_apex_virtual_utils.py
1 ##############################################################################
2 # Copyright (c) 2016 Dan Radez (dradez@redhat.com) (Red Hat)
3 #
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
9
10 import subprocess
11 import unittest
12
13 from mock import patch
14
15 from apex.virtual.utils import DEFAULT_VIRT_IP
16 from apex.virtual.utils import get_virt_ip
17 from apex.virtual.utils import generate_inventory
18 from apex.virtual.utils import host_setup
19 from apex.virtual.utils import virt_customize
20
21 from nose.tools import (
22     assert_is_instance,
23     assert_regexp_matches,
24     assert_raises,
25     assert_equal)
26
27
28 class TestVirtualUtils(unittest.TestCase):
29     @classmethod
30     def setup_class(cls):
31         """This method is run once for each class before any tests are run"""
32
33     @classmethod
34     def teardown_class(cls):
35         """This method is run once for each class _after_ all tests are run"""
36
37     def setup(self):
38         """This method is run once before _each_ test method is executed"""
39
40     def teardown(self):
41         """This method is run once after _each_ test method is executed"""
42
43     @patch('apex.virtual.utils.subprocess.check_output')
44     def test_get_virt_ip(self, mock_subprocess):
45         mock_subprocess.return_value = '<xml></xml>'
46         assert_equal(get_virt_ip(), DEFAULT_VIRT_IP)
47
48     @patch('apex.virtual.utils.subprocess.check_output')
49     def test_get_virt_ip_not_default(self, mock_subprocess):
50         mock_subprocess.return_value = '''<xml>
51 <ip address='1.2.3.4' netmask='255.255.255.0'/>
52 </xml>'''
53         assert_equal(get_virt_ip(), '1.2.3.4')
54
55     @patch('apex.virtual.utils.subprocess.check_output')
56     def test_get_virt_ip_raises(self, mock_subprocess):
57         mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
58         assert_equal(get_virt_ip(), DEFAULT_VIRT_IP)
59
60     @patch('apex.virtual.utils.common_utils')
61     def test_generate_inventory(self, mock_common_utils):
62         assert_is_instance(generate_inventory('target_file'), dict)
63
64     @patch('apex.virtual.utils.common_utils')
65     def test_generate_inventory_ha_enabled(self, mock_common_utils):
66         assert_is_instance(generate_inventory('target_file', ha_enabled=True),
67                            dict)
68
69     @patch('apex.virtual.utils.iptc')
70     @patch('apex.virtual.utils.subprocess.check_call')
71     @patch('apex.virtual.utils.vbmc_lib')
72     def test_host_setup(self, mock_vbmc_lib, mock_subprocess, mock_iptc):
73         host_setup({'test': 2468})
74         mock_subprocess.assert_called_with(['vbmc', 'start', 'test'])
75
76     @patch('apex.virtual.utils.iptc')
77     @patch('apex.virtual.utils.subprocess.check_call')
78     @patch('apex.virtual.utils.vbmc_lib')
79     def test_host_setup_raise_called_process_error(self, mock_vbmc_lib,
80                                                    mock_subprocess, mock_iptc):
81         mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
82         assert_raises(subprocess.CalledProcessError, host_setup, {'tst': 2468})
83
84     @patch('apex.virtual.utils.os.path')
85     @patch('apex.virtual.utils.subprocess.check_output')
86     def test_virt_customize(self, mock_subprocess, mock_os_path):
87         virt_customize([{'--operation': 'arg'}], 'target')
88
89     @patch('apex.virtual.utils.subprocess.check_output')
90     def test_virt_customize_file_not_found(self, mock_subprocess):
91         assert_raises(FileNotFoundError,
92                       virt_customize,
93                       [{'--operation': 'arg'}], 'target')
94
95     @patch('apex.virtual.utils.os.path')
96     @patch('apex.virtual.utils.subprocess.check_output')
97     def test_virt_customize_raises(self, mock_subprocess, mock_os_path):
98         mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
99         assert_raises(subprocess.CalledProcessError,
100                       virt_customize,
101                       [{'--operation': 'arg'}], 'target')