"""Common assertions.""" from project_manager.constant import header_const def assert_header(expected_header, actual_header): """Assert content type and correlation id in headers are same.""" expected_content_type = expected_header[header_const.CONTENT_TYPE] expected_correlation_id = expected_header[header_const.CORRELATION_ID] actual_content_type = actual_header[header_const.CONTENT_TYPE] actual_correlation_id = actual_header[header_const.CORRELATION_ID] assert expected_content_type == actual_content_type assert expected_correlation_id == actual_correlation_id