update tests

microsoft · Oct 25, 2024 · cd268b2 · cd268b2
1 parent cb686e5
commit cd268b2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/c_api_tests.cpp b/test/c_api_tests.cpp
@@ -285,11 +285,11 @@ TEST(CAPITests, GetOutputCAPI) {
   generator->GenerateNextToken();
 }
 
-TEST(CAPITests, SetUnsetTerminate) {
+TEST(CAPITests, SetTerminate) {
 #if TEST_PHI2
 
   auto GeneratorSetTerminateCall = [](OgaGenerator* generator) {
-    generator->SetUnsetTerminate(true);
+    generator->SetTerminate();
   };
 
   auto GenerateOutput = [](OgaGenerator* generator, std::unique_ptr<OgaTokenizerStream> tokenizer_stream) {
@@ -326,7 +326,7 @@ TEST(CAPITests, SetUnsetTerminate) {
     th.join();  // Wait for each thread to finish
   }
   EXPECT_EQ(generator->IsSessionTerminated(), true);
-  generator->SetUnsetTerminate(false);
+  generator->UnsetTerminate();
   EXPECT_EQ(generator->IsSessionTerminated(), false);
 #endif
 }