From a822f6995b6f4ae268f03fe670a8e6679379cc35 Mon Sep 17 00:00:00 2001
From: Liam Young <liam.young@canonical.com>
Date: Wed, 1 Apr 2020 13:45:47 +0000
Subject: [PATCH] Added unit tests

---
 .../utilities/test_zaza_utilities_generic.py       | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/unit_tests/utilities/test_zaza_utilities_generic.py b/unit_tests/utilities/test_zaza_utilities_generic.py
index 0f3ff9c..f1a648f 100644
--- a/unit_tests/utilities/test_zaza_utilities_generic.py
+++ b/unit_tests/utilities/test_zaza_utilities_generic.py
@@ -429,6 +429,20 @@ class TestGenericUtils(ut_utils.BaseTestCase):
         actual = generic_utils.get_unit_hostnames(_units)
 
         self.assertEqual(actual, expected)
+        expected_run_calls = [
+            mock.call('testunit/1', 'hostname'),
+            mock.call('testunit/2', 'hostname')]
+        self._run.assert_has_calls(expected_run_calls)
+
+        self._run.reset_mock()
+        self._run.side_effect = [{"Stdout": _hostname1},
+                                 {"Stdout": _hostname2}]
+        expected_run_calls = [
+            mock.call('testunit/1', 'hostname -f'),
+            mock.call('testunit/2', 'hostname -f')]
+
+        actual = generic_utils.get_unit_hostnames(_units, fqdn=True)
+        self._run.assert_has_calls(expected_run_calls)
 
     def test_port_knock_units(self):
         self.patch(