Add validation tests for ANN_setLoopTimeout

Also adds mStarted validation.

Fix: 149819396
Test: NNT_static
Change-Id: I2f947a4ead527b7cff6bf93439c3295e59626f7a

1 files changed, 9 insertions, 0 deletions

diff --git a/nn/runtime/test/TestValidation.cpp b/nn/runtime/test/TestValidation.cpp
index 9e1f910ce..93570f4e1 100644
--- a/nn/runtime/test/TestValidation.cpp
+++ b/nn/runtime/test/TestValidation.cpp
@@ -1089,6 +1089,10 @@ TEST_F(ValidationTestCompilation, ExecutionUsability) {
         auto testTooLate = [this, execution, &in0, &out0, memory] {
             // Try a bunch of things that are impermissible if the execution has started.
 
+            // Set loop timeout.
+            ASSERT_EQ(ANeuralNetworksExecution_setLoopTimeout(execution, kShortWaitInNanoseconds),
+                      ANEURALNETWORKS_BAD_STATE);
+
             // Set inputs and outputs.
             ASSERT_EQ(ANeuralNetworksExecution_setInput(execution, 0, nullptr, &in0, sizeof(in0)),
                       ANEURALNETWORKS_BAD_STATE);
@@ -1175,6 +1179,11 @@ TEST_F(ValidationTestCompilation, ExecutionUsability) {
     }
 }
 
+TEST_F(ValidationTestExecution, SetLoopTimeout) {
+    EXPECT_EQ(ANeuralNetworksExecution_setLoopTimeout(nullptr, kShortWaitInNanoseconds),
+              ANEURALNETWORKS_UNEXPECTED_NULL);
+}
+
 TEST_F(ValidationTestExecution, SetInput) {
     char buffer[20];
     EXPECT_EQ(ANeuralNetworksExecution_setInput(nullptr, 0, nullptr, buffer, sizeof(float)),