Coverage for /pythoncovmergedfiles/medio/medio/src/pdfminer.six/fuzzing/fuzzed_data_provider.py: 77%

1import io 

2 

3from atheris import FuzzedDataProvider 

4 

5 

6class PdfminerFuzzedDataProvider(FuzzedDataProvider): # type: ignore[misc] 

7 def ConsumeRandomBytes(self) -> bytes: 

8 int_range = self.ConsumeIntInRange(0, self.remaining_bytes()) 

9 return bytes(self.ConsumeBytes(int_range)) 

10 

11 def ConsumeRandomString(self) -> str: 

12 int_range = self.ConsumeIntInRange(0, self.remaining_bytes()) 

13 return str(self.ConsumeUnicodeNoSurrogates(int_range)) 

14 

15 def ConsumeRemainingString(self) -> str: 

16 return str(self.ConsumeUnicodeNoSurrogates(self.remaining_bytes())) 

17 

18 def ConsumeRemainingBytes(self) -> bytes: 

19 return bytes(self.ConsumeBytes(self.remaining_bytes())) 

20 

21 def ConsumeMemoryFile(self, all_data: bool = False) -> io.BytesIO: 

22 if all_data: 

23 return io.BytesIO(self.ConsumeRemainingBytes()) 

24 else: 

25 return io.BytesIO(self.ConsumeRandomBytes()) 

26 

27 def ConsumeOptionalIntList( 

28 self, 

29 max_count: int, 

30 min: int, 

31 max: int, 

32 ) -> list[int] | None: 

33 if self.ConsumeBool(): 

34 count = self.ConsumeIntInRange(0, max_count) 

35 return [int(i) for i in self.ConsumeIntListInRange(count, min, max)] 

36 return None