Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_fileio.py: 47%
36 statements
« prev ^ index » next coverage.py v7.3.1, created at 2023-09-25 06:17 +0000
« prev ^ index » next coverage.py v7.3.1, created at 2023-09-25 06:17 +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.storage.fileio import BlobReader
35global_data = None
36def TestOneInput(data):
37 global_data = data
38 fdp = atheris.FuzzedDataProvider(data)
40 blob = mock.Mock()
41 def read_fuzz_data(start=0, end=None, **_):
42 return global_data[start:end]
44 blob.download_as_bytes = mock.Mock(side_effect=read_fuzz_data)
45 blob.size = len(data)
46 blob.chunk_size = None
47 download_kwargs = {"if_metageneration_match": 1}
48 reader = BlobReader(blob, **download_kwargs)
49 try:
50 reader.read(fdp.ConsumeIntInRange(0, len(data)*2))
51 except UnicodeDecodeError:
52 return
54 reader.seek(
55 pos = fdp.ConsumeIntInRange(0, len(data)*2),
56 whence = fdp.ConsumeIntInRange(0, 2)
57 )
60def main():
61 atheris.instrument_all()
62 atheris.Setup(sys.argv, TestOneInput, enable_python_coverage=True)
63 atheris.Fuzz()
65if __name__ == "__main__":
66 main()