diff options
author | John L. Villalovos <john@sodarock.com> | 2021-11-07 14:33:39 -0800 |
---|---|---|
committer | John L. Villalovos <john@sodarock.com> | 2021-11-15 14:31:12 -0800 |
commit | 46773a82565cef231dc3391c12f296ac307cb95c (patch) | |
tree | 32cc9473ec28cc3d10be3baff28a74e8a17ac718 /gitlab/v4/objects/clusters.py | |
parent | 94feb8a5534d43a464b717275846faa75783427e (diff) | |
download | gitlab-46773a82565cef231dc3391c12f296ac307cb95c.tar.gz |
chore: ensure get() methods have correct type-hintsjlvillal/mypy_ensure_type_hints
Fix classes which don't have correct 'get()' methods for classes
derived from GetMixin.
Add a unit test which verifies that classes have the correct return
type in their 'get()' method.
Diffstat (limited to 'gitlab/v4/objects/clusters.py')
-rw-r--r-- | gitlab/v4/objects/clusters.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/gitlab/v4/objects/clusters.py b/gitlab/v4/objects/clusters.py index 4821b70..5491654 100644 --- a/gitlab/v4/objects/clusters.py +++ b/gitlab/v4/objects/clusters.py @@ -1,4 +1,4 @@ -from typing import Any, cast, Dict, Optional +from typing import Any, cast, Dict, Optional, Union from gitlab import exceptions as exc from gitlab.base import RequiredOptional, RESTManager, RESTObject @@ -57,6 +57,11 @@ class GroupClusterManager(CRUDMixin, RESTManager): path = f"{self.path}/user" return cast(GroupCluster, CreateMixin.create(self, data, path=path, **kwargs)) + def get( + self, id: Union[str, int], lazy: bool = False, **kwargs: Any + ) -> GroupCluster: + return cast(GroupCluster, super().get(id=id, lazy=lazy, **kwargs)) + class ProjectCluster(SaveMixin, ObjectDeleteMixin, RESTObject): pass @@ -102,3 +107,8 @@ class ProjectClusterManager(CRUDMixin, RESTManager): """ path = f"{self.path}/user" return cast(ProjectCluster, CreateMixin.create(self, data, path=path, **kwargs)) + + def get( + self, id: Union[str, int], lazy: bool = False, **kwargs: Any + ) -> ProjectCluster: + return cast(ProjectCluster, super().get(id=id, lazy=lazy, **kwargs)) |