summaryrefslogtreecommitdiff
path: root/tests/test_memcache.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_memcache.py')
-rw-r--r--tests/test_memcache.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/test_memcache.py b/tests/test_memcache.py
index 40b6524..4cdf7a6 100644
--- a/tests/test_memcache.py
+++ b/tests/test_memcache.py
@@ -51,6 +51,20 @@ class TestMemcache(unittest.TestCase):
self.assertEqual(result, True)
self.assertEqual(self.mc.get("long"), None)
+ def test_default(self):
+ key = "default"
+ default = object()
+ result = self.mc.get(key, default=default)
+ self.assertEqual(result, default)
+
+ self.mc.set("default", None)
+ result = self.mc.get(key, default=default)
+ self.assertIsNone(result)
+
+ self.mc.set("default", 123)
+ result = self.mc.get(key, default=default)
+ self.assertEqual(result, 123)
+
@mock.patch.object(_Host, 'send_cmd')
@mock.patch.object(_Host, 'readline')
def test_touch(self, mock_readline, mock_send_cmd):