class TestPoolStatusCache(SubManFixture):
"""
Class for testing PoolStatusCache
"""
def setUp(self):
super(TestPoolStatusCache, self).setUp()
self.pool_status_cache = PoolStatusCache()
self.pool_status_cache.write_cache = Mock()
def test_load_data(self):
cached = {
'pools': {
'pool1': 'Pool 1',
'pool2': 'Pool 2'
},
'tags': ['p1', 'p2']
}
mock_file = Mock()
mock_file.read = Mock(return_value=json.dumps(cached))
data = self.pool_status_cache._load_data(mock_file)
self.assertEqual(data, cached)
def test_load_from_server(self):
uep = Mock()
dummy_pools = {
'pools': {
'pool1': 'Pool 1',
'pool2': 'Pool 2'
},
'tags': ['p1', 'p2']
}
uep.getEntitlementList = Mock(return_value=dummy_pools)
self.pool_status_cache.read_status(uep, "THISISAUUID")
self.assertEqual(dummy_pools, self.pool_status_cache.server_status)
def setUp(self):
super(TestPoolStatusCache, self).setUp()
self.pool_status_cache = PoolStatusCache()
self.pool_status_cache.write_cache = Mock()