"""Module with tests for post_asset_final_status method.""" from unittest.mock import patch from src import app as index from src.common import lambda_exceptions import pytest post_asset_data = { 'filename': 'unique_filename.wav', 'status': 'encoding_processing', 'errors': None, 'timestamp': '2012-05-02T00:54:06.655Z', 'final_assets': [] } @patch('owsrequest.request.process') def test_post_asset_status_success(process): """Test for post_asset_status success.""" process.return_value.status_code = 200 process.return_value.json.return_value = {'message': 'ok'} result = index.post_asset_final_status(post_asset_data) assert result @patch('owsrequest.request.process') def test_post_asset_request_bad_status_failure(process): """Test for post_asset_status ows-assets 500 failure..""" process.return_value.status_code = 404 with pytest.raises(lambda_exceptions.PostAssetNotFoundError): index.post_asset_final_status(post_asset_data) @patch('owsrequest.request.process') def test_post_asset_status_request_exception_failure(process): """Test for post_asset_status ows-assets 500 failure..""" process.return_value.status_code = 400 with pytest.raises(lambda_exceptions.PostAssetError): index.post_asset_final_status(post_asset_data)