From 36e5368f3c68ede454bfc7f914c910babfbc804b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20N=C3=BA=C3=B1ez?= <david@nucypher.com>
Date: Fri, 23 Sep 2022 10:05:36 +0200
Subject: [PATCH] Add more basic unit tests for ReturnValueTest

---
 tests/unit/conditions/test_return_value.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/tests/unit/conditions/test_return_value.py b/tests/unit/conditions/test_return_value.py
index cfbb2eec8..8303d66db 100644
--- a/tests/unit/conditions/test_return_value.py
+++ b/tests/unit/conditions/test_return_value.py
@@ -4,11 +4,21 @@ import pytest
 from nucypher.policy.conditions.lingo import ReturnValueTest
 
 
-def test_return_result_test_simple():
+def test_return_value_test_integer():
     test = ReturnValueTest(comparator='>', value='0')
     assert test.eval('1')
     assert not test.eval('-1')
 
+    test = ReturnValueTest(comparator='>', value=0)
+    assert test.eval(1)
+    assert not test.eval(-1)
+
+
+def test_return_value_test_string():
+    test = ReturnValueTest(comparator='==', value='"foo"')
+    assert test.eval('"foo"')
+    assert not test.eval('"bar"')
+
 
 def test_return_value_sanitization():
     with pytest.raises(ValueError):