BIT-2131: Display folders when filtering by org in vault (#1170)

app/src/main/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/VaultItemListingViewModel.kt

@@ -672,9 +672,7 @@ class VaultItemListingViewModel @Inject constructor(
                 itemListingType = currentState
                     .itemListingType
                     .updateWithAdditionalDataIfNecessary(
-                        folderList = vaultData
+                        folderList = vaultData.folderViewList,
-                            .folderViewList
-                            .toFilteredList(state.vaultFilterType),
                         collectionList = vaultData
                             .collectionViewList
                             .toFilteredList(state.vaultFilterType),

app/src/main/java/com/x8bit/bitwarden/ui/vault/feature/itemlisting/util/VaultItemListingDataExtensions.kt

@@ -209,7 +209,9 @@ fun List<SendView>.toViewState(
         )
     }
 
-/** * Updates a [VaultItemListingState.ItemListingType] with the given data if necessary. */
+/**
+ * Updates a [VaultItemListingState.ItemListingType] with the given data if necessary.
+ */
 fun VaultItemListingState.ItemListingType.updateWithAdditionalDataIfNecessary(
     folderList: List<FolderView>,
     collectionList: List<CollectionView>,

app/src/main/java/com/x8bit/bitwarden/ui/vault/feature/vault/util/VaultDataExtensions.kt

@@ -46,7 +46,12 @@ fun VaultData.toViewState(
     val filteredCipherViewList = filteredCipherViewListWithDeletedItems
         .filter { it.deletedDate == null }
 
-    val filteredFolderViewList = folderViewList.toFilteredList(vaultFilterType).getFilteredFolders()
+    val filteredFolderViewList = folderViewList
+        .toFilteredList(
+            cipherList = filteredCipherViewList,
+            vaultFilterType = vaultFilterType,
+        )
+        .getFilteredFolders()
 
     val filteredCollectionViewList = collectionViewList
         .toFilteredList(vaultFilterType)
@@ -257,17 +262,20 @@ fun List<CipherView>.toFilteredList(
  */
 @JvmName("toFilteredFolderList")
 fun List<FolderView>.toFilteredList(
+    cipherList: List<CipherView>,
     vaultFilterType: VaultFilterType,
 ): List<FolderView> =
     this
-        .filter {
+        .filter { folder ->
             when (vaultFilterType) {
-                // Folders are only included when including the user's personal data.
                 VaultFilterType.AllVaults,
                 VaultFilterType.MyVault,
                 -> true
 
-                is VaultFilterType.OrganizationVault -> false
+                // Only include folders containing an item associated with this organization.
+                is VaultFilterType.OrganizationVault -> {
+                    cipherList.any { it.folderId == folder.id }
+                }
             }
         }
 

app/src/test/java/com/x8bit/bitwarden/ui/vault/feature/vault/util/VaultDataExtensionsTest.kt

@@ -152,7 +152,10 @@ class VaultDataExtensionsTest {
                 createMockCollectionView(number = 1),
                 createMockCollectionView(number = 2),
             ),
-            folderViewList = listOf(createMockFolderView(number = 1)),
+            folderViewList = listOf(
+                createMockFolderView(number = 1),
+                createMockFolderView(number = 3),
+            ),
             sendViewList = listOf(createMockSendView(number = 1)),
         )
 
@@ -173,7 +176,13 @@ class VaultDataExtensionsTest {
                 identityItemsCount = 0,
                 secureNoteItemsCount = 0,
                 favoriteItems = listOf(),
-                folderItems = listOf(),
+                folderItems = listOf(
+                    VaultState.ViewState.FolderItem(
+                        id = "mockId-1",
+                        name = "mockName-1".asText(),
+                        itemCount = 1,
+                    ),
+                ),
                 collectionItems = listOf(
                     VaultState.ViewState.CollectionItem(
                         id = "mockId-1",