From 876c53d3a308c85743275271a5e934f2c0df88c0 Mon Sep 17 00:00:00 2001
From: beagold <86345081+beagold@users.noreply.github.com>
Date: Sun, 7 Apr 2024 19:29:27 +0200
Subject: [PATCH] Add ability to edit own user banner (#1871)

---
 changes/1871.feature.md        |  1 +
 hikari/api/rest.py             |  4 ++++
 hikari/impl/rest.py            |  8 ++++++++
 tests/hikari/impl/test_rest.py | 24 ++++++++++++++++++++++++
 4 files changed, 37 insertions(+)
 create mode 100644 changes/1871.feature.md

diff --git a/changes/1871.feature.md b/changes/1871.feature.md
new file mode 100644
index 0000000000..8f1e2d8955
--- /dev/null
+++ b/changes/1871.feature.md
@@ -0,0 +1 @@
+Add ability to edit own user banner
diff --git a/hikari/api/rest.py b/hikari/api/rest.py
index 4c79af053b..d6d89909bf 100644
--- a/hikari/api/rest.py
+++ b/hikari/api/rest.py
@@ -2567,6 +2567,7 @@ async def edit_my_user(
         *,
         username: undefined.UndefinedOr[str] = undefined.UNDEFINED,
         avatar: undefined.UndefinedNoneOr[files.Resourceish] = undefined.UNDEFINED,
+        banner: undefined.UndefinedNoneOr[files.Resourceish] = undefined.UNDEFINED,
     ) -> users.OwnUser:
         """Edit the token's associated user.
 
@@ -2577,6 +2578,9 @@ async def edit_my_user(
         avatar : undefined.UndefinedNoneOr[hikari.files.Resourceish]
             If provided, the new avatar. If [`None`][],
             the avatar will be removed.
+        banner : undefined.UndefinedNoneOr[hikari.files.Resourceish]
+            If provided, the new banner. If [`None`][],
+            the banner will be removed.
 
         Returns
         -------
diff --git a/hikari/impl/rest.py b/hikari/impl/rest.py
index f7320e10fc..f4db3d94f5 100644
--- a/hikari/impl/rest.py
+++ b/hikari/impl/rest.py
@@ -2062,6 +2062,7 @@ async def edit_my_user(
         *,
         username: undefined.UndefinedOr[str] = undefined.UNDEFINED,
         avatar: undefined.UndefinedNoneOr[files.Resourceish] = undefined.UNDEFINED,
+        banner: undefined.UndefinedNoneOr[files.Resourceish] = undefined.UNDEFINED,
     ) -> users.OwnUser:
         route = routes.PATCH_MY_USER.compile()
         body = data_binding.JSONObjectBuilder()
@@ -2074,6 +2075,13 @@ async def edit_my_user(
             async with avatar_resource.stream(executor=self._executor) as stream:
                 body.put("avatar", await stream.data_uri())
 
+        if banner is None:
+            body.put("banner", None)
+        elif banner is not undefined.UNDEFINED:
+            banner_resource = files.ensure_resource(banner)
+            async with banner_resource.stream(executor=self._executor) as stream:
+                body.put("banner", await stream.data_uri())
+
         response = await self._request(route, json=body)
         assert isinstance(response, dict)
         return self._entity_factory.deserialize_my_user(response)
diff --git a/tests/hikari/impl/test_rest.py b/tests/hikari/impl/test_rest.py
index e479af2380..a781f1f6ca 100644
--- a/tests/hikari/impl/test_rest.py
+++ b/tests/hikari/impl/test_rest.py
@@ -3519,6 +3519,30 @@ async def test_edit_my_user_when_avatar_is_file(self, rest_client, file_resource
         rest_client._request.assert_awaited_once_with(expected_route, json=expected_json)
         rest_client._entity_factory.deserialize_my_user.assert_called_once_with({"id": "123"})
 
+    async def test_edit_my_user_when_banner_is_None(self, rest_client):
+        user = StubModel(123)
+        expected_route = routes.PATCH_MY_USER.compile()
+        expected_json = {"username": "new username", "banner": None}
+        rest_client._request = mock.AsyncMock(return_value={"id": "123"})
+        rest_client._entity_factory.deserialize_my_user = mock.Mock(return_value=user)
+
+        assert await rest_client.edit_my_user(username="new username", banner=None) is user
+
+        rest_client._request.assert_awaited_once_with(expected_route, json=expected_json)
+        rest_client._entity_factory.deserialize_my_user.assert_called_once_with({"id": "123"})
+
+    async def test_edit_my_user_when_banner_is_file(self, rest_client, file_resource_patch):
+        user = StubModel(123)
+        expected_route = routes.PATCH_MY_USER.compile()
+        expected_json = {"username": "new username", "banner": "some data"}
+        rest_client._request = mock.AsyncMock(return_value={"id": "123"})
+        rest_client._entity_factory.deserialize_my_user = mock.Mock(return_value=user)
+
+        assert await rest_client.edit_my_user(username="new username", banner="somebanner.png") is user
+
+        rest_client._request.assert_awaited_once_with(expected_route, json=expected_json)
+        rest_client._entity_factory.deserialize_my_user.assert_called_once_with({"id": "123"})
+
     async def test_fetch_my_connections(self, rest_client):
         connection1 = StubModel(123)
         connection2 = StubModel(456)