def test_upload_failed_with_message_and_errors(requests_mock):
"""
Verifies that correct error message is logged on bad requests if 'message' and 'errors' is set.
:param requests_mock.Mocker requests_mock: The requests mocker
"""
requests_mock.post('http://valid_url',
json={
'message': 'Error message',
'errors': 'Additional errors'
},
status_code=int(HTTPStatus.BAD_REQUEST))
with LogCapture(names='berry_cam.threads.uploader') as log:
uploader = Uploader('http://valid_url', 'valid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
time.sleep(1)
uploader.stop()
uploader.join(1.5)
log.check_present(
('berry_cam.threads.uploader', 'ERROR',
'Upload failed. Status code: {0}, message: Error message'.format(
HTTPStatus.BAD_REQUEST)))
log.check_present(
('berry_cam.threads.uploader', 'ERROR', 'Additional errors'))
assert not uploader.is_alive()
def test_fast_fail_empty_queue():
"""
Verifies that the thread stops even if the upload queue is empty.
"""
uploader = Uploader('http://invalid_url', 'invalid_key', 2)
uploader.start()
uploader.stop()
uploader.join(1.5)
assert not uploader.is_alive()
def test_fast_fail_between_retries():
"""
Verifies that if the thread stops fast after .stop() is called.
"""
uploader = Uploader('http://invalid_url', 'invalid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
uploader.stop()
uploader.join(1.5)
assert not uploader.is_alive()
def test_upload_success(requests_mock):
"""
Verifies that correct data is sent on successful uploads.
:param requests_mock.Mocker requests_mock: The requests mocker
"""
requests_mock.post('http://valid_url/')
with LogCapture(names='berry_cam.threads.uploader') as log:
uploader = Uploader('http://valid_url', 'valid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
time.sleep(1)
uploader.stop()
uploader.join(1.5)
log.check_present(('berry_cam.threads.uploader', 'INFO',
'Upload succeeded after 0 tries'))
assert len(requests_mock.request_history) == 1
assert b'api_key' in requests_mock.request_history[0].body
assert b'valid_key' in requests_mock.request_history[0].body
assert not uploader.is_alive()
def test_upload_failed_with_other_failure(requests_mock):
"""
Verifies that correct error message is logged on other failure messages than 'message' or 'errors'.
:param requests_mock.Mocker requests_mock: The requests mocker
"""
requests_mock.post('http://valid_url',
json='Other failure',
status_code=int(HTTPStatus.BAD_REQUEST))
with LogCapture(names='berry_cam.threads.uploader') as log:
uploader = Uploader('http://valid_url', 'valid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
time.sleep(1)
uploader.stop()
uploader.join(1.5)
log.check_present(
('berry_cam.threads.uploader', 'ERROR',
"Upload failed. Status code: {0}, message: b'\"Other failure\"'".
format(HTTPStatus.BAD_REQUEST)))
assert not uploader.is_alive()
def test_invalid_url():
"""
Verifies that the thread stops after some time if an invalid sever url is given.
"""
with LogCapture(names='berry_cam.threads.uploader') as log:
uploader = Uploader('http://invalid_url', 'invalid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
uploader.join(3)
log.check_present((
'berry_cam.threads.uploader', 'ERROR',
'Uploader: Error while connecting to server. Retrying...'
), ('berry_cam.threads.uploader', 'ERROR',
'Uploader: Failed to upload file after 2 tries, giving up. Are you sure the server is up?'
))
assert not uploader.is_alive()
def test_valid_url_invalid_api_key(requests_mock):
"""
Verifies that the thread stops after some time if an invalid api key is given.
:param requests_mock.Mocker requests_mock: The requests mocker
"""
requests_mock.post('http://valid_url/', status_code=HTTPStatus.FORBIDDEN)
with LogCapture(names='berry_cam.threads.uploader') as log:
uploader = Uploader('http://valid_url', 'invalid_key', 2)
uploader.start()
uploader.upload_queue.put(TESTIMAGE)
uploader.join(3)
log.check_present(
('berry_cam.threads.uploader', 'ERROR',
'Uploader: Access denied. Please check your api key.'))
assert not uploader.is_alive()
# Open config file
yaml_path = os.path.join(os.path.dirname(__file__), 'conf.yaml')
with open(yaml_path) as config_file:
config = yaml.safe_load(config_file)
# Init heartbeat thread to notify the server that the camera is up
heartbeat = Heartbeat(
config['camera']['name'],
'{}/api/camera/'.format(config['image_server']['server_url']),
config['image_server']['api_key'],
config['image_server']['retry_count'])
threads.append(heartbeat)
# Init uploader thread that will upload new images
uploader = Uploader(
'{}/api/picture/'.format(config['image_server']['server_url']),
config['image_server']['api_key'],
config['image_server']['retry_count'])
threads.append(uploader)
# Check PIR config
pin_number_type = None
if config['pir']['number_type'] == "BCM":
pin_number_type = GPIO.BCM
elif config['pir']['number_type'] == "BOARD":
pin_number_type = GPIO.BOARD
else:
logging.error(
"Invalid pir number type found. Can be either BOARD or BCM."
"Instead, found %s", config['pir']['number_type'])
stop()