vcsim: fix ModifyListView to return unresolved references

BREAKING: api: view.ListView.{Add,Remove,Reset} methods now return unresolved references
bruceadowns · Oct 11, 2023 · 0754d75 · 0754d75
1 parent 88629ae
commit 0754d75
Show file tree

Hide file tree

Showing 6 changed files with 71 additions and 35 deletions.
diff --git a/simulator/race_test.go b/simulator/race_test.go
@@ -187,7 +187,7 @@ func TestRace(t *testing.T) {
 					if terr != nil {
 						t.Error(terr)
 					}
-					terr = lv.Add(ctx, []types.ManagedObjectReference{r.Result.(types.ManagedObjectReference)})
+					_, terr = lv.Add(ctx, []types.ManagedObjectReference{r.Result.(types.ManagedObjectReference)})
 					if terr != nil {
 						t.Error(terr)
 					}
@@ -221,7 +221,7 @@ func TestRace(t *testing.T) {
 						time.AfterFunc(100*time.Millisecond, func() {
 							defer wg.Done()
 
-							_ = lv.Remove(ctx, []types.ManagedObjectReference{vm.Reference()})
+							_, _ = lv.Remove(ctx, []types.ManagedObjectReference{vm.Reference()})
 							task, _ := vm.PowerOff(ctx)
 							_ = task.Wait(ctx)
 						})

diff --git a/simulator/view_manager.go b/simulator/view_manager.go
@@ -227,8 +227,8 @@ func (m *ViewManager) CreateListView(ctx *Context, req *types.CreateListView) so
 	body := new(methods.CreateListViewBody)
 	list := new(ListView)
 
-	if err := list.add(ctx, req.Obj); err != nil {
-		body.Fault_ = Fault("", err)
+	if refs := list.add(ctx, req.Obj); len(refs) != 0 {
+		body.Fault_ = Fault("", &types.ManagedObjectNotFound{Obj: refs[0]})
 		return body
 	}
 
@@ -249,15 +249,16 @@ func (v *ListView) update(ctx *Context) {
 	ctx.Map.Update(v, []types.PropertyChange{{Name: "view", Val: v.View}})
 }
 
-func (v *ListView) add(ctx *Context, refs []types.ManagedObjectReference) *types.ManagedObjectNotFound {
+func (v *ListView) add(ctx *Context, refs []types.ManagedObjectReference) []types.ManagedObjectReference {
+	var unresolved []types.ManagedObjectReference
 	for _, ref := range refs {
 		obj := ctx.Session.Get(ref)
 		if obj == nil {
-			return &types.ManagedObjectNotFound{Obj: ref}
+			unresolved = append(unresolved, ref)
 		}
 		v.View = append(v.View, ref)
 	}
-	return nil
+	return unresolved
 }
 
 func (v *ListView) DestroyView(ctx *Context, c *types.DestroyView) soap.HasFault {
@@ -266,19 +267,19 @@ func (v *ListView) DestroyView(ctx *Context, c *types.DestroyView) soap.HasFault
 }
 
 func (v *ListView) ModifyListView(ctx *Context, req *types.ModifyListView) soap.HasFault {
-	body := new(methods.ModifyListViewBody)
+	body := &methods.ModifyListViewBody{
+		Res: new(types.ModifyListViewResponse),
+	}
+
+	body.Res.Returnval = v.add(ctx, req.Add)
 
 	for _, ref := range req.Remove {
 		RemoveReference(&v.View, ref)
+		if ctx.Map.Get(ref) == nil {
+			body.Res.Returnval = append(body.Res.Returnval, ref)
+		}
 	}
 
-	if err := v.add(ctx, req.Add); err != nil {
-		body.Fault_ = Fault("", err)
-		return body
-	}
-
-	body.Res = new(types.ModifyListViewResponse)
-
 	if len(req.Remove) != 0 || len(req.Add) != 0 {
 		v.update(ctx)
 	}
@@ -287,16 +288,13 @@ func (v *ListView) ModifyListView(ctx *Context, req *types.ModifyListView) soap.
 }
 
 func (v *ListView) ResetListView(ctx *Context, req *types.ResetListView) soap.HasFault {
-	body := new(methods.ResetListViewBody)
+	body := &methods.ResetListViewBody{
+		Res: new(types.ResetListViewResponse),
+	}
 
 	v.View = nil
 
-	if err := v.add(ctx, req.Obj); err != nil {
-		body.Fault_ = Fault("", err)
-		return body
-	}
-
-	body.Res = new(types.ResetListViewResponse)
+	body.Res.Returnval = v.add(ctx, req.Obj)
 
 	v.update(ctx)
 

diff --git a/simulator/view_manager_test.go b/simulator/view_manager_test.go
@@ -18,6 +18,7 @@ package simulator
 
 import (
 	"context"
+	"log"
 	"sync"
 	"testing"
 
@@ -198,3 +199,28 @@ func TestViewManager_CreateContainerView(t *testing.T) {
 		t.Errorf("Failed to run simulation: %s", err.Error())
 	}
 }
+
+func TestListViewModify(t *testing.T) {
+	Test(func(ctx context.Context, c *vim25.Client) {
+		list, err := view.NewManager(c).CreateListView(ctx, nil)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		defer list.Destroy(ctx)
+
+		add := []types.ManagedObjectReference{
+			c.ServiceContent.RootFolder,
+			{Type: "Folder", Value: "invalid"},
+		}
+
+		refs, err := list.Add(ctx, add)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if len(refs) != 1 {
+			t.Errorf("unresolved refs=%s", refs)
+		}
+	})
+}
diff --git a/view/example_test.go b/view/example_test.go
@@ -211,7 +211,7 @@ func ExampleListView_tasks() {
 
 						switch info.State {
 						case types.TaskInfoStateSuccess, types.TaskInfoStateError:
-							_ = list.Remove(ctx, []types.ManagedObjectReference{update.Obj})
+							_, _ = list.Remove(ctx, []types.ManagedObjectReference{update.Obj})
 							result[info.State]++
 							n--
 							wg.Done()
@@ -228,7 +228,7 @@ func ExampleListView_tasks() {
 			if err != nil {
 				return err
 			}
-			err = list.Add(ctx, []types.ManagedObjectReference{task.Reference()})
+			_, err = list.Add(ctx, []types.ManagedObjectReference{task.Reference()})
 			if err != nil {
 				return err
 			}

diff --git a/view/list_view.go b/view/list_view.go
@@ -34,29 +34,41 @@ func NewListView(c *vim25.Client, ref types.ManagedObjectReference) *ListView {
 	}
 }
 
-func (v ListView) Add(ctx context.Context, refs []types.ManagedObjectReference) error {
+func (v ListView) Add(ctx context.Context, refs []types.ManagedObjectReference) ([]types.ManagedObjectReference, error) {
 	req := types.ModifyListView{
 		This: v.Reference(),
 		Add:  refs,
 	}
-	_, err := methods.ModifyListView(ctx, v.Client(), &req)
-	return err
+	res, err := methods.ModifyListView(ctx, v.Client(), &req)
+	if err != nil {
+		return nil, err
+	}
+
+	return res.Returnval, nil
 }
 
-func (v ListView) Remove(ctx context.Context, refs []types.ManagedObjectReference) error {
+func (v ListView) Remove(ctx context.Context, refs []types.ManagedObjectReference) ([]types.ManagedObjectReference, error) {
 	req := types.ModifyListView{
 		This:   v.Reference(),
 		Remove: refs,
 	}
-	_, err := methods.ModifyListView(ctx, v.Client(), &req)
-	return err
+	res, err := methods.ModifyListView(ctx, v.Client(), &req)
+	if err != nil {
+		return nil, err
+	}
+
+	return res.Returnval, nil
 }
 
-func (v ListView) Reset(ctx context.Context, refs []types.ManagedObjectReference) error {
+func (v ListView) Reset(ctx context.Context, refs []types.ManagedObjectReference) ([]types.ManagedObjectReference, error) {
 	req := types.ResetListView{
 		This: v.Reference(),
 		Obj:  refs,
 	}
-	_, err := methods.ResetListView(ctx, v.Client(), &req)
-	return err
+	res, err := methods.ResetListView(ctx, v.Client(), &req)
+	if err != nil {
+		return nil, err
+	}
+
+	return res.Returnval, nil
 }
diff --git a/view/task_view.go b/view/task_view.go
@@ -73,7 +73,7 @@ func (v TaskView) Collect(ctx context.Context, f func([]types.TaskInfo)) error {
 					tasks = change.Val.(types.ArrayOfManagedObjectReference).ManagedObjectReference
 					if len(tasks) != 0 {
 						reset = func() {
-							_ = v.Reset(ctx, tasks)
+							_, _ = v.Reset(ctx, tasks)
 
 							// Remember any tasks we've reported as complete already,
 							// to avoid reporting multiple times when Reset is triggered.
@@ -113,7 +113,7 @@ func (v TaskView) Collect(ctx context.Context, f func([]types.TaskInfo)) error {
 		if reset != nil {
 			reset()
 		} else if len(prune) != 0 {
-			_ = v.Remove(ctx, prune)
+			_, _ = v.Remove(ctx, prune)
 		}
 
 		if len(tasks) != 0 && len(infos) == 0 {