Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_entries.py: 30%
40 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
30import google.cloud.logging_v2.entries as entries
31from google.cloud.logging_v2.client import Client
32from google.cloud.logging_v2.resource import Resource
34def create_dummy_log_entry(fdp):
35 return entries.LogEntry(
36 log_name=fdp.ConsumeString(20),
37 labels={
38 fdp.ConsumeString(10):fdp.ConsumeString(20),
39 fdp.ConsumeString(10):fdp.ConsumeString(20)
40 },
41 insert_id=fdp.ConsumeString(20),
42 timestamp=fdp.ConsumeString(20),
43 resource=Resource(type="global", labels={}),
44 trace=fdp.ConsumeString(20),
45 span_id=fdp.ConsumeString(20),
46 trace_sampled=fdp.ConsumeBool(),
47 source_location=LogEntrySourceLocation(
48 file=fdp.ConsumeString(20),
49 line=fdp.ConsumeString(20),
50 function=fdp.ConsumeString(20)
51 ),
52 operation=LogEntryOperation(
53 id=fdp.ConsumeString(20),
54 producer=fdp.ConsumeString(20),
55 first=fdp.ConsumeBool(),
56 last=fdp.ConsumeBool()
57 )
58 )
60def TestInput(data):
61 if len(data) < 1:
62 return
64 fdp = atheris.FuzzedDataProvider(data)
66 try:
67 entries._int_or_none(fdp.ConsumeInt(100))
68 entries.logger_name_from_path(fdp.ConsumeString(100))
69 entries.logger_name_from_path(fdp.ConsumeString(100),fdp.ConsumeString(50))
71 log_entry = create_dummy_log_entry(fdp)
72 log_entry.to_api_repr()
74 TextEntry(log_entry).to_api_repr()
75 StructEntry(log_entry).to_api_repr()
77 protobuf_entry = ProtobufEntry(log_entry)
78 protobuf_entry.payload_pb()
79 protobuf_entry.payload_json()
80 protobuf_entry.to_api_repr()
81 except ValueError as e:
82 if "did not match expected pattern" not in str(e):
83 raise e
85def main():
86 atheris.instrument_all()
87 atheris.Setup(sys.argv, TestInput, enable_python_coverage=True)
88 atheris.Fuzz()
90if __name__ == "__main__":
91 main()