From 97186f6a773ae9038d60a3cc7a9d1684462c05d6 Mon Sep 17 00:00:00 2001
From: Venkateshprasad <32921645+ven-k@users.noreply.github.com>
Date: Mon, 16 Oct 2023 16:24:57 +0530
Subject: [PATCH] test: conditional components and equations at the top level
 of `@mtkmodel`

---
 test/model_parsing.jl | 71 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 68 insertions(+), 3 deletions(-)

diff --git a/test/model_parsing.jl b/test/model_parsing.jl
index ea4d28803f..fd2e1f82ac 100644
--- a/test/model_parsing.jl
+++ b/test/model_parsing.jl
@@ -342,17 +342,82 @@ end
         end
     end
 
-    @mtkbuild a1 = A(eq_flag = 1)
+    @named a1 = A(eq_flag = 1)
+    a1 = complete(a1)
     @test getdefault(a1.c.val) == 1
     @test all([a1.eq ~ 0, a1.eq ~ 1] .∈ [equations(a1)])
+    a1 = complete(a1)
 
-    @mtkbuild a2 = A(c_flag = 2)
+    @named a2 = A(c_flag = 2)
+    a2 = complete(a2)
     @test getdefault(a2.c.val) == 2
     @test all([a2.eq ~ 0, a2.eq ~ 2] .∈ [equations(a2)])
 
     @test all(:comp0 .∈ [nameof.(a1.systems), nameof.(a2.systems)])
 
-    @mtkbuild a3 = A(eq_flag = false, c_flag = 3)
+    @named a3 = A(eq_flag = false, c_flag = 3)
+    a3 = complete(a3)
     @test getdefault(a3.c.val) == 3
     @test all([a3.eq ~ 0, a3.eq ~ 3] .∈ [equations(a3)])
+
+    @mtkmodel B begin
+        @structural_parameters begin
+            condition = 2
+        end
+        @parameters begin
+            a0
+            a1
+            a2
+            a3
+        end
+
+        @components begin
+            c0 = C()
+        end
+
+        @equations begin
+            a0 ~ 0
+        end
+
+        if condition == 1
+            @equations begin
+                a1 ~ 0
+            end
+            @components begin
+                c1 = C()
+            end
+        elseif condition == 2
+            @equations begin
+                a2 ~ 0
+            end
+            @components begin
+                c2 = C()
+            end
+        else
+            @equations begin
+                a3 ~ 0
+            end
+            @components begin
+                c3 = C()
+            end
+        end
+    end
+
+    @named b1 = B(condition = 1)
+    b1 = complete(b1)
+    @named b2 = B(condition = 2)
+    b2 = complete(b2)
+    @named b3 = B(condition = 10)
+    b3 = complete(b3)
+
+    @test nameof.(b1.systems) == [:c1, :c0]
+    @test nameof.(b2.systems) == [:c2, :c0]
+    @test nameof.(b3.systems) == [:c3, :c0]
+
+    @test Equation[b1.a1 ~ 0
+        b1.a0 ~ 0] == equations(b1)
+    @test Equation[b2.a2 ~ 0
+        b1.a0 ~ 0] == equations(b2)
+    @test Equation[b3.a3 ~ 0
+        b1.a0 ~ 0] == equations(b3)
 end