tests: mock uid when running as root (canonical#113)

Joshua Powers · OddBloke · commit 22054368db4c · 2019-12-16T10:08:25.000-05:00
The query command checks the user's uid when running and takes two
different code paths. As a normal user is returns fake data, that these
tests were expecting. As a root user, the actual user and vendor data
files are ready.

LP: #1856096
diff --git a/cloudinit/cmd/tests/test_query.py b/cloudinit/cmd/tests/test_query.py
@@ -150,7 +150,9 @@ def test_handle_args_dumps_all_instance_data(self):
             instance_data=self.instance_data, list_keys=False,
             user_data='ud', vendor_data='vd', varname=None)
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual(
             '{\n "my_var": "it worked",\n "userdata": "<%s> file:ud",\n'
             ' "vendordata": "<%s> file:vd"\n}\n' % (
@@ -165,7 +167,9 @@ def test_handle_args_returns_top_level_varname(self):
             instance_data=self.instance_data, list_keys=False,
             user_data='ud', vendor_data='vd', varname='my_var')
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual('it worked\n', m_stdout.getvalue())
 
     def test_handle_args_returns_nested_varname(self):
@@ -177,7 +181,9 @@ def test_handle_args_returns_nested_varname(self):
             instance_data=self.instance_data, user_data='ud', vendor_data='vd',
             list_keys=False, varname='v1.key_2')
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual('value-2\n', m_stdout.getvalue())
 
     def test_handle_args_returns_standardized_vars_to_top_level_aliases(self):
@@ -206,7 +212,9 @@ def test_handle_args_returns_standardized_vars_to_top_level_aliases(self):
             instance_data=self.instance_data, user_data='ud', vendor_data='vd',
             list_keys=False, varname=None)
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual(expected, m_stdout.getvalue())
 
     def test_handle_args_list_keys_sorts_top_level_keys_when_no_varname(self):
@@ -221,7 +229,9 @@ def test_handle_args_list_keys_sorts_top_level_keys_when_no_varname(self):
             instance_data=self.instance_data, list_keys=True, user_data='ud',
             vendor_data='vd', varname=None)
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual(expected, m_stdout.getvalue())
 
     def test_handle_args_list_keys_sorts_nested_keys_when_varname(self):
@@ -236,7 +246,9 @@ def test_handle_args_list_keys_sorts_nested_keys_when_varname(self):
             instance_data=self.instance_data, list_keys=True,
             user_data='ud', vendor_data='vd', varname='v1')
         with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-            self.assertEqual(0, query.handle_args('anyname', args))
+            with mock.patch('os.getuid') as m_getuid:
+                m_getuid.return_value = 100
+                self.assertEqual(0, query.handle_args('anyname', args))
         self.assertEqual(expected, m_stdout.getvalue())
 
     def test_handle_args_list_keys_errors_when_varname_is_not_a_dict(self):
@@ -252,7 +264,9 @@ def test_handle_args_list_keys_errors_when_varname_is_not_a_dict(self):
             vendor_data='vd',  varname='top')
         with mock.patch('sys.stderr', new_callable=StringIO) as m_stderr:
             with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
-                self.assertEqual(1, query.handle_args('anyname', args))
+                with mock.patch('os.getuid') as m_getuid:
+                    m_getuid.return_value = 100
+                    self.assertEqual(1, query.handle_args('anyname', args))
         self.assertEqual('', m_stdout.getvalue())
         self.assertIn(expected_error, m_stderr.getvalue())
 
