Extend unit tests

avisi-cloud · Mar 28, 2023 · 6107b19 · 6107b19
1 parent 74ed2e4
commit 6107b19
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt b/src/test/kotlin/nl/avisi/structurizr/site/generatr/site/model/MenuViewModelTest.kt
@@ -23,6 +23,7 @@ class MenuViewModelTest : ViewModelTest() {
 
         assertThat(viewModel.generalItems).containsExactly(
             LinkViewModel(pageViewModel, "Home", HomePageViewModel.url()),
+            LinkViewModel(pageViewModel, "Search", SearchViewModel.url()),
         )
     }
 
@@ -36,7 +37,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems[1]).isEqualTo(
+        assertThat(viewModel.generalItems[2]).isEqualTo(
             LinkViewModel(pageViewModel, "Software Systems", SoftwareSystemsPageViewModel.url())
         )
     }
@@ -56,7 +57,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems[1]).isEqualTo(
+        assertThat(viewModel.generalItems[2]).isEqualTo(
             LinkViewModel(pageViewModel, "Decisions", WorkspaceDecisionsPageViewModel.url(), false)
         )
     }
@@ -73,7 +74,7 @@ class MenuViewModelTest : ViewModelTest() {
         val pageViewModel = createPageViewModel(generatorContext)
         val viewModel = MenuViewModel(generatorContext, pageViewModel)
 
-        assertThat(viewModel.generalItems.drop(1)).containsExactly(
+        assertThat(viewModel.generalItems.drop(2)).containsExactly(
             LinkViewModel(pageViewModel, "Doc 1", WorkspaceDocumentationSectionPageViewModel.url(section1)),
             LinkViewModel(pageViewModel, "Doc Title 2", WorkspaceDocumentationSectionPageViewModel.url(section2))
         )