def test_update_cache_rrule(self, process_store, CachedUpdatePersister, config): rrule_info = { 'freq': 'DAILY', 'dtstart': '2014-10-30T13:21:18', } impl = MagicMock() persister = CachedUpdatePersister(impl, update_cache_rrule=rrule_info) persister.initialize_component(config) assert persister.read() is impl.read.return_value assert process_store['model'] is impl.read.return_value assert impl.read.call_count == 1
def test_dont_cache(self, process_store, CachedUpdatePersister, config): config['__mode__'] = 'fit' rrule_info = { 'freq': 'DAILY', 'dtstart': '2014-10-30T13:21:18', } impl = MagicMock() persister = CachedUpdatePersister(impl, update_cache_rrule=rrule_info) persister.initialize_component(config) assert persister.read() is impl.read.return_value assert len(process_store) == 0 assert persister.thread is None
def persister(self, CachedUpdatePersister, config): persister = CachedUpdatePersister(MagicMock()) persister.initialize_component(config) return persister