Exemplo n.º 1
0
    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())