@@ -50,25 +50,23 @@ class HostMemoryAccessor : public MemoryAccessor {
5050};
5151
5252// ipcTestParams:
53- // pool_ops, pool_params, provider_ops, provider_params, memoryAccessor, free_not_supp
54- // free_not_supp (bool) - provider does not support the free() op
53+ // pool_ops, pool_params, provider_ops, provider_params, memoryAccessor
5554using ipcTestParams =
5655 std::tuple<umf_memory_pool_ops_t *, void *, umf_memory_provider_ops_t *,
57- void *, MemoryAccessor *, bool >;
56+ void *, MemoryAccessor *>;
5857
5958struct umfIpcTest : umf_test::test,
6059 ::testing::WithParamInterface<ipcTestParams> {
6160 umfIpcTest () {}
6261 void SetUp () override {
6362 test::SetUp ();
64- auto [pool_ops, pool_params, provider_ops, provider_params, accessor,
65- free_not_supp] = this ->GetParam ();
63+ auto [pool_ops, pool_params, provider_ops, provider_params, accessor] =
64+ this ->GetParam ();
6665 poolOps = pool_ops;
6766 poolParams = pool_params;
6867 providerOps = provider_ops;
6968 providerParams = provider_params;
7069 memAccessor = accessor;
71- freeNotSupported = free_not_supp;
7270 }
7371
7472 void TearDown () override { test::TearDown (); }
@@ -143,18 +141,8 @@ struct umfIpcTest : umf_test::test,
143141 void *poolParams = nullptr ;
144142 umf_memory_provider_ops_t *providerOps = nullptr ;
145143 void *providerParams = nullptr ;
146- bool freeNotSupported = false ;
147144};
148145
149- static inline umf_result_t
150- get_umf_result_of_free (bool freeNotSupported, umf_result_t expected_result) {
151- if (freeNotSupported) {
152- return UMF_RESULT_ERROR_NOT_SUPPORTED;
153- }
154-
155- return expected_result;
156- }
157-
158146TEST_P (umfIpcTest, GetIPCHandleSize) {
159147 size_t size = 0 ;
160148 umf::pool_unique_handle_t pool = makePool ();
@@ -196,8 +184,7 @@ TEST_P(umfIpcTest, GetIPCHandleInvalidArgs) {
196184 EXPECT_EQ (ret, UMF_RESULT_ERROR_INVALID_ARGUMENT);
197185
198186 ret = umfFree (ptr);
199- EXPECT_EQ (ret,
200- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
187+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
201188}
202189
203190TEST_P (umfIpcTest, CloseIPCHandleInvalidPtr) {
@@ -258,8 +245,7 @@ TEST_P(umfIpcTest, BasicFlow) {
258245 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
259246
260247 ret = umfPoolFree (pool.get (), ptr);
261- EXPECT_EQ (ret,
262- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
248+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
263249
264250 pool.reset (nullptr );
265251 EXPECT_EQ (stat.getCount , 1 );
@@ -322,8 +308,7 @@ TEST_P(umfIpcTest, GetPoolByOpenedHandle) {
322308
323309 for (size_t i = 0 ; i < NUM_ALLOCS; ++i) {
324310 umf_result_t ret = umfFree (ptrs[i]);
325- EXPECT_EQ (ret,
326- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
311+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
327312 }
328313}
329314
@@ -349,8 +334,7 @@ TEST_P(umfIpcTest, AllocFreeAllocTest) {
349334 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
350335
351336 ret = umfPoolFree (pool.get (), ptr);
352- EXPECT_EQ (ret,
353- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
337+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
354338
355339 ptr = umfPoolMalloc (pool.get (), SIZE);
356340 ASSERT_NE (ptr, nullptr );
@@ -372,8 +356,7 @@ TEST_P(umfIpcTest, AllocFreeAllocTest) {
372356 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
373357
374358 ret = umfPoolFree (pool.get (), ptr);
375- EXPECT_EQ (ret,
376- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
359+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
377360
378361 pool.reset (nullptr );
379362 EXPECT_EQ (stat.getCount , stat.putCount );
@@ -424,8 +407,7 @@ TEST_P(umfIpcTest, openInTwoPools) {
424407 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
425408
426409 ret = umfPoolFree (pool1.get (), ptr);
427- EXPECT_EQ (ret,
428- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
410+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
429411
430412 pool1.reset (nullptr );
431413 pool2.reset (nullptr );
@@ -476,8 +458,7 @@ TEST_P(umfIpcTest, ConcurrentGetPutHandles) {
476458
477459 for (void *ptr : ptrs) {
478460 umf_result_t ret = umfPoolFree (pool.get (), ptr);
479- EXPECT_EQ (ret,
480- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
461+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
481462 }
482463
483464 pool.reset (nullptr );
@@ -539,8 +520,7 @@ TEST_P(umfIpcTest, ConcurrentOpenCloseHandles) {
539520
540521 for (void *ptr : ptrs) {
541522 umf_result_t ret = umfPoolFree (pool.get (), ptr);
542- EXPECT_EQ (ret,
543- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
523+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
544524 }
545525
546526 pool.reset (nullptr );
0 commit comments