Move context cache tests to their proper location

tests/unit/utils/test_cache.py

@@ -14,11 +14,15 @@ import tempfile
 import shutil
 
 # Import Salt Testing libs
-from tests.support.unit import TestCase
+from tests.support.unit import TestCase, skipIf
+from tests.support.mock import NO_MOCK, NO_MOCK_REASON
 
 # Import salt libs
 import salt.config
 import salt.loader
+import salt.payload
+import salt.utils.data
+import salt.utils.files
 import salt.utils.cache as cache
 
 
@@ -93,3 +97,67 @@ class CacheContextTestCase(TestCase):
 
         self.assertEqual(cache_test_func()['called'], 0)
         self.assertEqual(cache_test_func()['called'], 1)
+
+
+__context__ = {'a': 'b'}
+__opts__ = {'cachedir': '/tmp'}
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class ContextCacheTest(TestCase):
+    '''
+    Test case for salt.utils.cache.ContextCache
+    '''
+    def setUp(self):
+        '''
+        Clear the cache before every test
+        '''
+        context_dir = os.path.join(__opts__['cachedir'], 'context')
+        if os.path.isdir(context_dir):
+            shutil.rmtree(context_dir)
+
+    def test_set_cache(self):
+        '''
+        Tests to ensure the cache is written correctly
+        '''
+        @cache.context_cache
+        def _test_set_cache():
+            '''
+            This will inherit globals from the test module itself.
+            Normally these are injected by the salt loader [salt.loader]
+            '''
+            pass
+
+        _test_set_cache()
+
+        target_cache_file = os.path.join(__opts__['cachedir'], 'context', '{0}.p'.format(__name__))
+        self.assertTrue(os.path.isfile(target_cache_file), 'Context cache did not write cache file')
+
+        # Test manual de-serialize
+        with salt.utils.files.fopen(target_cache_file, 'rb') as fp_:
+            target_cache_data = salt.utils.data.decode(salt.payload.Serial(__opts__).load(fp_))
+        self.assertDictEqual(__context__, target_cache_data)
+
+        # Test cache de-serialize
+        cc = cache.ContextCache(__opts__, __name__)
+        retrieved_cache = cc.get_cache_context()
+        self.assertDictEqual(retrieved_cache, __context__)
+
+    def test_refill_cache(self):
+        '''
+        Tests to ensure that the context cache can rehydrate a wrapped function
+        '''
+        # First populate the cache
+        @cache.context_cache
+        def _test_set_cache():
+            pass
+        _test_set_cache()
+
+        # Then try to rehydate a func
+        @cache.context_cache
+        def _test_refill_cache(comparison_context):
+            self.assertEqual(__context__, comparison_context)
+
+        global __context__
+        __context__ = {}
+        _test_refill_cache({'a': 'b'})  # Compare to the context before it was emptied