Add tests for get_or_set.

leighmcculloch · Aug 1, 2016 · 57e6a36 · 57e6a36
1 parent 7da60d3
commit 57e6a36
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/filecache_test.rb b/tests/filecache_test.rb
@@ -12,6 +12,25 @@ def test_basic
     f.set(KEY1, VALUE1)
     assert_equal(f.get(KEY1), VALUE1, "set/get")
 
+    f.delete(KEY1)
+    assert_nil(f.get(KEY1), "delete")
+
+    assert_equal(
+      f.get_or_set(KEY1) do
+        VALUE1
+      end,
+      VALUE1,
+      "set_or_get"
+    )
+
+    assert_equal(
+      f.get_or_set(KEY1) do
+        VALUE2
+      end,
+      VALUE1,
+      "set_or_get"
+    )
+
     f.delete(KEY1)
     assert_nil(f.get(KEY1), "delete")