Plumbing logs from tarball into the correct logs parameter.

Signed-off-by: Rob Dobson <[email protected]>

Author: rdobson

hwinfo/tools/inspector.py

@@ -412,7 +412,10 @@ def main():
     validate_args(args)
 
     if args.logs:
-        host = HostFromLogs(args.logs)
+        if ".tar" in args.logs:
+            host = HostFromTarball(args.logs)
+        else:
+            host = HostFromLogs(args.logs)
     else:
         host = Host(args.machine, args.username, args.password)
 

hwinfo/tools/tests/test_inspector.py

@@ -402,6 +402,14 @@ def test_validate_local_machine(self):
         args.password = None
         inspector.validate_args(args)
 
+    @patch('hwinfo.tools.inspector.print_system_info')
+    @patch('hwinfo.tools.inspector.HostFromTarball')
+    def test_from_tarball(self, host_cls, print_system_info):
+        sys.argv = ['hwinfo', '-l', 'ack-submission.tar.gz']
+        mhost = host_cls.return_value = mock.MagicMock()
+        inspector.main()
+        host_cls.assert_called_with('ack-submission.tar.gz')
+
 class PrintSystemInfoTests(unittest.TestCase):
 
     @patch('hwinfo.tools.inspector.print_unit')