From 024d3bb4be9987f626bf0c843828f95d1c660535 Mon Sep 17 00:00:00 2001
From: Dan Radez <dradez@redhat.com>
Date: Tue, 19 Sep 2017 15:57:01 -0400
Subject: Adding python unittests for apex/virtual/*

Change-Id: I13dd395cd6270cbf0a02855b1d29794ecca06d76
Signed-off-by: Dan Radez <dradez@redhat.com>
(cherry picked from commit 9d24ddc56a76175a4b7514cea5c789e2612b2528)
---
 apex/tests/test_apex_virtual_utils.py | 101 ++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)
 create mode 100644 apex/tests/test_apex_virtual_utils.py

(limited to 'apex/tests/test_apex_virtual_utils.py')

diff --git a/apex/tests/test_apex_virtual_utils.py b/apex/tests/test_apex_virtual_utils.py
new file mode 100644
index 00000000..643069f3
--- /dev/null
+++ b/apex/tests/test_apex_virtual_utils.py
@@ -0,0 +1,101 @@
+##############################################################################
+# Copyright (c) 2016 Dan Radez (dradez@redhat.com) (Red Hat)
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import subprocess
+import unittest
+
+from mock import patch
+
+from apex.virtual.utils import DEFAULT_VIRT_IP
+from apex.virtual.utils import get_virt_ip
+from apex.virtual.utils import generate_inventory
+from apex.virtual.utils import host_setup
+from apex.virtual.utils import virt_customize
+
+from nose.tools import (
+    assert_is_instance,
+    assert_regexp_matches,
+    assert_raises,
+    assert_equal)
+
+
+class TestVirtualUtils(unittest.TestCase):
+    @classmethod
+    def setup_class(cls):
+        """This method is run once for each class before any tests are run"""
+
+    @classmethod
+    def teardown_class(cls):
+        """This method is run once for each class _after_ all tests are run"""
+
+    def setup(self):
+        """This method is run once before _each_ test method is executed"""
+
+    def teardown(self):
+        """This method is run once after _each_ test method is executed"""
+
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_get_virt_ip(self, mock_subprocess):
+        mock_subprocess.return_value = '<xml></xml>'
+        assert_equal(get_virt_ip(), DEFAULT_VIRT_IP)
+
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_get_virt_ip_not_default(self, mock_subprocess):
+        mock_subprocess.return_value = '''<xml>
+<ip address='1.2.3.4' netmask='255.255.255.0'/>
+</xml>'''
+        assert_equal(get_virt_ip(), '1.2.3.4')
+
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_get_virt_ip_raises(self, mock_subprocess):
+        mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
+        assert_equal(get_virt_ip(), DEFAULT_VIRT_IP)
+
+    @patch('apex.virtual.utils.common_utils')
+    def test_generate_inventory(self, mock_common_utils):
+        assert_is_instance(generate_inventory('target_file'), dict)
+
+    @patch('apex.virtual.utils.common_utils')
+    def test_generate_inventory_ha_enabled(self, mock_common_utils):
+        assert_is_instance(generate_inventory('target_file', ha_enabled=True),
+                           dict)
+
+    @patch('apex.virtual.utils.iptc')
+    @patch('apex.virtual.utils.subprocess.check_call')
+    @patch('apex.virtual.utils.vbmc_lib')
+    def test_host_setup(self, mock_vbmc_lib, mock_subprocess, mock_iptc):
+        host_setup({'test': 2468})
+        mock_subprocess.assert_called_with(['vbmc', 'start', 'test'])
+
+    @patch('apex.virtual.utils.iptc')
+    @patch('apex.virtual.utils.subprocess.check_call')
+    @patch('apex.virtual.utils.vbmc_lib')
+    def test_host_setup_raise_called_process_error(self, mock_vbmc_lib,
+                                                   mock_subprocess, mock_iptc):
+        mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
+        assert_raises(subprocess.CalledProcessError, host_setup, {'tst': 2468})
+
+    @patch('apex.virtual.utils.os.path')
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_virt_customize(self, mock_subprocess, mock_os_path):
+        virt_customize([{'--operation': 'arg'}], 'target')
+
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_virt_customize_file_not_found(self, mock_subprocess):
+        assert_raises(FileNotFoundError,
+                      virt_customize,
+                      [{'--operation': 'arg'}], 'target')
+
+    @patch('apex.virtual.utils.os.path')
+    @patch('apex.virtual.utils.subprocess.check_output')
+    def test_virt_customize_raises(self, mock_subprocess, mock_os_path):
+        mock_subprocess.side_effect = subprocess.CalledProcessError(1, 'cmd')
+        assert_raises(subprocess.CalledProcessError,
+                      virt_customize,
+                      [{'--operation': 'arg'}], 'target')
-- 
cgit 1.2.3-korg