fix: should browse until complete by default

electricbubble · Mar 3, 2022 · e67ef0a · e67ef0a
1 parent 091a095
commit e67ef0a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/installationproxy.go b/installationproxy.go
@@ -47,11 +47,18 @@ func (p *installationProxy) Browse(opts ...InstallationProxyOption) (currentList
 	if err = respPkt.Unmarshal(&reply); err != nil {
 		return nil, err
 	}
-	if reply.Status != "BrowsingApplications" {
-		return nil, fmt.Errorf("installation proxy 'Browse' status: %s", reply.Status)
-	}
 
 	currentList = reply.CurrentList
+
+	for reply.Status != "Complete" {
+		if respPkt, err = p.client.ReceivePacket(); err != nil {
+			return nil, err
+		}
+		if err = respPkt.Unmarshal(&reply); err != nil {
+			return nil, err
+		}
+		currentList = append(currentList, reply.CurrentList)
+	}
 	return
 }
 

diff --git a/installationproxy_test.go b/installationproxy_test.go
@@ -36,8 +36,12 @@ func Test_installationProxy_Browse(t *testing.T) {
 	t.Log(len(currentList))
 
 	for _, cl := range currentList {
-		app := cl.(map[string]interface{})
-		t.Log(app)
+		app, ok := cl.(map[string]interface{})
+		if ok {
+			t.Log(app)
+		} else {
+			t.Log(cl)
+		}
 	}
 }