From 0b6bb7a98579c0c5c9bda03a1486eaac124b2d11 Mon Sep 17 00:00:00 2001
From: victor-meng <56978073+victor-meng@users.noreply.github.com>
Date: Thu, 5 May 2022 18:27:13 -0700
Subject: [PATCH] Fix stored procedures, UDF, and triggers sometimes don't show
 up in the resource tree after expanding (#1267)

---
 src/Explorer/Tree/ResourceTree.tsx | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/Explorer/Tree/ResourceTree.tsx b/src/Explorer/Tree/ResourceTree.tsx
index dd8c3c9c6..77f694598 100644
--- a/src/Explorer/Tree/ResourceTree.tsx
+++ b/src/Explorer/Tree/ResourceTree.tsx
@@ -611,7 +611,8 @@ export const ResourceTree: React.FC<ResourceTreeProps> = ({ container }: Resourc
             ]),
         contextMenu: ResourceTreeContextMenuButtonFactory.createStoreProcedureContextMenuItems(container, sp),
       })),
-      onClick: () => {
+      onClick: async () => {
+        await collection.loadStoredProcedures();
         collection.selectedSubnodeKind(ViewModels.CollectionTabKind.StoredProcedures);
         refreshActiveTab(
           (tab: TabsBase) =>
@@ -635,7 +636,8 @@ export const ResourceTree: React.FC<ResourceTreeProps> = ({ container }: Resourc
             ]),
         contextMenu: ResourceTreeContextMenuButtonFactory.createUserDefinedFunctionContextMenuItems(container, udf),
       })),
-      onClick: () => {
+      onClick: async () => {
+        await collection.loadUserDefinedFunctions();
         collection.selectedSubnodeKind(ViewModels.CollectionTabKind.UserDefinedFunctions);
         refreshActiveTab(
           (tab: TabsBase) =>
@@ -657,7 +659,8 @@ export const ResourceTree: React.FC<ResourceTreeProps> = ({ container }: Resourc
             .isDataNodeSelected(collection.databaseId, collection.id(), [ViewModels.CollectionTabKind.Triggers]),
         contextMenu: ResourceTreeContextMenuButtonFactory.createTriggerContextMenuItems(container, trigger),
       })),
-      onClick: () => {
+      onClick: async () => {
+        await collection.loadTriggers();
         collection.selectedSubnodeKind(ViewModels.CollectionTabKind.Triggers);
         refreshActiveTab(
           (tab: TabsBase) =>