From 282481920b201e9779214b1b3e11074fdf94ab59 Mon Sep 17 00:00:00 2001
From: Bartek Kryza <bkryza@gmail.com>
Date: Thu, 8 Sep 2022 23:47:59 +0200
Subject: [PATCH] Extended nested classes and enums test case t00004

---
 tests/t00004/t00004.cc   | 16 ++++++++++++++--
 tests/t00004/test_case.h |  6 ++++++
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/tests/t00004/t00004.cc b/tests/t00004/t00004.cc
index d2f0a128..6c702235 100644
--- a/tests/t00004/t00004.cc
+++ b/tests/t00004/t00004.cc
@@ -14,12 +14,24 @@ public:
     public:
         enum class Lights { Green, Yellow, Red };
 
-        class AAA {
-        };
+        class AAA { };
     };
 
     void foo2() const { }
 };
 
+template <typename T> class C {
+public:
+    T t;
+
+    class AA {
+        class AAA { };
+
+        enum class CCC { CCC_1, CCC_2 };
+    };
+
+    enum class CC { CC_1, CC_2 };
+};
+
 }
 }
diff --git a/tests/t00004/test_case.h b/tests/t00004/test_case.h
index d0dd351f..d6bc5bc2 100644
--- a/tests/t00004/test_case.h
+++ b/tests/t00004/test_case.h
@@ -52,6 +52,12 @@ TEST_CASE("t00004", "[test-case][class]")
     REQUIRE_THAT(puml, (IsMethod<Public, Const>("foo")));
     REQUIRE_THAT(puml, (IsMethod<Public, Const>("foo2")));
 
+    REQUIRE_THAT(puml, IsClassTemplate("C", "T"));
+    REQUIRE_THAT(puml, IsInnerClass(_A("C<T>"), _A("C<T>::AA")));
+    REQUIRE_THAT(puml, IsInnerClass(_A("C<T>::AA"), _A("C<T>::AA::AAA")));
+    REQUIRE_THAT(puml, IsInnerClass(_A("C<T>"), _A("C<T>::CC")));
+    REQUIRE_THAT(puml, IsInnerClass(_A("C<T>::AA"), _A("C<T>::AA::CCC")));
+
     save_puml(
         "./" + config.output_directory() + "/" + diagram->name + ".puml", puml);
 }