react-native-tree-multi-select 1.2.0 → 1.2.3

package/lib/typescript/__tests__/initNodeMap.helper.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"initNodeMap.helper.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/initNodeMap.helper.test.ts"],"names":[],"mappings":""}

package/lib/typescript/__tests__/search.helper.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=search.helper.test.d.ts.map

package/lib/typescript/__tests__/search.helper.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.helper.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/search.helper.test.ts"],"names":[],"mappings":""}

package/lib/typescript/__tests__/selectAll.helper.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=selectAll.helper.test.d.ts.map

package/lib/typescript/__tests__/selectAll.helper.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"selectAll.helper.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/selectAll.helper.test.ts"],"names":[],"mappings":""}

package/lib/typescript/__tests__/store.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=store.test.d.ts.map

package/lib/typescript/__tests__/store.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/store.test.ts"],"names":[],"mappings":""}

package/lib/typescript/__tests__/toggleCheckbox.helper.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=toggleCheckbox.helper.test.d.ts.map

package/lib/typescript/__tests__/toggleCheckbox.helper.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toggleCheckbox.helper.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/toggleCheckbox.helper.test.ts"],"names":[],"mappings":""}

package/lib/typescript/helpers/expandCollapse.helper.d.ts

@@ -15,4 +15,16 @@ export declare function expandAll(): void;
  * Collapse all nodes in the tree.
  */
 export declare function collapseAll(): void;
+/**
+ * Expand tree nodes of given ids. If the id is of a child, it also expands
+ * the parent which it belongs to.
+ * @param ids Ids of nodes to expand.
+ */
+export declare function expandNodes(ids: string[]): void;
+/**
+ * Collapse tree nodes of given ids. If the id is of a parent, it also collapses
+ * the children inside it.
+ * @param ids Ids of nodes to collapse.
+ */
+export declare function collapseNodes(ids: string[]): void;
 //# sourceMappingURL=expandCollapse.helper.d.ts.map

package/lib/typescript/helpers/expandCollapse.helper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"expandCollapse.helper.d.ts","sourceRoot":"","sources":["../../../src/helpers/expandCollapse.helper.ts"],"names":[],"mappings":"AAGA;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,MAAM,QAmE5C;AAED;;GAEG;AACH,wBAAgB,SAAS,SAKxB;AAED;;GAEG;AACH,wBAAgB,WAAW,SAK1B"}
+{"version":3,"file":"expandCollapse.helper.d.ts","sourceRoot":"","sources":["../../../src/helpers/expandCollapse.helper.ts"],"names":[],"mappings":"AAGA;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,MAAM,QA+C5C;AAED;;GAEG;AACH,wBAAgB,SAAS,SAKxB;AAED;;GAEG;AACH,wBAAgB,WAAW,SAK1B;AAED;;;;GAIG;AACH,wBAAgB,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,QAwBxC;AAED;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,QAuB1C"}

package/lib/typescript/types/treeView.types.d.ts

