From 295bde821c62745c13bcbaa202d7d0e96d69dc74 Mon Sep 17 00:00:00 2001
From: Michael Z <michael.zhao@citrix.com>
Date: Tue, 7 May 2019 14:23:03 +0800
Subject: [PATCH] CP-30423: Fix UT code build error.

Signed-off-by: Michael Z <michael.zhao@citrix.com>
---
 .../UnitTestHelper/MockObjectBuilders/MockActionFactory.cs  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/XenAdminTests/UnitTests/UnitTestHelper/MockObjectBuilders/MockActionFactory.cs b/XenAdminTests/UnitTests/UnitTestHelper/MockObjectBuilders/MockActionFactory.cs
index 6bff2bae3..eca2af9be 100644
--- a/XenAdminTests/UnitTests/UnitTestHelper/MockObjectBuilders/MockActionFactory.cs
+++ b/XenAdminTests/UnitTests/UnitTestHelper/MockObjectBuilders/MockActionFactory.cs
@@ -77,15 +77,15 @@ namespace XenAdminTests.UnitTests.UnitTestHelper.MockObjectBuilders
 
         private AsyncAction gpuAssignAction(string id)
         {
-            Mock<GPU_group> group = ObjectManager.NewXenObject<GPU_group>(id);
-            Mock<VGPU_type> vgpuType = ObjectManager.NewXenObject<VGPU_type>(id);
+            Mock<VGPU> vgpu = ObjectManager.NewXenObject<VGPU>(id);
+            List<VGPU> vgpus = new List<VGPU>() { vgpu.Object };
             Mock<VM> vm = ObjectManager.NewXenObject<VM>(id);
             ObjectManager.MockConnectionFor(id).Setup(c => c.ResolveAll(It.IsAny<List<XenRef<VGPU>>>())).Returns(new List<VGPU>());
             ObjectManager.MockProxyFor(id).Setup(
                 p =>
                 p.async_vgpu_create(It.IsAny<string>(), It.IsAny<string>(), It.IsAny<string>(), "0",
                               It.IsAny<object>(), It.IsAny<string>())).Returns(new Response<string>("ok"));
-            return new GpuAssignAction(vm.Object, group.Object, vgpuType.Object);
+            return new GpuAssignAction(vm.Object, vgpus);
         }
     }
 }