Tests: test output of action query

Author: rezib

tests/Controller.py

@@ -8,6 +8,7 @@
 from unittest.mock import patch, Mock
 import subprocess
 from io import StringIO
+import textwrap
 
 from TestUtils import (
     make_temp_dir, RiftTestCase, RiftProjectTestCase
@@ -44,22 +45,11 @@ def test_main_version(self):
         self.assert_except(SystemExit, "0", main, ['--version'])
 
 
-class ControllerProjectTest(RiftProjectTestCase):
+class ControllerProjectActionQueryTest(RiftProjectTestCase):
     """
-    Tests class for Controller
+    Tests class for Controller action query
     """
 
-    def _check_qemuuserstatic(self):
-        """Skip the test if none qemu-$arch-static executable is found for all
-        architectures declared in project configuration."""
-        if not any(
-            [
-                os.path.exists(f"/usr/bin/qemu-{arch}-static")
-                for arch in self.config.get('arch')
-            ]
-        ):
-            self.skipTest("qemu-user-static is not available")
-
     def test_action_query(self):
         """simple 'rift query' is ok """
         self.assertEqual(main(['query']), 0)
@@ -77,6 +67,60 @@ def test_action_query_on_bad_pkg(self):
         self.make_pkg(name='pkg2', metadata={})
         self.assertEqual(main(['query']), 0)
 
+    @patch('sys.stdout', new_callable=StringIO)
+    def test_action_query_output_default(self, mock_stdout):
+        self.make_pkg(name="pkg1")
+        self.make_pkg(name="pkg2", version='2.1', release='3')
+        self.assertEqual(main(['query']), 0)
+        self.assertIn(
+            "NAME MODULE       MAINTAINERS VERSION RELEASE MODULEMANAGER",
+            mock_stdout.getvalue())
+        self.assertIn(textwrap.dedent("""
+            ---- ------       ----------- ------- ------- -------------
+            pkg1 Great module Myself      1.0     1       buddy@somewhere.org
+            pkg2 Great module Myself      2.1     3       buddy@somewhere.org
+            """),
+            mock_stdout.getvalue())
+
+    @patch('sys.stdout', new_callable=StringIO)
+    def test_action_query_output_format(self, mock_stdout):
+        self.make_pkg(name="pkg1")
+        self.make_pkg(name="pkg2", version='2.1', release='3')
+        self.assertEqual(
+            main([
+                'query', '--format',
+                '%name %module %origin %reason %tests %version %arch %release '
+                '%changelogname %changelogtime %maintainers %modulemanager '
+                '%buildrequires']), 0)
+        self.assertIn(
+            "NAME MODULE       ORIGIN REASON          TESTS VERSION ARCH   "
+            "RELEASE CHANGELOGNAME                      CHANGELOGTIME "
+            "MAINTAINERS MODULEMANAGER       BUILDREQUIRES",
+            mock_stdout.getvalue())
+        self.assertIn(textwrap.dedent("""
+            ---- ------       ------ ------          ----- ------- ----   ------- -------------                      ------------- ----------- -------------       -------------
+            pkg1 Great module Vendor Missing feature 0     1.0     noarch 1       Myself <buddy@somewhere.org> 1.0-1 2019-02-26    Myself      buddy@somewhere.org br-package
+            pkg2 Great module Vendor Missing feature 0     2.1     noarch 3       Myself <buddy@somewhere.org> 2.1-3 2019-02-26    Myself      buddy@somewhere.org br-package
+            """),
+            mock_stdout.getvalue())
+
+
+class ControllerProjectTest(RiftProjectTestCase):
+    """
+    Tests class for Controller
+    """
+
+    def _check_qemuuserstatic(self):
+        """Skip the test if none qemu-$arch-static executable is found for all
+        architectures declared in project configuration."""
+        if not any(
+            [
+                os.path.exists(f"/usr/bin/qemu-{arch}-static")
+                for arch in self.config.get('arch')
+            ]
+        ):
+            self.skipTest("qemu-user-static is not available")
+
     @patch('rift.Controller.remove_packages')
     @patch('rift.Controller.validate_pkgs')
     @patch('rift.Controller.get_packages_from_patch')