add more tests

charles-cooper · charles-cooper · commit 474a15f3cd5e · 2025-05-29T10:38:17.000+02:00
diff --git a/tests/unit/cli/vyper_json/test_output_selection.py b/tests/unit/cli/vyper_json/test_output_selection.py
@@ -102,9 +102,18 @@ def bar():
     self.foo()
     A.foo()
     assert 1 != 12
+    self.ctor_and_runtime()
 
 @internal
 def only_from_ctor():
+    self.ctor_recursive()
+
+@internal
+def ctor_recursive():
+    self.ctor_and_runtime()
+
+@internal
+def ctor_and_runtime():
     pass
 
 @deploy
@@ -122,19 +131,24 @@ def __init__():
 
     assert "foo (0)" in function_infos
     assert "foo (1)" in function_infos
-    assert "bar (2)" in function_infos
-    assert "only_from_ctor (4)" in function_infos
+    assert "ctor_and_runtime (2)" in function_infos
+    assert "bar (3)" in function_infos
+    assert "ctor_recursive (5)" in function_infos
+    assert "only_from_ctor (6)" in function_infos
     # faa is unreachable, should not be in metadata or bytecode
     assert not any("faa" in key for key in function_infos.keys())
 
     assert function_infos["foo (0)"]["function_id"] == 0
     assert function_infos["foo (1)"]["function_id"] == 1
-    assert function_infos["bar (2)"]["function_id"] == 2
+    assert function_infos["ctor_and_runtime (2)"]["function_id"] == 2
+    assert function_infos["bar (3)"]["function_id"] == 3
+    assert function_infos["ctor_recursive (5)"]["function_id"] == 5
+    assert function_infos["only_from_ctor (6)"]["function_id"] == 6
 
     assert function_infos["foo (0)"]["module_path"] == "B.vy"
     assert function_infos["foo (1)"]["module_path"] == "A.vy"
-    assert function_infos["bar (2)"]["module_path"] == "B.vy"
+    assert function_infos["bar (3)"]["module_path"] == "B.vy"
 
     assert function_infos["foo (0)"]["source_id"] == input_bundle.load_file("B.vy").source_id
     assert function_infos["foo (1)"]["source_id"] == input_bundle.load_file("A.vy").source_id
-    assert function_infos["bar (2)"]["source_id"] == input_bundle.load_file("B.vy").source_id
+    assert function_infos["bar (3)"]["source_id"] == input_bundle.load_file("B.vy").source_id
