Tests: Added new commands to memray attach

Signed-off-by: Ivona Stojanovic <[email protected]>

Author: ivonastojanovic

tests/unit/test_attach.py

@@ -18,4 +18,68 @@ def test_memray_attach_aggregated_without_output_file(
             main(["attach", "--aggregate", "1234"])
 
         captured = capsys.readouterr()
+        print("Error", captured.err)
         assert "Can't use aggregated mode without an output file." in captured.err
+
+
+class TestAttachSubCommandOptions:
+    @pytest.mark.parametrize(
+        "option",
+        [
+            "--native",
+            "--force",
+            "-f",
+            "--aggregate",
+            "--follow-fork",
+            "--trace-python-allocators",
+            "--no-compress",
+        ],
+    )
+    def test_memray_attach_stop_tracking_option_with_other_options(
+        self, option, capsys
+    ):
+        # WHEN
+        with pytest.raises(SystemExit):
+            main(["attach", "1234", "--stop-tracking", option])
+
+        captured = capsys.readouterr()
+        assert (
+            "Can't use --stop-tracking with" in captured.err and option in captured.err
+        )
+
+    @pytest.mark.parametrize(
+        "arg1,arg2",
+        [
+            ("--stop-tracking", "--heap-limit=10"),
+            ("--stop-tracking", "--duration=10"),
+            ("--heap-limit=10", "--duration=10"),
+        ],
+    )
+    def test_memray_attach_stop_tracking_option_with_other_mode_options(
+        self, arg1, arg2, capsys
+    ):
+        # WHEN
+        with pytest.raises(SystemExit):
+            main(["attach", "1234", arg1, arg2])
+
+        captured = capsys.readouterr()
+        arg1_name = arg1.split("=")[0]
+        arg2_name = arg2.split("=")[0]
+        assert (
+            f"argument {arg2_name}: not allowed with argument {arg1_name}"
+            in captured.err
+        )
+
+    @pytest.mark.parametrize(
+        "option, argument",
+        [("--output", "test.bin")],
+    )
+    def test_memray_attach_stop_tracking_option_with_output_options(
+        self, option, argument, capsys
+    ):
+        # WHEN
+        with pytest.raises(SystemExit):
+            main(["attach", "1234", "--stop-tracking", option, argument])
+
+        captured = capsys.readouterr()
+        assert f"Can't use --stop-tracking with {option}" in captured.err