@@ -44,6 +44,7 @@ export interface TreeViewProps extends NodeListProps {
     onCheck?: (checkedIds: string[]) => void;
     onExpand?: (expandedIds: string[]) => void;
     preselectedIds?: string[];
+    preExpandedIds?: string[];
 }
 type CheckboxProps = Omit<RNPaperCheckboxAndroidProps, "onPress" | "status">;
 export interface CheckBoxViewProps {
@@ -66,6 +67,8 @@ export interface TreeViewRef {
     unselectAllFiltered: () => void;
     expandAll: () => void;
     collapseAll: () => void;
+    expandNodes: (ids: string[]) => void;
+    collapseNodes: (ids: string[]) => void;
     setSearchText: (searchText: string, searchKeys?: string[]) => void;
 }
 export {};

package/lib/typescript/types/treeView.types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"treeView.types.d.ts","sourceRoot":"","sources":["../../../src/types/treeView.types.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACR,SAAS,EACT,SAAS,EACT,qBAAqB,EACrB,SAAS,EACZ,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EACH,KAAK,KAAK,IAAI,2BAA2B,EAC5C,MAAM,4DAA4D,CAAC;AAEpE,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,eAAe,CAAC;AAE1D,MAAM,WAAW,eAAe;IAC5B,UAAU,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,QAAQ;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC;IACtB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACtB;AAED,MAAM,WAAW,qBAAsB,SAAQ,QAAQ;IACnD,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,MAAM,iBAAiB,CAAC,KAAK,GAAG,GAAG,IAAI,IAAI,CAC7C,cAAc,CAAC,KAAK,CAAC,EACrB,MAAM,GACJ,YAAY,CACjB,CAAC;AAEF,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IAEd,YAAY,EAAE,iBAAiB,CAAC;IAChC,UAAU,EAAE,OAAO,CAAC;IAEpB,OAAO,EAAE,MAAM,IAAI,CAAC;IACpB,QAAQ,EAAE,MAAM,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,sBAAsB;IACnC,sBAAsB,CAAC,EAAE,6BAA6B,CAAC;IAEvD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B,iBAAiB,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAC3D,2BAA2B,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;IACnE,gCAAgC,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAE9E,sBAAsB,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;CAC9D;AAED,MAAM,WAAW,SAAU,SAAQ,sBAAsB;IACrD,IAAI,EAAE,qBAAqB,CAAC;IAC5B,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,aAAc,SAAQ,sBAAsB;IACzD,kBAAkB,CAAC,EAAE,iBAAiB,CAAC;CAC1C;AAED,MAAM,WAAW,aAAc,SAAQ,aAAa;IAChD,IAAI,EAAE,QAAQ,EAAE,CAAC;IAEjB,OAAO,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IACzC,QAAQ,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAE3C,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC7B;AAED,KAAK,aAAa,GAAG,IAAI,CAAC,2BAA2B,EAAE,SAAS,GAAG,QAAQ,CAAC,CAAC;AAE7E,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,iBAAiB,CAAC;IACzB,aAAa,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IACxC,IAAI,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,6BAA6B;IAE1C,wBAAwB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAChD,uBAAuB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAC/C,kBAAkB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAG1C,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACzB;AAED,MAAM,MAAM,wBAAwB,GAChC,iBAAiB,GACf,6BAA6B,CAAC;AAEpC,MAAM,WAAW,WAAW;IACxB,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,EAAE,MAAM,IAAI,CAAC;IAExB,iBAAiB,EAAE,MAAM,IAAI,CAAC;IAC9B,mBAAmB,EAAE,MAAM,IAAI,CAAC;IAEhC,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,EAAE,MAAM,IAAI,CAAC;IAExB,aAAa,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;CACtE"}
+{"version":3,"file":"treeView.types.d.ts","sourceRoot":"","sources":["../../../src/types/treeView.types.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EACR,SAAS,EACT,SAAS,EACT,qBAAqB,EACrB,SAAS,EACZ,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EACH,KAAK,KAAK,IAAI,2BAA2B,EAC5C,MAAM,4DAA4D,CAAC;AAEpE,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,eAAe,CAAC;AAE1D,MAAM,WAAW,eAAe;IAC5B,UAAU,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,QAAQ;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,QAAQ,EAAE,CAAC;IACtB,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACtB;AAED,MAAM,WAAW,qBAAsB,SAAQ,QAAQ;IACnD,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,MAAM,iBAAiB,CAAC,KAAK,GAAG,GAAG,IAAI,IAAI,CAC7C,cAAc,CAAC,KAAK,CAAC,EACrB,MAAM,GACJ,YAAY,CACjB,CAAC;AAEF,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IAEd,YAAY,EAAE,iBAAiB,CAAC;IAChC,UAAU,EAAE,OAAO,CAAC;IAEpB,OAAO,EAAE,MAAM,IAAI,CAAC;IACpB,QAAQ,EAAE,MAAM,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,sBAAsB;IACnC,sBAAsB,CAAC,EAAE,6BAA6B,CAAC;IAEvD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B,iBAAiB,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;IAC3D,2BAA2B,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;IACnE,gCAAgC,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;IAE9E,sBAAsB,CAAC,EAAE,KAAK,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;CAC9D;AAED,MAAM,WAAW,SAAU,SAAQ,sBAAsB;IACrD,IAAI,EAAE,qBAAqB,CAAC;IAC5B,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,aAAc,SAAQ,sBAAsB;IACzD,kBAAkB,CAAC,EAAE,iBAAiB,CAAC;CAC1C;AAED,MAAM,WAAW,aAAc,SAAQ,aAAa;IAChD,IAAI,EAAE,QAAQ,EAAE,CAAC;IAEjB,OAAO,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IACzC,QAAQ,CAAC,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAE3C,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAE1B,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC7B;AAED,KAAK,aAAa,GAAG,IAAI,CAAC,2BAA2B,EAAE,SAAS,GAAG,QAAQ,CAAC,CAAC;AAE7E,MAAM,WAAW,iBAAiB;IAC9B,KAAK,EAAE,iBAAiB,CAAC;IACzB,aAAa,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;IACxC,IAAI,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,6BAA6B;IAE1C,wBAAwB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAChD,uBAAuB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAC/C,kBAAkB,CAAC,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAG1C,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACzB;AAED,MAAM,MAAM,wBAAwB,GAChC,iBAAiB,GACf,6BAA6B,CAAC;AAEpC,MAAM,WAAW,WAAW;IACxB,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,EAAE,MAAM,IAAI,CAAC;IAExB,iBAAiB,EAAE,MAAM,IAAI,CAAC;IAC9B,mBAAmB,EAAE,MAAM,IAAI,CAAC;IAEhC,SAAS,EAAE,MAAM,IAAI,CAAC;IACtB,WAAW,EAAE,MAAM,IAAI,CAAC;IAExB,WAAW,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IACrC,aAAa,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;IAEvC,aAAa,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,EAAE,KAAK,IAAI,CAAC;CACtE"}

package/lib/typescript/utils/usePreviousState.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Get previous value of a state
+ * @param value state
+ * @returns previous value of @param value after it's updated
+ */
+export default function usePreviousState<T>(value: T): T | undefined;
+//# sourceMappingURL=usePreviousState.d.ts.map

package/lib/typescript/utils/usePreviousState.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePreviousState.d.ts","sourceRoot":"","sources":["../../../src/utils/usePreviousState.ts"],"names":[],"mappings":"AAEA;;;;GAIG;AACH,MAAM,CAAC,OAAO,UAAU,gBAAgB,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,iBAQnD"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-native-tree-multi-select",
-  "version": "1.2.0",
+  "version": "1.2.3",
   "description": "Super-fast Tree view with multi-selection capabilities, using checkboxes and search filtering.",
   "main": "lib/commonjs/index",
   "module": "lib/module/index",
@@ -37,7 +37,8 @@
     "build:android": "cd example/android && ./gradlew assembleDebug --no-daemon --console=plain -PreactNativeArchitectures=arm64-v8a",
     "build:ios": "cd example/ios && pod install && xcodebuild -workspace TreeMultiSelectExample.xcworkspace -scheme TreeMultiSelectExample -configuration Debug -sdk iphonesimulator CC=clang CPLUSPLUS=clang++ LD=clang LDPLUSPLUS=clang++ GCC_OPTIMIZATION_LEVEL=0 GCC_PRECOMPILE_PREFIX_HEADER=YES ASSETCATALOG_COMPILER_OPTIMIZATION=time DEBUG_INFORMATION_FORMAT=dwarf COMPILER_INDEX_STORE_ENABLE=NO",
     "bootstrap": "yarn example && yarn install && yarn example pods",
-    "clean": "del-cli android/build example/android/build example/android/app/build example/ios/build"
+    "clean": "del-cli android/build example/android/build example/android/app/build example/ios/build",
+    "generate-changelog": "auto-changelog"
   },
   "keywords": [
     "react-native",
@@ -88,6 +89,7 @@
     "@types/react-native": "0.70.0",
     "@types/react-native-vector-icons": "^6.4.13",
     "@types/react-test-renderer": "^18.0.0",
+    "auto-changelog": "^2.4.0",
     "commitlint": "^17.0.2",
     "del-cli": "^5.0.0",
     "eslint": "^8.4.1",
@@ -119,7 +121,7 @@
   "engines": {
     "node": ">= 16.0.0"
   },
-  "packageManager": "^yarn@1.22.15",
+  "packageManager": "yarn@1.22.22",
   "jest": {
     "preset": "react-native",
     "setupFilesAfterEnv": [

package/scripts/bootstrap.js

@@ -0,0 +1,29 @@
+const os = require('os');
+const path = require('path');
+const child_process = require('child_process');
+
+const root = path.resolve(__dirname, '..');
+const args = process.argv.slice(2);
+const options = {
+  cwd: process.cwd(),
+  env: process.env,
+  stdio: 'inherit',
+  encoding: 'utf-8',
+};
+
+if (os.type() === 'Windows_NT') {
+  options.shell = true;
+}
+
+let result;
+
+if (process.cwd() !== root || args.length) {
+  // We're not in the root of the project, or additional arguments were passed
+  // In this case, forward the command to `yarn`
+  result = child_process.spawnSync('yarn', args, options);
+} else {
+  // If `yarn` is run without arguments, perform bootstrap
+  result = child_process.spawnSync('yarn', ['bootstrap'], options);
+}
+
+process.exitCode = result.status;

package/src/TreeView.tsx

@@ -14,9 +14,12 @@ import {
   initializeNodeMaps,
   expandAll,
   collapseAll,
-  toggleCheckboxes
+  toggleCheckboxes,
+  expandNodes,
+  collapseNodes
 } from './helpers';
 import { useTreeViewStore } from './store/treeView.store';
+import usePreviousState from './utils/usePreviousState';
 
 const _TreeView = React.forwardRef<TreeViewRef, TreeViewProps>(
   (props, ref) => {
@@ -28,6 +31,8 @@ const _TreeView = React.forwardRef<TreeViewRef, TreeViewProps>(
 
       preselectedIds = [],
 
+      preExpandedIds = [],
+
       treeFlashListProps,
       checkBoxViewStyleProps,
       indentationMultiplier,
@@ -66,9 +71,14 @@ const _TreeView = React.forwardRef<TreeViewRef, TreeViewProps>(
       expandAll,
       collapseAll,
 
+      expandNodes,
+      collapseNodes,
+
       setSearchText
     }));
 
+    const prevSearchText = usePreviousState(searchText);
+
     function setSearchText(text: string, keys: string[] = ["name"]) {
       updateSearchText(text);
       updateSearchKeys(keys);
@@ -79,8 +89,11 @@ const _TreeView = React.forwardRef<TreeViewRef, TreeViewProps>(
 
       initializeNodeMaps(data);
 
-      // Check any preselected nodes
+      // Check any pre-selected nodes
       toggleCheckboxes(preselectedIds, true);
+
+      // Expand pre-expanded nodes
+      expandNodes(preExpandedIds);
       // eslint-disable-next-line react-hooks/exhaustive-deps
     }, []);
 
@@ -108,12 +121,20 @@ const _TreeView = React.forwardRef<TreeViewRef, TreeViewProps>(
           )));
         });
       }
