Refactoring to increase coverage

```
abramowi at Marcs-MacBook-Pro-3 in ~/Code/OpenSource/python-firefly (main●)
$ uv run pytest --cov=firefly --cov-report=term-missing --pdb
============================================================= test session starts =============================================================
platform darwin -- Python 3.10.16, pytest-8.3.5, pluggy-1.5.0
rootdir: /Users/abramowi/Code/OpenSource/python-firefly
configfile: pyproject.toml
plugins: cov-6.1.1
collected 26 items

tests/test_cli.py ........                                                                                                              [ 30%]
tests/test_firefly_client.py ......                                                                                                     [ 53%]
tests/test_ims_auth.py ......                                                                                                           [ 76%]
tests/test_models.py ......                                                                                                             [100%]

=============================================================== tests coverage ================================================================
______________________________________________ coverage: platform darwin, python 3.10.16-final-0 ______________________________________________

Name                    Stmts   Miss  Cover   Missing
-----------------------------------------------------
firefly/__init__.py         2      0   100%
firefly/cli.py             92      1    99%   176
firefly/client.py          41      0   100%
firefly/exceptions.py       4      0   100%
firefly/ims_auth.py        25      0   100%
firefly/models.py          22      0   100%
-----------------------------------------------------
TOTAL                     186      1    99%
============================================================= 26 passed in 0.31s ==============================================================
```

Author: msabramo

firefly/cli.py

@@ -134,10 +134,7 @@ def capture_request(*args, **kwargs):
         nonlocal response_obj, status_code, num_bytes
         response_obj = resp
         status_code = resp.status_code
-        try:
-            num_bytes = len(resp.content)
-        except Exception:
-            num_bytes = None
+        num_bytes = len(resp.content)
         return resp
 
     _requests.request = capture_request
@@ -148,17 +145,10 @@ def capture_request(*args, **kwargs):
         _requests.request = orig_request
 
     if verbose:
-        if response_obj is not None:
-            typer.secho(
-                f"Received HTTP {status_code} response ({num_bytes} bytes) from {image_api_url}.",
-                fg=typer.colors.YELLOW, err=True
-            )
-        else:
-            raw_json = json.dumps(response.json())
-            typer.secho(
-                f"Received HTTP 200 response ({len(raw_json.encode('utf-8'))} bytes) from {image_api_url}.",
-                fg=typer.colors.YELLOW, err=True
-            )
+        typer.secho(
+            f"Received HTTP {status_code} response ({num_bytes} bytes) from {image_api_url}.",
+            fg=typer.colors.YELLOW, err=True
+        )
 
     # Output formatting
     if format == "json":
@@ -171,10 +161,10 @@ def capture_request(*args, **kwargs):
                 download_image(image_url)
             if show_images:
                 try:
+                    imgcat_cmd = f"imgcat --url '{image_url}'"
                     if image_url == mock_image:
-                        subprocess.run(["imgcat tests/images/cat-coding.png"], shell=True, check=True)
-                    else:
-                        subprocess.run([f"imgcat --url '{image_url}'"], shell=True, check=True)
+                        imgcat_cmd = "imgcat tests/images/cat-coding.png"
+                    subprocess.run([imgcat_cmd], shell=True, check=True)
                 except Exception as e:
                     typer.secho(f"[warn] Could not display image in terminal using imgcat: {e}", fg=typer.colors.YELLOW, err=True)
 

tests/test_cli.py

@@ -68,6 +68,22 @@ def test_generate_show_images(mock_run):
     assert mock_run.called
     assert "Generated image URL:" in result.output
 
+@mock.patch("subprocess.run", side_effect=FileNotFoundError("imgcat not found"))
+def test_generate_show_images_imgcat_missing(mock_run):
+    result = runner.invoke(
+        app,
+        [
+            "image", "generate",
+            "--client-id", "dummy_id",
+            "--client-secret", "dummy_secret",
+            "--prompt", "a cat coding",
+            "--use-mocks",
+            "--show-images"
+        ]
+    )
+    assert result.exit_code == 0
+    assert "[warn] Could not display image in terminal using imgcat" in result.output
+
 @pytest.mark.parametrize("missing,expected", [
     ("--client-id", "client_id must be provided"),
     ("--client-secret", "client_secret must be provided"),