| 124 | |
| 125 | @contextlib.contextmanager |
| 126 | def mock_retry_now(self, now): |
| 127 | if sys.version_info[0] > 3 or sys.version_info[0] == 3 and sys.version_info[1] >= 11: |
| 128 | attr = "github.GithubRetry.GithubRetry._GithubRetry__datetime" |
| 129 | else: |
| 130 | attr = "github.GithubRetry._GithubRetry__datetime" |
| 131 | with mock.patch(attr) as dt: |
| 132 | dt.now = lambda tz=None: datetime.fromtimestamp(now, tz=tz) |
| 133 | dt.fromtimestamp = datetime.fromtimestamp |
| 134 | yield |
| 135 | |
| 136 | def test_primary_rate_error_with_reset(self): |
| 137 | retry = github.GithubRetry(total=3) |