Coverage for /pythoncovmergedfiles/medio/medio/src/fuzz_validators.py: 60%
40 statements
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:32 +0000
« prev ^ index » next coverage.py v7.2.7, created at 2023-06-07 06:32 +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 2023 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.
26import sys
27import atheris
29import wtforms
32class FuzzField:
33 def __init__(self, text):
34 self.text = text
35 self.data = text
37 def gettext(self, string):
38 return self.text
41def TestOneInput(data):
42 fdp = atheris.FuzzedDataProvider(data)
44 try:
45 c1 = wtforms.validators.HostnameValidation()
46 c1(fdp.ConsumeUnicodeNoSurrogates(1024))
47 except (wtforms.validators.StopValidation,
48 wtforms.validators.ValidationError):
49 pass
51 try:
52 validator = wtforms.validators.URL()
53 validator(fdp.ConsumeUnicodeNoSurrogates(1024),
54 FuzzField(fdp.ConsumeUnicodeNoSurrogates(1024)))
55 except wtforms.validators.ValidationError:
56 pass
58 try:
59 validator = wtforms.validators.email()
60 validator(fdp.ConsumeUnicodeNoSurrogates(1024),
61 FuzzField(fdp.ConsumeUnicodeNoSurrogates(1024)))
62 except wtforms.validators.ValidationError:
63 pass
66def main():
67 atheris.instrument_all()
68 atheris.Setup(sys.argv, TestOneInput)
69 atheris.Fuzz()
72if __name__ == "__main__":
73 main()