From a9a7c84703eb7250012fd68af3e4ca77eae07029 Mon Sep 17 00:00:00 2001
From: "Chris St. Pierre" <chris.a.st.pierre@gmail.com>
Date: Wed, 14 Aug 2013 10:22:44 -0400
Subject: doc: minor fixes

---
 testsuite/Testsrc/Testlib/TestServer/TestCache.py | 54 +++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 testsuite/Testsrc/Testlib/TestServer/TestCache.py

(limited to 'testsuite/Testsrc')

diff --git a/testsuite/Testsrc/Testlib/TestServer/TestCache.py b/testsuite/Testsrc/Testlib/TestServer/TestCache.py
new file mode 100644
index 000000000..7c26e52b8
--- /dev/null
+++ b/testsuite/Testsrc/Testlib/TestServer/TestCache.py
@@ -0,0 +1,54 @@
+import os
+import sys
+
+# add all parent testsuite directories to sys.path to allow (most)
+# relative imports in python 2.4
+path = os.path.dirname(__file__)
+while path != "/":
+    if os.path.basename(path).lower().startswith("test"):
+        sys.path.append(path)
+    if os.path.basename(path) == "testsuite":
+        break
+    path = os.path.dirname(path)
+from common import *
+
+from Bcfg2.Server.Cache import *
+
+
+class TestCache(Bcfg2TestCase):
+    def test_cache(self):
+        md_cache = Cache("Metadata")
+        md_cache['foo.example.com'] = 'foo metadata'
+        md_cache['bar.example.com'] = 'bar metadata'
+        self.assertItemsEqual(list(iter(md_cache)),
+                              ["foo.example.com", "bar.example.com"])
+
+        probe_cache = Cache("Probes", "data")
+        probe_cache['foo.example.com'] = 'foo probe data'
+        probe_cache['bar.example.com'] = 'bar probe data'
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              ["foo.example.com", "bar.example.com"])
+
+        md_cache.expire("foo.example.com")
+        self.assertItemsEqual(list(iter(md_cache)), ["bar.example.com"])
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              ["foo.example.com", "bar.example.com"])
+
+        probe_cache.expire("bar.example.com")
+        self.assertItemsEqual(list(iter(md_cache)), ["bar.example.com"])
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              ["foo.example.com"])
+
+        probe_cache['bar.example.com'] = 'bar probe data'
+        self.assertItemsEqual(list(iter(md_cache)), ["bar.example.com"])
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              ["foo.example.com", "bar.example.com"])
+
+        expire("bar.example.com")
+        self.assertEqual(len(md_cache), 0)
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              ["foo.example.com"])
+
+        probe_cache2 = Cache("Probes", "data")
+        self.assertItemsEqual(list(iter(probe_cache)),
+                              list(iter(probe_cache2)))
-- 
cgit v1.2.3-1-g7c22