[TEST] Add test to set model path of ml-service
authorgichan2-jang <gichan2.jang@samsung.com>
Fri, 12 Apr 2024 04:26:23 +0000 (13:26 +0900)
committerjaeyun-jung <39614140+jaeyun-jung@users.noreply.github.com>
Fri, 12 Apr 2024 06:02:41 +0000 (15:02 +0900)
Add test to set model path of ml-service.

Signed-off-by: gichan2-jang <gichan2.jang@samsung.com>
tests/capi/unittest_capi_service_offloading.cc

index dba89a9..9c30d44 100644 (file)
@@ -338,9 +338,10 @@ TEST_F (MLOffloadingService, registerModel)
   if (root_path == NULL)
     return;
 
-  g_autofree gchar *test_model = g_build_filename (root_path, "tests",
-      "test_models", "models", "mobilenet_v1_1.0_224_quant.tflite", NULL);
-  EXPECT_TRUE (g_file_test (test_model, G_FILE_TEST_EXISTS));
+  g_autofree gchar *model_dir
+      = g_build_filename (root_path, "tests", "test_models", "models", NULL);
+  g_autofree gchar *test_model
+      = g_build_filename (model_dir, "mobilenet_v1_1.0_224_quant.tflite", NULL);
 
   g_autofree gchar *contents = NULL;
   gsize len = 0;
@@ -350,6 +351,9 @@ TEST_F (MLOffloadingService, registerModel)
   status = ml_service_set_event_cb (server_h, _ml_service_event_cb, &test_data);
   EXPECT_EQ (status, ML_ERROR_NONE);
 
+  status = ml_service_offloading_set_information (server_h, "path", model_dir);
+  EXPECT_EQ (status, ML_ERROR_NONE);
+
   status = ml_tensors_info_create (&in_info);
   EXPECT_EQ (ML_ERROR_NONE, status);
   ml_tensors_info_set_count (in_info, 1);