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())
