5

Author: RebeccaMahany

pkg/windows/windowsupdate/mem_test.go

@@ -6,7 +6,7 @@ import (
 	"runtime"
 	"testing"
 
-	"github.com/go-ole/go-ole"
+	"github.com/go-ole/go-ole/oleutil"
 	"github.com/kolide/launcher/pkg/windows/oleconv"
 	"github.com/scjalliance/comshim"
 	"github.com/shirou/gopsutil/v3/process"
@@ -23,23 +23,16 @@ func TestMemory(t *testing.T) {
 	session, err := NewUpdateSession()
 	require.NoError(t, err)
 
-	searcher, err := session.CreateUpdateSearcher()
-	require.NoError(t, err)
-
-	fmt.Println("before getting name ID")
+	fmt.Println("created update session")
 	takeAndPrintMemoryUsageSnapshot(t)
 
-	/*
-		searchVariant, err := oleutil.CallMethod(searcher.disp, "Search", "Type='Software'")
-		require.NoError(t, err)
-	*/
-	displayID, err := searcher.disp.GetSingleIDOfName("Search")
+	searcher, err := session.CreateUpdateSearcher()
 	require.NoError(t, err)
 
-	fmt.Println("before Search")
+	fmt.Println("created update searcher")
 	takeAndPrintMemoryUsageSnapshot(t)
 
-	searchVariant, err := searcher.disp.Invoke(displayID, ole.DISPATCH_METHOD, "Type='Software'")
+	searchVariant, err := oleutil.CallMethod(searcher.disp, "Search", "Type='Software'")
 	require.NoError(t, err)
 
 	fmt.Println("after Search")