mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-31 04:42:09 +00:00
42b27b9a9d
* Add isort * Apply isort on py files * Fix circular import * Fix format for notebooks * Fix format --------- Co-authored-by: Chi Wang <wang.chi@microsoft.com>
55 lines
1.8 KiB
Python
Executable File
55 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3 -m pytest
|
|
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from autogen.cache.disk_cache import DiskCache
|
|
|
|
|
|
class TestDiskCache(unittest.TestCase):
|
|
def setUp(self):
|
|
self.seed = "test_seed"
|
|
|
|
@patch("autogen.cache.disk_cache.diskcache.Cache", return_value=MagicMock())
|
|
def test_init(self, mock_cache):
|
|
cache = DiskCache(self.seed)
|
|
self.assertIsInstance(cache.cache, MagicMock)
|
|
mock_cache.assert_called_with(self.seed)
|
|
|
|
@patch("autogen.cache.disk_cache.diskcache.Cache", return_value=MagicMock())
|
|
def test_get(self, mock_cache):
|
|
key = "key"
|
|
value = "value"
|
|
cache = DiskCache(self.seed)
|
|
cache.cache.get.return_value = value
|
|
self.assertEqual(cache.get(key), value)
|
|
cache.cache.get.assert_called_with(key, None)
|
|
|
|
cache.cache.get.return_value = None
|
|
self.assertIsNone(cache.get(key, None))
|
|
|
|
@patch("autogen.cache.disk_cache.diskcache.Cache", return_value=MagicMock())
|
|
def test_set(self, mock_cache):
|
|
key = "key"
|
|
value = "value"
|
|
cache = DiskCache(self.seed)
|
|
cache.set(key, value)
|
|
cache.cache.set.assert_called_with(key, value)
|
|
|
|
@patch("autogen.cache.disk_cache.diskcache.Cache", return_value=MagicMock())
|
|
def test_context_manager(self, mock_cache):
|
|
with DiskCache(self.seed) as cache:
|
|
self.assertIsInstance(cache, DiskCache)
|
|
mock_cache_instance = cache.cache
|
|
mock_cache_instance.close.assert_called()
|
|
|
|
@patch("autogen.cache.disk_cache.diskcache.Cache", return_value=MagicMock())
|
|
def test_close(self, mock_cache):
|
|
cache = DiskCache(self.seed)
|
|
cache.close()
|
|
cache.cache.close.assert_called()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|