Add test for send_reset_password_email

PriorLabs · May 19, 2024 · 7c383e5 · 7c383e5
1 parent 43d7ecc
commit 7c383e5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tabpfn_client/tests/unit/test_client.py b/tabpfn_client/tests/unit/test_client.py
@@ -97,6 +97,12 @@ def test_valid_auth_token(self, mock_server):
         mock_server.router.get(mock_server.endpoints.protected_root.path).respond(200)
         self.assertTrue(self.client.try_authenticate("true_token"))
 
+    @with_mock_server()
+    def test_send_reset_password_email(self, mock_server):
+        mock_server.router.post(mock_server.endpoints.send_reset_password_email.path).respond(
+            200, json={"message": "Password reset email sent!"})
+        self.assertEqual(self.client.send_reset_password_email("test"), (True, "Password reset email sent!"))
+
     @with_mock_server()
     def test_retrieve_greeting_messages(self, mock_server):
         mock_server.router.get(mock_server.endpoints.retrieve_greeting_messages.path).respond(