Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_helpers.py: 71%
51 statements
« prev ^ index » next coverage.py v7.3.2, created at 2023-12-08 06:40 +0000
« prev ^ index » next coverage.py v7.3.2, created at 2023-12-08 06:40 +0000
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 2022 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 atheris
28import sys
29import json
31# We instrument all functions when initiating atheris
32from google.cloud import _helpers
34def TestOneInput(data):
35 if len(data) < 100:
36 return
38 fdp = atheris.FuzzedDataProvider(data)
39 try:
40 loaded_data = json.loads(fdp.ConsumeString(300))
41 success = True
42 except Exception:
43 success = False
45 if success:
46 try:
47 _helpers._ensure_tuple_or_list(loaded_data)
48 except TypeError:
49 pass
51 try:
52 _helpers._bytes_to_unicode(data)
53 except ValueError:
54 pass
56 fdp = atheris.FuzzedDataProvider(data)
57 try:
58 _helpers._time_from_iso8601_time_naive(fdp.ConsumeString(100))
59 except ValueError:
60 pass
62 fdp = atheris.FuzzedDataProvider(data)
63 try:
64 _helpers._rfc3339_nanos_to_datetime(fdp.ConsumeString(100))
65 except ValueError:
66 pass
68 fdp = atheris.FuzzedDataProvider(data)
69 try:
70 _helpers._name_from_project_path(
71 fdp.ConsumeString(60),
72 fdp.ConsumeString(60),
73 fdp.ConsumeString(60)
74 )
75 except ValueError:
76 pass
78def main():
79 atheris.instrument_all()
80 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
81 atheris.Fuzz()
83if __name__ == "__main__":
84 main()