chore: enable mypy check strict_equality

JohnVillalovos · JohnVillalovos · commit d8ba031b4c5b · 2022-07-14T19:04:27.000-07:00
Enable the `mypy` `strict_equality` check.
diff --git a/gitlab/v4/objects/users.py b/gitlab/v4/objects/users.py
@@ -163,7 +163,7 @@ class User(SaveMixin, ObjectDeleteMixin, RESTObject):
 
     @cli.register_custom_action("User")
     @exc.on_http_error(exc.GitlabBlockError)
-    def block(self, **kwargs: Any) -> Union[Dict[str, Any], requests.Response]:
+    def block(self, **kwargs: Any) -> bool:
         """Block the user.
 
         Args:
@@ -177,10 +177,13 @@ def block(self, **kwargs: Any) -> Union[Dict[str, Any], requests.Response]:
             Whether the user status has been changed
         """
         path = f"/users/{self.encoded_id}/block"
-        server_data = self.manager.gitlab.http_post(path, **kwargs)
-        if server_data is True:
-            self._attrs["state"] = "blocked"
-        return server_data
+        # NOTE: Undocumented behavior of the GitLab API is that it returns a boolean
+        server_data = cast(bool, self.manager.gitlab.http_post(path, **kwargs))
+        if isinstance(server_data, bool):
+            if server_data is True:
+                self._attrs["state"] = "blocked"
+            return server_data
+        return False
 
     @cli.register_custom_action("User")
     @exc.on_http_error(exc.GitlabFollowError)
@@ -220,7 +223,7 @@ def unfollow(self, **kwargs: Any) -> Union[Dict[str, Any], requests.Response]:
 
     @cli.register_custom_action("User")
     @exc.on_http_error(exc.GitlabUnblockError)
-    def unblock(self, **kwargs: Any) -> Union[Dict[str, Any], requests.Response]:
+    def unblock(self, **kwargs: Any) -> bool:
         """Unblock the user.
 
         Args:
@@ -234,10 +237,13 @@ def unblock(self, **kwargs: Any) -> Union[Dict[str, Any], requests.Response]:
             Whether the user status has been changed
         """
         path = f"/users/{self.encoded_id}/unblock"
-        server_data = self.manager.gitlab.http_post(path, **kwargs)
-        if server_data is True:
-            self._attrs["state"] = "active"
-        return server_data
+        # NOTE: Undocumented behavior of the GitLab API is that it returns a boolean
+        server_data = cast(bool, self.manager.gitlab.http_post(path, **kwargs))
+        if isinstance(server_data, bool):
+            if server_data is True:
+                self._attrs["state"] = "active"
+            return server_data
+        return False
 
     @cli.register_custom_action("User")
     @exc.on_http_error(exc.GitlabDeactivateError)
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,6 +13,8 @@ disallow_incomplete_defs = true
 disallow_subclassing_any = true
 disallow_untyped_decorators = true
 disallow_untyped_defs = true
+no_implicit_reexport = true
+strict_equality = true
 warn_redundant_casts = true
 warn_unused_configs = true
 warn_unused_ignores = true
@@ -21,8 +23,6 @@ warn_unused_ignores = true
 # disallow_any_generics = true
 # disallow_untyped_calls = true
 # no_implicit_optional = true
-no_implicit_reexport = true
-# strict_equality = true
 # warn_return_any = true
 
 [[tool.mypy.overrides]] # Overrides for currently untyped modules
diff --git a/tests/functional/api/test_users.py b/tests/functional/api/test_users.py
@@ -28,14 +28,24 @@ def test_create_user(gl, fixture_dir):
 
 
 def test_block_user(gl, user):
-    user.block()
+    result = user.block()
+    assert result is True
     users = gl.users.list(blocked=True)
     assert user in users
 
-    user.unblock()
+    # block again
+    result = user.block()
+    assert result is False
+
+    result = user.unblock()
+    assert result is True
     users = gl.users.list(blocked=False)
     assert user in users
 
+    # unblock again
+    result = user.unblock()
+    assert result is False
+
 
 def test_ban_user(gl, user):
     user.ban()
