def test_capture_size_limit_not_reached_when_no_file(self): cache = StreamingCache(cache_dir=None) self.assertFalse(cache.exists('my_label')) ie.current_env().set_cache_manager(cache, 'dummy pipeline') limiter = capture_limiters.SizeLimiter(1) self.assertFalse(limiter.is_triggered())
def limiters(self): # type: () -> List[capture_limiters.Limiter] if self._test_limiters: return self._test_limiters return [ capture_limiters.SizeLimiter(self._capture_size_limit), capture_limiters.DurationLimiter(self._capture_duration) ]
def test_capture_size_limit_not_reached_when_file_size_under_limit(self): ib.options.capture_size_limit = 100 cache = StreamingCache(cache_dir=None) # Build a sink object to track the label as a capture in the test. cache.sink(['my_label'], is_capture=True) cache.write([TestStreamFileRecord()], 'my_label') self.assertTrue(cache.exists('my_label')) ie.current_env().set_cache_manager(cache, 'dummy pipeline') limiter = capture_limiters.SizeLimiter(ib.options.capture_size_limit) self.assertFalse(limiter.is_triggered())
def test_capture_size_limit_reached_when_file_size_above_limit(self): ib.options.capture_size_limit = 1 cache = StreamingCache(cache_dir=None) cache.sink(['my_label'], is_capture=True) cache.write([ TestStreamFileRecord(recorded_event=TestStreamPayload.Event( element_event=TestStreamPayload.Event.AddElements(elements=[ TestStreamPayload.TimestampedElement( encoded_element=coders.FastPrimitivesCoder().encode( 'a'), timestamp=0) ]))) ], 'my_label') self.assertTrue(cache.exists('my_label')) ie.current_env().set_cache_manager(cache) limiter = capture_limiters.SizeLimiter(1) self.assertTrue(limiter.is_triggered())
def test_capture_size_limit_not_reached_when_no_cache(self): self.assertEqual(len(ie.current_env()._cache_managers), 0) limiter = capture_limiters.SizeLimiter(1) self.assertFalse(limiter.is_triggered())