def test_set(self, mock_open, mock_cache_file, mock_cache_object, mock_flush_index, mock_datetime, mock_index): self.build_mock_index_and_datetime_now(mock_index, mock_datetime) cache = Cache('test') value = cache.set('key_obj', 'value_obj', as_fileobj=False) self.assertTrue(mock_flush_index.called) self.assertTrue(mock_cache_object.called) self.assertEqual(value, 'value_obj') self.assertEqual(cache.index['key_obj'], mock_datetime.now()) mock_flush_index.reset_mock() value = cache.set('key_file', 'value_file', as_fileobj=True) self.assertTrue(mock_flush_index.called) self.assertTrue(mock_cache_file.called) self.assertEqual(value, 'value_file') self.assertEqual(cache.index['key_file'], mock_datetime.now())