-      else {
+      else if (prevSearchText && prevSearchText !== "") {
+        /* Collapse all nodes only if previous search query was non-empty: this is 
+        done to prevent node collapse on first render if preExpandedIds is provided */
         InteractionManager.runAfterInteractions(() => {
           updateExpanded(new Set());
         });
       }
-    }, [getIds, initialTreeViewData, searchText, updateExpanded]);
+    }, [
+      getIds,
+      initialTreeViewData,
+      prevSearchText,
+      searchText,
+      updateExpanded
+    ]);
 
     React.useEffect(() => {
       return () => {

package/src/__mocks__/generateTree.mock.ts

@@ -0,0 +1,125 @@
+import { TreeNode, __FlattenedTreeNode__ } from "../types/treeView.types";
+
+export const tree3d2b = [
+    {
+        "id": "1",
+        "name": "node1",
+        "children": [
+            {
+                "id": "1.1",
+                "name": "node1.1",
+                "children": [
+                    {
+                        "id": "1.1.1",
+                        "name": "node1.1.1"
+                    },
+                    {
+                        "id": "1.1.2",
+                        "name": "node1.1.2"
+                    }
+                ]
+            },
+            {
+                "id": "1.2",
+                "name": "node1.2",
+                "children": [
+                    {
+                        "id": "1.2.1",
+                        "name": "node1.2.1"
+                    },
+                    {
+                        "id": "1.2.2",
+                        "name": "node1.2.2"
+                    }
+                ]
+            }
+        ]
+    },
+    {
+        "id": "2",
+        "name": "node2",
+        "children": [
+            {
+                "id": "2.1",
+                "name": "node2.1",
+                "children": [
+                    {
+                        "id": "2.1.1",
+                        "name": "node2.1.1"
+                    },
+                    {
+                        "id": "2.1.2",
+                        "name": "node2.1.2"
+                    }
+                ]
+            },
+            {
+                "id": "2.2",
+                "name": "node2.2",
+                "children": [
+                    {
+                        "id": "2.2.1",
+                        "name": "node2.2.1"
+                    },
+                    {
+                        "id": "2.2.2",
+                        "name": "node2.2.2"
+                    }
+                ]
+            }
+        ]
+    }
+];
+
+export function generateTree(
+    depth: number,
+    breadth: number,
+    prefix: string = ''
+): TreeNode[] {
+    let nodes: TreeNode[] = [];
+
+    for (let i = 1; i <= breadth; i++) {
+        let node: TreeNode = {
+            id: `${prefix}${i}`,
+            name: `node${prefix}${i}`
+        };
+
+        if (depth > 1) {
+            node.children = generateTree(depth - 1, breadth, `${prefix}${i}.`);
+        }
+
+        nodes.push(node);
+    }
+
+    return nodes;
+}
+
+export function generateExpectedFlatTree(
+    nodes: TreeNode[],
+    expandedIds: Set<string>,
+    level: number = 0
+): __FlattenedTreeNode__[] {
+    let flattened: __FlattenedTreeNode__[] = [];
+    for (let node of nodes) {
+        flattened.push({ ...node, level: level });
+        if (node.children && expandedIds.has(node.id)) {
+            flattened = [
+                ...flattened,
+                ...generateExpectedFlatTree(node.children, expandedIds, level + 1)
+            ];
+        }
+    }
+    return flattened;
+}
+
+export function createRandomNumberSet() {
+    let randomNumbers = new Set<string>();
+
+    while (randomNumbers.size < 10) {
+        // Random number between 0 and 99
+        let randomNumber = Math.floor(Math.random() * 100);
+        randomNumbers.add(randomNumber.toString());
+    }
+
+    return randomNumbers;
+}

package/src/__mocks__/zustand.ts

@@ -0,0 +1,24 @@
+import { act } from '@testing-library/react-native';
+import { StateCreator } from 'zustand';
+
+const { create: actualCreate } = jest.requireActual('zustand');
+
+// a variable to hold reset functions for all stores declared in the app
+const storeResetFns = new Set<() => void>();
+
+// when creating a store, we get its initial state, create a reset function and add it in the set
+export const create = <S>(createState: StateCreator<S>) => {
+    const store = actualCreate(createState);
+    const initialState = store.getState();
+    storeResetFns.add(() => store.setState(initialState, true));
+    return store;
+};
+
+// Reset all stores after each test run
+beforeEach(async () => {
+    await act(() =>
+        storeResetFns.forEach(resetFn => {
+            resetFn();
+        })
+    );
+});

package/src/__tests__/expandCollapse.helper.test.ts

@@ -0,0 +1,189 @@
+jest.mock('zustand');
+
+import { useTreeViewStore } from '../store/treeView.store';
+import { tree3d2b } from "../__mocks__/generateTree.mock";
+import {
+    collapseAll,
+    collapseNodes,
+    expandAll,
+    expandNodes,
+    handleToggleExpand,
+    initializeNodeMaps
+} from '../helpers';
+import { act } from 'react-test-renderer';
+
+describe('handleToggleExpand', () => {
+    beforeEach(() => {
+        useTreeViewStore.setState(useTreeViewStore.getState(), true);
+
+        useTreeViewStore.getState().updateInitialTreeViewData(tree3d2b);
+        initializeNodeMaps(tree3d2b);
+    });
+
+    test('handleToggleExpand correctly toggles the expanded state of a tree node', () => {
+        act(() => {
+            handleToggleExpand('1');
+        });
+
+        // Node '1' should now be expanded
+        let { expanded } = useTreeViewStore.getState();
+        expect(expanded.has('1')).toBeTruthy();
+
+        act(() => {
+            handleToggleExpand('1.2');
+        });
+
+        // Node '1.2' should now be expanded, Node '1'(parent) should remain expanded
+        expanded = useTreeViewStore.getState().expanded;
+        expect(expanded.has('1.2')).toBeTruthy();
+        expect(expanded.has('1')).toBeTruthy();
+
+        act(() => {
+            handleToggleExpand('1');
+            handleToggleExpand('2');
+        });
+
+        // Node '1' and its descendants should now be collapsed but Node '2' should remain expanded
+        expanded = useTreeViewStore.getState().expanded;
+
+        expect(expanded.has('1')).toBeFalsy();
+        expect(expanded.has('1.1')).toBeFalsy();
+        expect(expanded.has('1.2')).toBeFalsy();
+        expect(expanded.has('1.2.1')).toBeFalsy();
+
+        expect(expanded.has('2')).toBeTruthy();
+    });
+});
+
+describe('expandAll', () => {
+    beforeEach(() => {
+        useTreeViewStore.setState(useTreeViewStore.getState(), true);
+
+        // Setup mock tree
+        useTreeViewStore.getState().updateInitialTreeViewData(tree3d2b);
+        initializeNodeMaps(tree3d2b);
+    });
+
+    it('calls expandAll on initial tree(all collapsed)', () => {
+        act(() => {
+            expandAll();
+        });
+
+        const { expanded, nodeMap } = useTreeViewStore.getState();
+
+        // Convert nodeMap.keys() iterator to a Set for comparison
+        const nodeKeys = new Set(nodeMap.keys());
+        expect(expanded).toEqual(nodeKeys);
+    });
+
+    it('expands all node in tree with some nodes which are already expanded', () => {
+        act(() => {
+            handleToggleExpand('1');
+            handleToggleExpand('2');
+            handleToggleExpand('1.1');
+            handleToggleExpand('1.2');
+            handleToggleExpand('1.1');
+
+            expandAll();
+        });
+
+        const { expanded, nodeMap } = useTreeViewStore.getState();
+
+        // Convert nodeMap.keys() iterator to a Set for comparison
+        const nodeKeys = new Set(nodeMap.keys());
+        expect(expanded).toEqual(nodeKeys);
+    });
+});
+
+describe('collapseAll', () => {
+    beforeEach(() => {
+        useTreeViewStore.setState(useTreeViewStore.getState(), true);
+
+        // Setup mock tree
+        useTreeViewStore.getState().updateInitialTreeViewData(tree3d2b);
+        initializeNodeMaps(tree3d2b);
+    });
+
+    it('calls collapseAll on initial tree(all collapsed)', () => {
+        act(() => {
+            collapseAll();
+        });
+
+        const { expanded } = useTreeViewStore.getState();
+        expect(expanded).toEqual(new Set<string>());
+    });
+
+    it('collapses all node in tree with some nodes which are already expanded', () => {
+        act(() => {
+            handleToggleExpand('1');
+            handleToggleExpand('2');
+            handleToggleExpand('1.1');
+            handleToggleExpand('1.2');
+            handleToggleExpand('1.1');
+
+            collapseAll();
+        });
+
+        const { expanded } = useTreeViewStore.getState();
+
+        expect(expanded).toEqual(new Set<string>());
+    });
+});
+
+describe('expandNodes & collapseNodes', () => {
+    beforeEach(() => {
+        useTreeViewStore.setState(useTreeViewStore.getState(), true);
+
+        useTreeViewStore.getState().updateInitialTreeViewData(tree3d2b);
+        initializeNodeMaps(tree3d2b);
+    });
+
+    it('expands and then collapses multiple nodes as needed', () => {
+        act(() => {
+            expandNodes(["1", "2.1", "2.2.2"]);
+        });
+
+        const { expanded: expandedAfterExpandNodes } = useTreeViewStore.getState();
+        // Both nodes and their parents should be expanded
+        expect(expandedAfterExpandNodes.has("1")).toBeTruthy();
+        expect(expandedAfterExpandNodes.has("1.1")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("1.1.1")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("1.1.2")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("1.2")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("1.2.1")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("1.2.2")).toBeFalsy();
+
+        expect(expandedAfterExpandNodes.has("2")).toBeTruthy();
+        expect(expandedAfterExpandNodes.has("2.1")).toBeTruthy();
+        expect(expandedAfterExpandNodes.has("2.1.1")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("2.1.2")).toBeFalsy();
+
+        expect(expandedAfterExpandNodes.has("2.2")).toBeTruthy();
+        expect(expandedAfterExpandNodes.has("2.2.1")).toBeFalsy();
+        expect(expandedAfterExpandNodes.has("2.2.2")).toBeTruthy();
+
+        // Then collapse the same nodes
+        act(() => {
+            collapseNodes(["1", "2.2"]);
+        });
+
+        const { expanded: expandedAfterCollapseNodes } = useTreeViewStore.getState();
+        // Both nodes and their children should be collapsed
+        expect(expandedAfterCollapseNodes.has("1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.1.1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.1.2")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.2")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.2.1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("1.2.2")).toBeFalsy();
+
+        expect(expandedAfterCollapseNodes.has("2")).toBeTruthy();
+        expect(expandedAfterCollapseNodes.has("2.1")).toBeTruthy();
+        expect(expandedAfterCollapseNodes.has("2.1.1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("2.1.2")).toBeFalsy();
+
+        expect(expandedAfterCollapseNodes.has("2.2")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("2.2.1")).toBeFalsy();
+        expect(expandedAfterCollapseNodes.has("2.2.2")).toBeFalsy();
+    });
+});

package/src/__tests__/flattenTree.helper.test.ts

@@ -0,0 +1,78 @@
+import { getFlattenedTreeData } from "../helpers";
+import { generateExpectedFlatTree, tree3d2b } from "../__mocks__/generateTree.mock";
+import { TreeNode } from "../types/treeView.types";
+
+describe('getFlattenedTreeData', () => {
+    let nodes: TreeNode[];
+    let expandedIds: Set<string>;
+
+    beforeEach(() => {
+        nodes = tree3d2b;
+        expandedIds = new Set<string>();
+    });
+
+    // Hardcoded test
+    test('should correctly flatten a simple tree', () => {
+        const simpleNodes: TreeNode[] = [{
+            id: '1',
+            name: 'node1',
+            children: [
+                {
+                    id: '1.1', name: 'node1.1', children: [
+                        { id: '1.1.1', name: 'node1.1.1' },
+                        { id: '1.1.2', name: 'node1.1.2' },
+                        { id: '1.1.3', name: 'node1.1.3' },
+                    ]
+                },
+                {
+                    id: '1.2', name: 'node1.2', children: [
+                        { id: '1.2.1', name: 'node1.2.1' },
+                        { id: '1.2.2', name: 'node1.2.2' },
+                        { id: '1.2.3', name: 'node1.2.3' },
+                    ]
+                },
+                {
+                    id: '1.3', name: 'node1.3', children: [
+                        { id: '1.3.1', name: 'node1.3.1' },
+                        { id: '1.3.2', name: 'node1.3.2' },
+                        { id: '1.3.3', name: 'node1.3.3' },
+                    ]
+                },
+            ],
+        }];
+        const simpleExpandedIds = new Set(['1', '1.1', '1.2', '1.3']);
+        const result = getFlattenedTreeData(simpleNodes, simpleExpandedIds);
+        expect(result).toEqual([
+            { id: '1', name: 'node1', level: 0, children: simpleNodes?.[0]?.children },
+            { id: '1.1', name: 'node1.1', level: 1, children: simpleNodes?.[0]?.children?.[0]?.children },
+            { id: '1.1.1', name: 'node1.1.1', level: 2 },
+            { id: '1.1.2', name: 'node1.1.2', level: 2 },
+            { id: '1.1.3', name: 'node1.1.3', level: 2 },
+            { id: '1.2', name: 'node1.2', level: 1, children: simpleNodes?.[0]?.children?.[1]?.children },
+            { id: '1.2.1', name: 'node1.2.1', level: 2 },
+            { id: '1.2.2', name: 'node1.2.2', level: 2 },
+            { id: '1.2.3', name: 'node1.2.3', level: 2 },
+            { id: '1.3', name: 'node1.3', level: 1, children: simpleNodes?.[0]?.children?.[2]?.children },
+            { id: '1.3.1', name: 'node1.3.1', level: 2 },
+            { id: '1.3.2', name: 'node1.3.2', level: 2 },
+            { id: '1.3.3', name: 'node1.3.3', level: 2 },
+        ]);
+    });
+
+    // Dynamic test
+    test('should return a flat array of nodes with correct level property', () => {
+        expandedIds.add('1');
+        expandedIds.add('1.1');
+        expandedIds.add('1.2');
+        expandedIds.add('1.3');
+        const result = getFlattenedTreeData(nodes, expandedIds);
+        const expected = generateExpectedFlatTree(nodes, expandedIds);
+        expect(result).toEqual(expected);
+    });
+
+    // Edge case: tree with no expanded nodes
+    test('should return a flat array of root nodes when no nodes are expanded', () => {
+        const result = getFlattenedTreeData(nodes, new Set());
+        expect(result).toEqual(nodes.map(node => ({ ...node, level: 0 })));
+    });
+});