feat: add more tests on all classes using DiagramIndexViewModel

avisi-cloud · Nov 9, 2024 · 569f689 · 569f689
1 parent 65c4f81
commit 569f689
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 2 deletions.
diff --git a/src/main/kotlin/nl/avisi/structurizr/site/generatr/StructurizrUtilities.kt b/src/main/kotlin/nl/avisi/structurizr/site/generatr/StructurizrUtilities.kt
@@ -6,8 +6,6 @@ import com.structurizr.model.SoftwareSystem
 import com.structurizr.model.StaticStructureElement
 import com.structurizr.view.ViewSet
 import nl.avisi.structurizr.site.generatr.site.GeneratorContext
-import nl.avisi.structurizr.site.generatr.site.model.DiagramViewModel
-import nl.avisi.structurizr.site.generatr.site.model.ImageViewViewModel
 
 val Workspace.includedSoftwareSystems: List<SoftwareSystem>
     get() = model.softwareSystems.filter {

diff --git a/...cturizr/site/generatr/site/model/SoftwareSystemContainerComponentCodePageViewModelTest.kt b/...cturizr/site/generatr/site/model/SoftwareSystemContainerComponentCodePageViewModelTest.kt
@@ -107,4 +107,12 @@ class SoftwareSystemContainerComponentCodePageViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.visible).isFalse()
     }
+
+    @Test
+    fun `show list is disabled`() {
+        val viewModel = SoftwareSystemContainerComponentCodePageViewModel(generatorContext, backendContainer, backendComponent)
+        assertThat(viewModel.diagramIndex.showList).isFalse()
+
+
+    }
 }
diff --git a/...tructurizr/site/generatr/site/model/SoftwareSystemContainerComponentsPageViewModelTest.kt b/...tructurizr/site/generatr/site/model/SoftwareSystemContainerComponentsPageViewModelTest.kt
@@ -148,4 +148,10 @@ class SoftwareSystemContainerComponentsPageViewModelTest : ViewModelTest() {
         val viewModel = SoftwareSystemContainerComponentsPageViewModel(generatorContext, softwareSystem.addContainer("Container"))
         assertThat(viewModel.visible).isFalse()
     }
+
+    @Test
+    fun `show list is enabled`() {
+        val viewModel = SoftwareSystemContainerComponentsPageViewModel(generatorContext, backendContainer)
+        assertThat(viewModel.diagramIndex.showList).isTrue()
+    }
 }
diff --git a/...nl/avisi/structurizr/site/generatr/site/model/SoftwareSystemContainerPageViewModelTest.kt b/...nl/avisi/structurizr/site/generatr/site/model/SoftwareSystemContainerPageViewModelTest.kt
@@ -75,4 +75,10 @@ class SoftwareSystemContainerPageViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.visible).isFalse()
     }
+
+    @Test
+    fun `show list is enabled`() {
+        val viewModel = SoftwareSystemContainerPageViewModel(generatorContext, softwareSystem)
+        assertThat(viewModel.diagramIndex.showList).isTrue()
+    }
 }
diff --git a/...l/avisi/structurizr/site/generatr/site/model/SoftwareSystemDeploymentPageViewModelTest.kt b/...l/avisi/structurizr/site/generatr/site/model/SoftwareSystemDeploymentPageViewModelTest.kt
@@ -4,6 +4,7 @@ import assertk.assertThat
 import assertk.assertions.containsExactly
 import assertk.assertions.isEqualTo
 import assertk.assertions.isFalse
+import assertk.assertions.isTrue
 import com.structurizr.model.SoftwareSystem
 import kotlin.test.Test
 
@@ -62,4 +63,10 @@ class SoftwareSystemDeploymentPageViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.visible).isFalse()
     }
+
+    @Test
+    fun `show list is enabled`() {
+        val viewModel = SoftwareSystemDeploymentPageViewModel(generatorContext, softwareSystem)
+        assertThat(viewModel.diagramIndex.showList).isTrue()
+    }
 }
diff --git a/...n/nl/avisi/structurizr/site/generatr/site/model/SoftwareSystemDynamicPageViewModelTest.kt b/...n/nl/avisi/structurizr/site/generatr/site/model/SoftwareSystemDynamicPageViewModelTest.kt
@@ -4,6 +4,7 @@ import assertk.assertThat
 import assertk.assertions.containsExactly
 import assertk.assertions.isEqualTo
 import assertk.assertions.isFalse
+import assertk.assertions.isTrue
 import com.structurizr.model.SoftwareSystem
 import kotlin.test.Test
 
@@ -64,4 +65,10 @@ class SoftwareSystemDynamicPageViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.visible).isFalse()
     }
+
+    @Test
+    fun `show list is enabled`() {
+        val viewModel = SoftwareSystemDynamicPageViewModel(generatorContext, softwareSystem)
+        assertThat(viewModel.diagramIndex.showList).isTrue()
+    }
 }