| # Copyright 2018 The LUCI Authors. All rights reserved. |
| # Use of this source code is governed under the Apache License, Version 2.0 |
| # that can be found in the LICENSE file. |
| |
| """Allows test-repeatable access to a random UUID.""" |
| |
| from __future__ import annotations |
| |
| import uuid |
| |
| from recipe_engine import recipe_api |
| |
| class UuidApi(recipe_api.RecipeApi): |
| def __init__(self, **kwargs): |
| super().__init__(**kwargs) |
| self._fake_time: int | None = None |
| self._fake_step: int | None = None |
| if self._test_data.enabled: |
| self._fake_uuid = self._test_data.get('seed', 4916) |
| self._fake_step = self._test_data.get('step', 3) |
| |
| def random(self) -> str: |
| """Returns a random UUID string.""" |
| if self._test_data.enabled: |
| self._fake_uuid += self._fake_step |
| return str(uuid.UUID(int=self._fake_uuid)) |
| return str(uuid.uuid4()) # pragma: no cover |
