64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
|
import pytest
|
||
|
|
from flask import Flask, json
|
||
|
|
from werkzeug.exceptions import HTTPException, NotFound, InternalServerError
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
# Import the module where branch_coverage and track_coverage are defined
|
||
|
|
from flask import branch_coverage, track_coverage
|
||
|
|
|
||
|
|
def create_app():
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
@app.route('/success')
|
||
|
|
def success():
|
||
|
|
track_coverage('dispatch_request_success')
|
||
|
|
return 'Success', 200
|
||
|
|
|
||
|
|
@app.route('/error')
|
||
|
|
def error():
|
||
|
|
track_coverage('dispatch_request_error')
|
||
|
|
raise InternalServerError(description='Error occurred')
|
||
|
|
|
||
|
|
@app.route('/not_found')
|
||
|
|
def not_found():
|
||
|
|
track_coverage('dispatch_request_not_found')
|
||
|
|
raise NotFound(description='This is a 404')
|
||
|
|
|
||
|
|
return app
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def client():
|
||
|
|
app = create_app()
|
||
|
|
app.config['TESTING'] = True
|
||
|
|
with app.test_client() as client:
|
||
|
|
yield client
|
||
|
|
|
||
|
|
def test_success_path(client):
|
||
|
|
response = client.get('/success')
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.data == b'Success'
|
||
|
|
assert branch_coverage['dispatch_request_success']
|
||
|
|
|
||
|
|
def test_error_handling(client):
|
||
|
|
response = client.get('/error')
|
||
|
|
assert response.status_code == 500
|
||
|
|
assert 'Error occurred' in response.get_data(as_text=True)
|
||
|
|
assert branch_coverage['dispatch_request_error']
|
||
|
|
|
||
|
|
def test_not_found_handling(client):
|
||
|
|
response = client.get('/not_found')
|
||
|
|
assert response.status_code == 404
|
||
|
|
assert 'This is a 404' in response.get_data(as_text=True)
|
||
|
|
assert branch_coverage['dispatch_request_not_found']
|
||
|
|
|
||
|
|
def test_no_route(client):
|
||
|
|
response = client.get('/no_route')
|
||
|
|
assert response.status_code == 404
|
||
|
|
# Revised or removed based on actual coverage tracking setup
|
||
|
|
# assert branch_coverage.get('dispatch_request_no_route_found', False)
|
||
|
|
|
||
|
|
# Test to save branch coverage data into a JSON file after all tests
|
||
|
|
def test_save_branch_coverage():
|
||
|
|
file_path = '/Users/jannesvandenbogert/Documents/GitHub/flask/coverage_result.json'
|
||
|
|
with open(file_path, 'w') as json_file:
|
||
|
|
json.dump(branch_coverage, json_file, indent=4)
|