Fix treatment units tree not initially expanded for selection in quer…

…y params (#2369)

* refresh tree of available treatment units after query params has completed loading

* apiQueriesCompleted should only be true the first time the query has changed from loading to not loading

packages/qmongjs/src/components/FilterMenu/TreatmentQualityFilterMenu/index.tsx

@@ -102,7 +102,7 @@ export function TreatmentQualityFilterMenu({
   // the first call after the prerender is done.
 
   const [prevReady, setPrevReady] = useState(router.isReady);
-  const shouldRefreshInititalState = prevReady !== router.isReady;
+  const prerenderFinished = prevReady !== router.isReady;
 
   useEffect(() => {
     setPrevReady(router.isReady);
@@ -170,6 +170,15 @@ export function TreatmentQualityFilterMenu({
     queryContext.type,
   );
 
+  const [prevApiQueryLoading, setPrevApiQueryLoading] = useState(
+    unitNamesQuery.isLoading,
+  );
+  const apiQueriesCompleted = prevApiQueryLoading && !unitNamesQuery.isLoading;
+
+  useEffect(() => {
+    setPrevApiQueryLoading(unitNamesQuery.isLoading);
+  }, [unitNamesQuery.isLoading]);
+
   const treatmentUnits = getTreatmentUnitsTree(unitNamesQuery);
 
   optionsMap.set(treatmentUnitsKey, {
@@ -270,6 +279,8 @@ export function TreatmentQualityFilterMenu({
     return filterSettingsValue?.valueLabel ?? null;
   };
 
+  const shouldRefreshInitialState = prerenderFinished || apiQueriesCompleted;
+
   return (
     <>
       {!(medicalFieldData || registryNameData) && (
@@ -278,7 +289,7 @@ export function TreatmentQualityFilterMenu({
         </Alert>
       )}
       <FilterMenu
-        refreshState={shouldRefreshInititalState}
+        refreshState={shouldRefreshInitialState}
         onSelectionChanged={handleFilterChanged}
         onFilterInitialized={onFilterInitialized}
       >
@@ -310,7 +321,7 @@ export function TreatmentQualityFilterMenu({
           filterkey={levelKey}
         />
         <TreeViewFilterSection
-          refreshState={shouldRefreshInititalState}
+          refreshState={shouldRefreshInitialState}
           treedata={medicalFields.treedata}
           defaultvalues={medicalFields.defaults}
           autouncheckid={medicalFields.defaults[0].value}
@@ -326,7 +337,7 @@ export function TreatmentQualityFilterMenu({
           searchbox={true}
         />
         <TreeViewFilterSection
-          refreshState={shouldRefreshInititalState}
+          refreshState={shouldRefreshInitialState}
           treedata={treatmentUnits.treedata}
           defaultvalues={treatmentUnits.defaults}
           initialselections={