Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_resources.py: 37%
30 statements
« prev ^ index » next coverage.py v7.2.2, created at 2023-03-26 07:30 +0000
« prev ^ index » next coverage.py v7.2.2, created at 2023-03-26 07:30 +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 mock
31with atheris.instrument_imports():
32 from google.cloud.logging_v2.handlers import _monitored_resources
35global_fdp = None
36def mock_retrieve_metadata_server(endpoint):
37 """Mock for retrieve_metadata_server"""
38 if global_fdp is None:
39 return None
40 if global_fdp.ConsumeIntInRange(1, 10) < 3:
41 return None
42 return global_fdp.ConsumeUnicodeNoSurrogates(30)
44def TestInput(data):
45 global global_fdp
46 global_fdp = atheris.FuzzedDataProvider(data)
48 # Mock the metadata server to avoid connections. The
49 # retrieve_metadata_server will return fuzzer-seeded data.
50 patch = mock.patch(
51 "google.cloud.logging_v2.handlers._monitored_resources.retrieve_metadata_server",
52 wraps=mock_retrieve_metadata_server,
53 )
54 # TODO: randomise relevant environment variables.
55 with patch:
56 _monitored_resources.detect_resource()
59def main():
60 atheris.Setup(sys.argv, TestInput, enable_python_coverage=True)
61 atheris.Fuzz()
63if __name__ == "__main__":
64 main()