Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_httpretty_e2e.py: 40%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1###### Coverage stub
2import atexit
3import coverage
4cov = coverage.coverage(data_file='.coverage', cover_pylib=True)
5cov.start()
6# Register an exist handler that will print coverage
7def exit_handler():
8 cov.stop()
9 cov.save()
10atexit.register(exit_handler)
11####### End of coverage stub
12#!/usr/bin/python3
13# Copyright 2023 Google LLC
14#
15# Licensed under the Apache License, Version 2.0 (the "License");
16# you may not use this file except in compliance with the License.
17# You may obtain a copy of the License at
18#
19# http://www.apache.org/licenses/LICENSE-2.0
20#
21# Unless required by applicable law or agreed to in writing, software
22# distributed under the License is distributed on an "AS IS" BASIS,
23# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
24# See the License for the specific language governing permissions and
25# limitations under the License.
27import sys
28import atheris
29import requests
30import httpretty
33def test_one(input_data):
34 fdp = atheris.FuzzedDataProvider(input_data)
36 # Enable httpretty
37 httpretty.enable(verbose=True, allow_net_connect=False)
39 # Create arguments
40 http_body = fdp.ConsumeUnicodeNoSurrogates(sys.maxsize)
41 if fdp.ConsumeBool():
42 header_dict = {
43 fdp.ConsumeUnicodeNoSurrogates(24): fdp.ConsumeUnicodeNoSurrogates(24),
44 fdp.ConsumeUnicodeNoSurrogates(24): fdp.ConsumeUnicodeNoSurrogates(24),
45 fdp.ConsumeUnicodeNoSurrogates(24): fdp.ConsumeUnicodeNoSurrogates(24),
46 fdp.ConsumeUnicodeNoSurrogates(24): fdp.ConsumeUnicodeNoSurrogates(24)
47 }
48 else:
49 header_dict = None
51 # Pass random arguments to register_uri and ensure we can get it
52 try:
53 httpretty.register_uri(httpretty.GET,
54 "http://fuzzing.com/",
55 body=http_body,
56 content_type=fdp.ConsumeUnicodeNoSurrogates(32),
57 adding_headers=header_dict)
58 response = requests.get('http://fuzzing.com')
59 except httpretty.HTTPrettyError:
60 pass
62 httpretty.reset()
65if __name__ == "__main__":
66 atheris.instrument_all()
67 atheris.Setup(sys.argv, test_one)
68 atheris.Fuzz()