Improve CLI test

douglaslassance · Oct 14, 2024 · 73beaec · 73beaec
1 parent 84b03aa
commit 73beaec
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/cases.py b/tests/cases.py
@@ -189,6 +189,13 @@ def test_cli(self):
         output = f"+------- {untracked_image_01} - - - {host} {user}\n"
         self.assertEqual(output, result.output)
 
+        result = runner.invoke(cli.claim, [untracked_image_01], obj=obj)
+        self.assertEqual(0, result.exit_code, result.output)
+        host = socket.gethostname()
+        user = getpass.getuser()
+        output = f"-------- {untracked_image_01} - - - - -\n"
+        self.assertEqual(output, result.output)
+
     def tearDown(self):
         try:
             shutil.rmtree(self.temp_dir)