Skip to content

Commit 0d8dfe2

Browse files
committed
Remove EXPECT_DEATH as the preparation of public absl
1 parent 468d83d commit 0d8dfe2

File tree

5 files changed

+5
-70
lines changed

5 files changed

+5
-70
lines changed

src/common.h

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,6 @@ namespace error {
8585

8686
void Abort();
8787
void Exit(int code);
88-
void SetTestCounter(int c);
89-
void ResetTestMode();
90-
bool GetTestCounter();
9188

9289
class Die {
9390
public:

src/error.cc

Lines changed: 5 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -29,29 +29,18 @@ ABSL_FLAG(int32_t, minloglevel, 0,
2929

3030
namespace sentencepiece {
3131
namespace error {
32-
int gTestCounter = 0;
3332

3433
void Abort() {
35-
if (GetTestCounter() == 1) {
36-
SetTestCounter(2);
37-
} else {
38-
std::cerr << "Program terminated with an unrecoverable error." << std::endl;
39-
ShutdownLibrary();
40-
exit(-1);
41-
}
34+
std::cerr << "Program terminated with an unrecoverable error." << std::endl;
35+
ShutdownLibrary();
36+
std::exit(-1);
4237
}
4338

4439
void Exit(int code) {
45-
if (GetTestCounter() == 1) {
46-
SetTestCounter(2);
47-
} else {
48-
ShutdownLibrary();
49-
exit(code);
50-
}
40+
ShutdownLibrary();
41+
std::exit(code);
5142
}
5243

53-
void SetTestCounter(int c) { gTestCounter = c; }
54-
bool GetTestCounter() { return gTestCounter; }
5544
} // namespace error
5645

5746
namespace util {

src/init_test.cc

Lines changed: 0 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -89,36 +89,6 @@ TEST(FlagsTest, ParseCommandLineFlagsTest3) {
8989
EXPECT_EQ(1, argc);
9090
}
9191

92-
TEST(FlagsTest, ParseCommandLineFlagsHelpTest) {
93-
const char *kFlags[] = {"program", "--help"};
94-
int argc = std::size(kFlags);
95-
char **argv = const_cast<char **>(kFlags);
96-
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), "");
97-
absl::SetFlag(&FLAGS_help, false);
98-
}
99-
100-
TEST(FlagsTest, ParseCommandLineFlagsVersionTest) {
101-
const char *kFlags[] = {"program", "--version"};
102-
int argc = std::size(kFlags);
103-
char **argv = const_cast<char **>(kFlags);
104-
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), "");
105-
absl::SetFlag(&FLAGS_version, false);
106-
}
107-
108-
TEST(FlagsTest, ParseCommandLineFlagsUnknownTest) {
109-
const char *kFlags[] = {"program", "--foo"};
110-
int argc = std::size(kFlags);
111-
char **argv = const_cast<char **>(kFlags);
112-
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), "");
113-
}
114-
115-
TEST(FlagsTest, ParseCommandLineFlagsInvalidBoolTest) {
116-
const char *kFlags[] = {"program", "--bool_f=X"};
117-
int argc = std::size(kFlags);
118-
char **argv = const_cast<char **>(kFlags);
119-
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), "");
120-
}
121-
12292
TEST(FlagsTest, ParseCommandLineFlagsEmptyStringArgs) {
12393
const char *kFlags[] = {"program", "--string_f="};
12494
int argc = std::size(kFlags);
@@ -137,12 +107,6 @@ TEST(FlagsTest, ParseCommandLineFlagsEmptyBoolArgs) {
137107
EXPECT_TRUE(absl::GetFlag(FLAGS_bool_f));
138108
}
139109

140-
TEST(FlagsTest, ParseCommandLineFlagsEmptyIntArgs) {
141-
const char *kFlags[] = {"program", "--int32_f"};
142-
int argc = std::size(kFlags);
143-
char **argv = const_cast<char **>(kFlags);
144-
EXPECT_DEATH(ParseCommandLineFlags(kFlags[0], &argc, &argv), );
145-
}
146110
} // namespace absl
147111

148112
#endif // _USE_EXTERNAL_ABSL

src/testharness.h

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -136,13 +136,6 @@ class Tester {
136136
#define EXPECT_OK(c) EXPECT_EQ(c, ::sentencepiece::util::OkStatus())
137137
#define EXPECT_NOT_OK(c) EXPECT_NE(c, ::sentencepiece::util::OkStatus())
138138

139-
#define EXPECT_DEATH(statement, condition) \
140-
{ \
141-
sentencepiece::error::SetTestCounter(1); \
142-
statement; \
143-
sentencepiece::error::SetTestCounter(0); \
144-
};
145-
146139
#define ASSERT_TRUE EXPECT_TRUE
147140
#define ASSERT_FALSE EXPECT_FALSE
148141
#define ASSERT_STREQ EXPECT_STREQ
@@ -154,7 +147,6 @@ class Tester {
154147
#define ASSERT_LT EXPECT_LT
155148
#define ASSERT_NEAR EXPECT_NEAR
156149
#define ASSERT_NOT_OK EXPECT_NOT_OK
157-
#define ASSERT_DEATH ASSERT_DEATH
158150

159151
template <typename T>
160152
class TestWithParam {

src/util_test.cc

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -313,19 +313,12 @@ TEST(UtilTest, MapUtilTest) {
313313

314314
EXPECT_EQ("A", port::FindOrDie(kMap, "a"));
315315
EXPECT_EQ("B", port::FindOrDie(kMap, "b"));
316-
EXPECT_DEATH(port::FindOrDie(kMap, "x"), "");
317316

318317
EXPECT_EQ("A", port::FindWithDefault(kMap, "a", "x"));
319318
EXPECT_EQ("B", port::FindWithDefault(kMap, "b", "x"));
320319
EXPECT_EQ("x", port::FindWithDefault(kMap, "d", "x"));
321320

322321
EXPECT_EQ("A", port::FindOrDie(kMap, "a"));
323-
EXPECT_DEATH(port::FindOrDie(kMap, "d"), "");
324-
}
325-
326-
TEST(UtilTest, MapUtilVecTest) {
327-
const std::map<std::vector<int>, std::string> kMap = {{{0, 1}, "A"}};
328-
EXPECT_DEATH(port::FindOrDie(kMap, {0, 2}), "");
329322
}
330323

331324
TEST(UtilTest, InputOutputBufferTest) {

0 commit comments

Comments
 (0)