chore: unit test

fhevm/tee_crypto_test.go

@@ -49,8 +49,6 @@ func TestTeeDecryptRun(t *testing.T) {
 
 func TestTeeOptimisticRequire(t *testing.T) {
 	signature := "teeOptimisticRequire(uint256)"
-	// func teeOptimisticRequireRun(environment EVMEnvironment, caller common.Address, addr common.Address, input []byte, readOnly bool, runSpan trace.Span) ([]byte, error) {
-
 	rapid.Check(t, func(t *rapid.T) {
 		testcases := []struct {
 			typ     tfhe.FheUintType
@@ -83,4 +81,37 @@ func TestTeeOptimisticRequire(t *testing.T) {
 			t.Fatalf("incorrect result, expected=%d, got=0", 1)
 		}
 	})
+
+	rapid.Check(t, func(t *rapid.T) {
+		testcases := []struct {
+			typ     tfhe.FheUintType
+			require uint64
+		}{
+			{tfhe.FheUint8, uint64(1)},
+			{tfhe.FheUint8, uint64(0)},
+		}
+
+		environment := newTestEVMEnvironment()
+		depth := 1
+		environment.depth = depth
+		for _, tc := range testcases {
+			addr := common.Address{}
+			readOnly := false
+			ct, err := importTeePlaintextToEVM(environment, depth, tc.require, tc.typ)
+			if err != nil {
+				t.Fatalf(err.Error())
+			}
+
+			input := toLibPrecompileInput(signature, false, ct.GetHash())
+			_, err = FheLibRun(environment, addr, addr, input, readOnly)
+			if err != nil {
+				t.Fatalf(err.Error())
+			}
+		}
+
+		optReqResult, _ := teeEvaluateRemainingOptimisticRequires(environment)
+		if optReqResult {
+			t.Fatalf("incorrect result, expected=%d, got=0", 1)
+		}
+	})
 }