better error for setting url of unknown demo

src/client.rs

@@ -287,7 +287,8 @@ impl ApiClient {
         hash: [u8; 16],
         key: &str,
     ) -> Result<(), Error> {
-        self.client
+        let response = self
+            .client
             .post(self.url(format!("/demos/{}/url", demo_id))?)
             .form(&[
                 ("hash", hex::encode(hash).as_str()),
@@ -297,8 +298,13 @@ impl ApiClient {
                 ("key", key),
             ])
             .send()
-            .await?
+            .await?;
-            .error_for_status()?;
+
+        if response.status() == StatusCode::NOT_FOUND {
+            return Err(Error::DemoNotFound(demo_id));
+        }
+
+        response.error_for_status()?;
 
         Ok(())
     }

tests/tests.rs

@@ -223,6 +223,24 @@ async fn test_set_url_invalid_hash() {
     assert!(matches!(res.unwrap_err(), Error::HashMisMatch));
 }
 
+#[tokio::test]
+async fn test_set_url_unknown_demo() {
+    let client = test_client().await;
+
+    let res = client
+        .set_url(
+            99,
+            "tests",
+            "tests",
+            "http://example.com/tests",
+            [0; 16],
+            "edit",
+        )
+        .await;
+    dbg!(&res);
+    assert!(matches!(res.unwrap_err(), Error::DemoNotFound(99)));
+}
+
 #[tokio::test]
 async fn test_set_url() {
     let client = test_client().await;