DOP-1575: Add mongo web shell as a directive (#244)

* DOP-1575: Add mongo web shell as a directive

* Put everything into rstspec.toml

Author: rayangler

snooty/rstspec.toml

@@ -10,6 +10,7 @@ card_type = ["small", "large"]
 output_format = ["html"]
 devhub_type = ["article", "quickstart", "how-to", "video", "live"]
 charts_theme = ["light", "dark"]
+mws_version = ["3.4", "3.6", "4.0", "4.2", "4.4", "latest"]
 
 [directive.default-domain]
 argument_type = "string"
@@ -617,6 +618,13 @@ inherit = "_guides-base"
 #    - second related page}
 # """
 
+[directive."mongodb:mongo-web-shell"]
+help = """An instance of the web version of a Mongo shell"""
+options.version = "mws_version"
+example = """.. mongo-web-shell::
+   ${1::version: (string)}
+"""
+
 [directive."mongodb:openapi"]
 help = """Include a reStructuredText file's contents."""
 argument_type = ["path", "uri"]

snooty/test_parser.py

@@ -1812,6 +1812,88 @@ def test_toctree() -> None:
     )
 
 
+def test_mongo_web_shell() -> None:
+    path = ROOT_PATH.joinpath(Path("test.rst"))
+    project_config = ProjectConfig(ROOT_PATH, "", source="./")
+    parser = rstparser.Parser(project_config, JSONVisitor)
+
+    # Test simple use of mongo-web-shell directive
+    page, diagnostics = parse_rst(
+        parser,
+        path,
+        """
+.. mongo-web-shell::
+""",
+    )
+    page.finish(diagnostics)
+    assert diagnostics == []
+    check_ast_testing_string(
+        page.ast,
+        """<root>
+        <directive domain="mongodb" name="mongo-web-shell"/>
+        </root>""",
+    )
+
+    # Test with option
+    page, diagnostics = parse_rst(
+        parser,
+        path,
+        """
+.. mongo-web-shell::
+   :version: 4.2
+""",
+    )
+    page.finish(diagnostics)
+    assert diagnostics == []
+    check_ast_testing_string(
+        page.ast,
+        """<root>
+        <directive domain="mongodb" name="mongo-web-shell" version="4.2"/>
+        </root>""",
+    )
+
+    # Test with invalid version
+    page, diagnostics = parse_rst(
+        parser,
+        path,
+        """
+.. mongo-web-shell::
+   :version: 4.1
+""",
+    )
+    page.finish(diagnostics)
+    assert len(diagnostics) == 1
+    assert isinstance(diagnostics[0], DocUtilsParseError)
+
+    # Test unknown options
+    page, diagnostics = parse_rst(
+        parser,
+        path,
+        """
+.. mongo-web-shell::
+   :width: 100%
+   :height: 300
+""",
+    )
+    page.finish(diagnostics)
+    assert len(diagnostics) == 1
+    assert isinstance(diagnostics[0], DocUtilsParseError)
+
+    # Test with unwanted content block
+    page, diagnostics = parse_rst(
+        parser,
+        path,
+        """
+.. mongo-web-shell::
+   
+   Hello world. This is unwanted content.
+""",
+    )
+    page.finish(diagnostics)
+    assert len(diagnostics) == 1
+    assert isinstance(diagnostics[0], DocUtilsParseError)
+
+
 def test_callable_target() -> None:
     path = ROOT_PATH.joinpath(Path("test.rst"))
     project_config = ProjectConfig(ROOT_PATH, "", source="./")