| 18 | |
| 19 | |
| 20 | class FakeSession: |
| 21 | def __init__(self, responses=None, get_response=None): |
| 22 | self.responses = list(responses or []) |
| 23 | self.get_response = get_response or FakeResponse(content=b"url-image") |
| 24 | self.posts = [] |
| 25 | self.gets = [] |
| 26 | |
| 27 | def post(self, url, headers=None, json=None, timeout=None): |
| 28 | self.posts.append({"url": url, "json": dict(json or {})}) |
| 29 | return self.responses.pop(0) |
| 30 | |
| 31 | def get(self, url, timeout=None): |
| 32 | self.gets.append(url) |
| 33 | return self.get_response |
| 34 | |
| 35 | |
| 36 | def encoded_payload(data=b"image-bytes"): |
no outgoing calls