def test_camera_stream_frames_trunc_left():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
frames = []
for data, frame in generate_frames('hkffkffhkff'):
encoder.frame = frame
if frame.complete:
frames.append(frame)
stream.write(data)
del frames[:3]
# As we've gotten rid of the start of the stream we need to re-calc the
# video and split sizes in the comparison meta-data
sizes = accumulate(f.frame_size for f in frames)
frames = [
PiVideoFrame(
f.index,
f.frame_type,
f.frame_size,
size,
size,
f.timestamp,
f.complete
)
for f, size in zip(frames, sizes)
]
assert stream.getvalue() == b'fkkffhkkff'
assert list(stream.frames) == frames
assert list(reversed(stream.frames)) == frames[::-1]
def initAndRun(self,
cameraStream: cams.PiCameraCircularIO,
file=None,
parent=None):
self.logger = self.logger.getChild(
"RTSP") if file is None else self.logger.getChild("Record")
self._myParent = parent
self._redirectPackages(cameraStream, file)
self.name = "cam_RTSP_queue" if file is None else "cam_file_queue"
self.setDaemon(False)
self.start()
if parent is not None:
try:
with self.lock:
with cameraStream.lock:
save_pos = cameraStream.tell()
try:
pos = cameraStream._find_all(
camf.PiVideoFrameType.sps_header)
if pos is not None:
cameraStream.seek(pos)
while True:
buf = cameraStream.read1()
if not buf:
break
self._queue.put(buf)
finally:
cameraStream.seek(save_pos)
cameraStream.clear()
except:
self.logger.exception("Konnte anfang nicht speichern")
def test_camera_stream_init():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
with pytest.raises(ValueError):
PiCameraCircularIO(None, size=1)
with pytest.raises(ValueError):
PiCameraCircularIO(camera)
with pytest.raises(ValueError):
PiCameraCircularIO(camera, size=1, seconds=1)
assert PiCameraCircularIO(camera, seconds=1, bitrate=1024).size == 1024 // 8
def test_camera_stream_frames():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
frames = []
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
if frame.complete:
frames.append(frame)
stream.write(data)
assert stream.getvalue() == b'hkkffkkff'
assert list(stream.frames) == frames
assert list(reversed(stream.frames)) == frames[::-1]
def test_camera_stream_copy_bad():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
with pytest.raises(ValueError):
PiCameraCircularIO(camera, size=10).copy_to('foo',
size=1000,
seconds=10)
def test_camera_stream_copy_all():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
stream.write(data)
output = io.BytesIO()
stream.copy_to(output)
assert output.getvalue() == b'hkkffkkff'
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
stream.write(data)
assert stream.getvalue() == b'fhkkffkkff'
output = io.BytesIO()
stream.copy_to(output)
assert output.getvalue() == b'hkkffkkff'
def test_camera_stream_copy_seconds():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
stream.write(data)
output = io.BytesIO()
stream.copy_to(output, seconds=1)
assert output.getvalue() == b''
stream.copy_to(output, seconds=10)
assert output.getvalue() == b'hkkffkkff'
def test_camera_stream_clear():
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
stream.write(data)
stream.clear()
assert stream.getvalue() == b''
assert list(stream.frames) == []
assert list(reversed(stream.frames)) == []
def test_camera_stream_frames_trunc_right():
# We don't officially support this but the code should work if entire
# frames are truncated (without leaving partial frame data) which is what
# we're testing for here (of course, the resulting H.264 stream won't be
# valid, but we're not testing that...)
camera = mock.Mock()
encoder = mock.Mock()
camera._encoders = {1: encoder}
stream = PiCameraCircularIO(camera, size=10)
frames = []
for data, frame in generate_frames('hkffkff'):
encoder.frame = frame
if frame.complete:
frames.append(frame)
stream.write(data)
stream.seek(7)
stream.truncate()
del frames[-2:]
assert stream.getvalue() == b'hkkffkk'
assert list(stream.frames) == frames
assert list(reversed(stream.frames)) == frames[::-1]
