diff options
author | Jakub Szafrański <s@samu.pl> | 2019-02-12 12:51:29 +0100 |
---|---|---|
committer | Jakub Szafrański <s@samu.pl> | 2019-02-12 13:00:55 +0100 |
commit | 2439d0a8d7e48750d7b73a0e8da9a3571f6617cc (patch) | |
tree | 542f2591eb9a0caa5f78a3a620110d4a0475dab1 /tests | |
parent | bad41222379102e3f18f6f2f7be3ee608de6fbff (diff) | |
download | python-memcached-2439d0a8d7e48750d7b73a0e8da9a3571f6617cc.tar.gz |
Add support for default value in get
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_memcache.py | 14 |
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): |