CA-122441: Add unit test for GPU_group supported/enabled VGPU types

Signed-off-by: John Else <john.else@citrix.com>

ocaml/test/OMakefile

@@ -33,6 +33,7 @@ OCAML_OBJS = \
 	test_pool_license \
 	test_platformdata \
 	test_sm_features \
+	test_gpu_group \
 	test_pci_db \
 	test_pci_helpers \
 	test_vgpu_type \

ocaml/test/suite.ml

@@ -25,6 +25,7 @@ let base_suite =
 			Test_pool_license.test;
 			Test_platformdata.test;
 			Test_sm_features.test;
+			Test_gpu_group.test;
 			Test_pci_db.test;
 			Test_pci_helpers.test;
 			Test_vgpu_type.test;

ocaml/test/test_gpu_group.ml

@@ -0,0 +1,79 @@
+(*
+ * Copyright (C) 2013 Citrix Systems Inc.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published
+ * by the Free Software Foundation; version 2.1 only. with the special
+ * exception on linking described in file LICENSE.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *)
+
+open OUnit
+open Test_vgpu_common
+
+let test_supported_enabled_types () =
+	let __context = Test_common.make_test_database () in
+	(* Create a GPU group containing a single K2 PGPU. *)
+	let host = List.hd (Db.Host.get_all ~__context) in
+	let gPU_group = Test_common.make_gpu_group ~__context () in
+	let pgpu = make_pgpu ~__context ~host ~gPU_group default_k2 in
+	(* Update the group's enabled and supported types, and check that they
+	 * contain all the types enabled and supported by the PGPU. *)
+	Xapi_gpu_group.update_supported_VGPU_types ~__context ~self:gPU_group;
+	Xapi_gpu_group.update_enabled_VGPU_types ~__context ~self:gPU_group;
+	let vgpu_types_and_refs =
+		List.map
+			(fun vgpu_type ->
+				(vgpu_type, Xapi_vgpu_type.find_or_create ~__context vgpu_type))
+			k2_vgpu_types
+	in
+	let group_supported_types =
+		Db.GPU_group.get_supported_VGPU_types ~__context ~self:gPU_group
+	in
+	let group_enabled_types =
+		Db.GPU_group.get_enabled_VGPU_types ~__context ~self:gPU_group
+	in
+	List.iter
+		(fun (vgpu_type, vgpu_type_ref) ->
+			let msg_supported =
+				Printf.sprintf
+					"GPU group does not list %s as supported"
+					vgpu_type.Xapi_vgpu_type.model_name
+			in
+			let msg_enabled =
+				Printf.sprintf
+					"GPU group does not list %s as enabled"
+					vgpu_type.Xapi_vgpu_type.model_name
+			in
+			assert_bool msg_supported (List.mem vgpu_type_ref group_supported_types);
+			assert_bool msg_enabled (List.mem vgpu_type_ref group_enabled_types))
+		vgpu_types_and_refs;
+	(* Invalidate the PGPU's host ref, and run a GC pass; this should destroy the
+	 * pgpu, and clear the group's supported and enabled types. *)
+	Db.PGPU.set_host ~__context ~self:pgpu ~value:Ref.null;
+	Db_gc.gc_PGPUs ~__context;
+	let group_supported_types =
+		Db.GPU_group.get_supported_VGPU_types ~__context ~self:gPU_group
+	in
+	let group_enabled_types =
+		Db.GPU_group.get_enabled_VGPU_types ~__context ~self:gPU_group
+	in
+	assert_equal
+		~msg:"PGPU has not been destroyed"
+		(Db.is_valid_ref __context pgpu) false;
+	assert_equal
+		~msg:"GPU group still has supported types after GC"
+		group_supported_types [];
+	assert_equal
+		~msg:"GPU group still has enabled types after GC"
+		group_enabled_types []
+
+let test =
+	"test_gpu_group" >:::
+		[
+			"test_supported_enabled_types" >:: test_supported_enabled_types;
+		]