Update tests

Author: DavidSouther

aws_doc_sdk_examples_tools/doc_gen_cli.py

@@ -66,18 +66,19 @@ def main():
     with open(args.write_json, "w") as out:
         out.write(serialized)
 
-    for root in args.from_root:
-        merged_doc_gen.collect_snippets(Path(root))
-
-    serialized_snippets = json.dumps(
-        {
-            "snippets": merged_doc_gen.snippets,
-            "snippet_files": merged_doc_gen.snippet_files,
-        },
-        cls=DocGenEncoder,
-    )
-    with open(args.write_snippets, "w") as out:
-        out.write(serialized_snippets)
+    if args.write_snippets:
+        for root in args.from_root:
+            merged_doc_gen.collect_snippets(Path(root))
+
+        serialized_snippets = json.dumps(
+            {
+                "snippets": merged_doc_gen.snippets,
+                "snippet_files": merged_doc_gen.snippet_files,
+            },
+            cls=DocGenEncoder,
+        )
+        with open(args.write_snippets, "w") as out:
+            out.write(serialized_snippets)
 
 
 if __name__ == "__main__":

aws_doc_sdk_examples_tools/doc_gen_cli_test.py

@@ -76,6 +76,7 @@ def test_doc_gen_strict_option(strict, should_raise, patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=strict,
         skip_entity_expansion=False,
     )
@@ -94,6 +95,7 @@ def test_skip_entity_expansion(patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=False,
         skip_entity_expansion=True,
     )
@@ -111,6 +113,7 @@ def test_default_entity_expansion(patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=False,
         skip_entity_expansion=False,
     )