@@ -47,25 +47,23 @@ class HostMemoryAccessor : public MemoryAccessor {
4747};
4848
4949// ipcTestParams:
50- // pool_ops, pool_params, provider_ops, provider_params, memoryAccessor, free_not_supp
51- // free_not_supp (bool) - provider does not support the free() op
50+ // pool_ops, pool_params, provider_ops, provider_params, memoryAccessor
5251using ipcTestParams =
5352 std::tuple<umf_memory_pool_ops_t *, void *, umf_memory_provider_ops_t *,
54- void *, MemoryAccessor *, bool >;
53+ void *, MemoryAccessor *>;
5554
5655struct umfIpcTest : umf_test::test,
5756 ::testing::WithParamInterface<ipcTestParams> {
5857 umfIpcTest () {}
5958 void SetUp () override {
6059 test::SetUp ();
61- auto [pool_ops, pool_params, provider_ops, provider_params, accessor,
62- free_not_supp] = this ->GetParam ();
60+ auto [pool_ops, pool_params, provider_ops, provider_params, accessor] =
61+ this ->GetParam ();
6362 poolOps = pool_ops;
6463 poolParams = pool_params;
6564 providerOps = provider_ops;
6665 providerParams = provider_params;
6766 memAccessor = accessor;
68- freeNotSupported = free_not_supp;
6967 }
7068
7169 void TearDown () override { test::TearDown (); }
@@ -124,18 +122,8 @@ struct umfIpcTest : umf_test::test,
124122 void *poolParams = nullptr ;
125123 umf_memory_provider_ops_t *providerOps = nullptr ;
126124 void *providerParams = nullptr ;
127- bool freeNotSupported = false ;
128125};
129126
130- static inline umf_result_t
131- get_umf_result_of_free (bool freeNotSupported, umf_result_t expected_result) {
132- if (freeNotSupported) {
133- return UMF_RESULT_ERROR_NOT_SUPPORTED;
134- }
135-
136- return expected_result;
137- }
138-
139127TEST_P (umfIpcTest, GetIPCHandleSize) {
140128 size_t size = 0 ;
141129 umf::pool_unique_handle_t pool = makePool ();
@@ -177,8 +165,7 @@ TEST_P(umfIpcTest, GetIPCHandleInvalidArgs) {
177165 EXPECT_EQ (ret, UMF_RESULT_ERROR_INVALID_ARGUMENT);
178166
179167 ret = umfFree (ptr);
180- EXPECT_EQ (ret,
181- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
168+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
182169}
183170
184171TEST_P (umfIpcTest, CloseIPCHandleInvalidPtr) {
@@ -239,8 +226,7 @@ TEST_P(umfIpcTest, BasicFlow) {
239226 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
240227
241228 ret = umfPoolFree (pool.get (), ptr);
242- EXPECT_EQ (ret,
243- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
229+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
244230
245231 pool.reset (nullptr );
246232 EXPECT_EQ (stat.getCount , 1 );
@@ -303,8 +289,7 @@ TEST_P(umfIpcTest, GetPoolByOpenedHandle) {
303289
304290 for (size_t i = 0 ; i < NUM_ALLOCS; ++i) {
305291 umf_result_t ret = umfFree (ptrs[i]);
306- EXPECT_EQ (ret,
307- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
292+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
308293 }
309294}
310295
@@ -330,8 +315,7 @@ TEST_P(umfIpcTest, AllocFreeAllocTest) {
330315 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
331316
332317 ret = umfPoolFree (pool.get (), ptr);
333- EXPECT_EQ (ret,
334- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
318+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
335319
336320 ptr = umfPoolMalloc (pool.get (), SIZE);
337321 ASSERT_NE (ptr, nullptr );
@@ -353,8 +337,7 @@ TEST_P(umfIpcTest, AllocFreeAllocTest) {
353337 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
354338
355339 ret = umfPoolFree (pool.get (), ptr);
356- EXPECT_EQ (ret,
357- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
340+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
358341
359342 pool.reset (nullptr );
360343 EXPECT_EQ (stat.getCount , stat.putCount );
@@ -405,8 +388,7 @@ TEST_P(umfIpcTest, openInTwoPools) {
405388 EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
406389
407390 ret = umfPoolFree (pool1.get (), ptr);
408- EXPECT_EQ (ret,
409- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
391+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
410392
411393 pool1.reset (nullptr );
412394 pool2.reset (nullptr );
@@ -457,8 +439,7 @@ TEST_P(umfIpcTest, ConcurrentGetPutHandles) {
457439
458440 for (void *ptr : ptrs) {
459441 umf_result_t ret = umfPoolFree (pool.get (), ptr);
460- EXPECT_EQ (ret,
461- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
442+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
462443 }
463444
464445 pool.reset (nullptr );
@@ -520,8 +501,7 @@ TEST_P(umfIpcTest, ConcurrentOpenCloseHandles) {
520501
521502 for (void *ptr : ptrs) {
522503 umf_result_t ret = umfPoolFree (pool.get (), ptr);
523- EXPECT_EQ (ret,
524- get_umf_result_of_free (freeNotSupported, UMF_RESULT_SUCCESS));
504+ EXPECT_EQ (ret, UMF_RESULT_SUCCESS);
525505 }
526506
527507 pool.reset (nullptr );
0 commit comments