Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_handlers.py: 41%
34 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 logging
30with atheris.instrument_imports():
31 import google.cloud.logging_v2.handlers.handlers as handlers
32 import google.cloud.logging_v2.handlers.structured_log as log
34def TestInput(data):
35 fdp = atheris.FuzzedDataProvider(data)
37 logname = fdp.ConsumeString(100)
38 message = fdp.ConsumeString(100)
39 record = logging.LogRecord(
40 logname, logging.INFO, None, None, message, None, None
41 )
43 handlers.CloudLoggingFilter._infer_source_location(record)
44 filter = handlers.CloudLoggingFilter(record)
46 handler = log.StructuredLogHandler(
47 labels={
48 fdp.ConsumeString(10):fdp.ConsumeString(20),
49 fdp.ConsumeString(10):fdp.ConsumeString(20)
50 },
51 project_id=fdp.ConsumeString(100)
52 )
53 try:
54 handler.format(record)
55 handlers._format_and_parse_message(record, handler)
56 handlers.setup_logging(handler)
57 except ValueError as e:
58 if "Formatting field not found in record" not in str(e):
59 raise e
60def main():
61 atheris.Setup(sys.argv, TestInput, enable_python_coverage=True)
62 atheris.Fuzz()
64if __name__ == "__main__":
65 main()