Skip to content

Commit b87c93d

Browse files
committed
catch exceptions from google test
1 parent c77d3a1 commit b87c93d

File tree

2 files changed

+16
-6
lines changed

2 files changed

+16
-6
lines changed

test/pools/pool_residency.cpp

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,12 @@ TEST_F(PoolResidencyTestFixture,
176176
}
177177

178178
int main(int argc, char **argv) {
179-
InitGoogleTest(&argc, argv);
180-
AddGlobalTestEnvironment(new MockedLevelZeroTestEnvironment);
181-
return RUN_ALL_TESTS();
179+
try {
180+
InitGoogleTest(&argc, argv);
181+
AddGlobalTestEnvironment(new MockedLevelZeroTestEnvironment);
182+
return RUN_ALL_TESTS();
183+
} catch (...) {
184+
std::cerr << "Exception occurred." << std::endl;
185+
return 1;
186+
}
182187
}

test/providers/provider_level_zero_residency.cpp

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,12 @@ TEST_F(LevelZeroResidencyTestFixture, removeDeviceTwiceShouldFail) {
8383
}
8484

8585
int main(int argc, char **argv) {
86-
InitGoogleTest(&argc, argv);
87-
AddGlobalTestEnvironment(new MockedLevelZeroTestEnvironment);
88-
return RUN_ALL_TESTS();
86+
try {
87+
InitGoogleTest(&argc, argv);
88+
AddGlobalTestEnvironment(new MockedLevelZeroTestEnvironment);
89+
return RUN_ALL_TESTS();
90+
} catch (...) {
91+
std::cerr << "Exception occurred." << std::endl;
92+
return 1;
93+
}
8994
}

0 commit comments

Comments
 (0)