Fixed class visitor continuation after inner class

2021-02-26 23:22:16 +01:00
parent 474f38c349
commit c1f4ecefbc
5 changed files with 13 additions and 4 deletions
--- a/tests/test_cases.cc
+++ b/tests/test_cases.cc
@@ -258,6 +258,8 @@ TEST_CASE("Test t00004", "[unit-test]")
    REQUIRE_THAT(puml, Contains("A +-- AA"));
    REQUIRE_THAT(puml, Contains("AA +-- AAA"));
    REQUIRE_THAT(puml, Contains("AA +-- Lights"));
+    REQUIRE_THAT(puml, Contains("+ foo() const"));
+    REQUIRE_THAT(puml, Contains("+ foo2() const"));

    save_puml(
        "./" + config.output_directory + "/" + diagram->name + ".puml", puml);