Add unit tests for the patch API

Signed-off-by: Rustin170506 <[email protected]>

Author: 0xPoe

src/tests/krate/publish/snapshots/all__krate__publish__links__crate_with_links_field-2.snap

@@ -45,6 +45,7 @@ expression: response.json()
     "readme_path": "/api/v1/crates/foo/1.0.0/readme",
     "rust_version": null,
     "updated_at": "[datetime]",
+    "yank_message": null,
     "yanked": false
   }
 }

src/tests/krate/publish/snapshots/all__krate__publish__manifest__boolean_readme-2.snap

@@ -45,6 +45,7 @@ expression: response.json()
     "readme_path": "/api/v1/crates/foo/1.0.0/readme",
     "rust_version": "1.69",
     "updated_at": "[datetime]",
+    "yank_message": null,
     "yanked": false
   }
 }

src/tests/krate/publish/snapshots/all__krate__publish__manifest__lib_and_bin_crate-2.snap

@@ -48,6 +48,7 @@ expression: response.json()
     "readme_path": "/api/v1/crates/foo/1.0.0/readme",
     "rust_version": null,
     "updated_at": "[datetime]",
+    "yank_message": null,
     "yanked": false
   }
 }

src/tests/krate/yanking.rs

@@ -220,3 +220,66 @@ async fn publish_after_yank_max_version() {
     let json = anon.show_crate("fyk_max").await;
     assert_eq!(json.krate.max_version, "2.0.0");
 }
+
+#[tokio::test(flavor = "multi_thread")]
+async fn patch_version_yank_unyank() {
+    let (_, anon, _, token) = TestApp::full().with_token();
+
+    // Upload a new crate
+    let crate_to_publish = PublishBuilder::new("patchable", "1.0.0");
+    token.publish_crate(crate_to_publish).await.good();
+
+    // Check initial state
+    let json = anon.show_version("patchable", "1.0.0").await;
+    assert!(!json.version.yanked);
+    assert_eq!(json.version.yank_message, None);
+
+    // Yank with message
+    token
+        .update_yank_status("patchable", "1.0.0", Some(true), Some("Yanking reason"))
+        .await
+        .good();
+
+    let json = anon.show_version("patchable", "1.0.0").await;
+    assert!(json.version.yanked);
+    assert_eq!(
+        json.version.yank_message,
+        Some("Yanking reason".to_string())
+    );
+
+    // Update yank message
+    token
+        .update_yank_status("patchable", "1.0.0", None, Some("Updated reason"))
+        .await
+        .good();
+
+    let json = anon.show_version("patchable", "1.0.0").await;
+    assert!(json.version.yanked);
+    assert_eq!(
+        json.version.yank_message,
+        Some("Updated reason".to_string())
+    );
+
+    // Unyank
+    token
+        .update_yank_status("patchable", "1.0.0", Some(false), None)
+        .await
+        .good();
+
+    let json = anon.show_version("patchable", "1.0.0").await;
+    assert!(!json.version.yanked);
+    assert_eq!(json.version.yank_message, None);
+
+    // Attempt to set yank message on unyanked version (should fail)
+    token
+        .update_yank_status("patchable", "1.0.0", None, Some("Invalid message"))
+        .await
+        .status()
+        .is_client_error();
+    // Attempt to unyank with message (should fail)
+    token
+        .update_yank_status("patchable", "1.0.0", Some(false), Some("Invalid message"))
+        .await
+        .status()
+        .is_client_error();
+}

src/tests/routes/crates/snapshots/all__routes__crates__read__show.snap

@@ -71,6 +71,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/foo_show/1.0.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     },
     {
@@ -103,6 +104,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/foo_show/0.5.1/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     },
     {
@@ -135,6 +137,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/foo_show/0.5.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]

src/tests/routes/crates/snapshots/all__routes__crates__reverse_dependencies__prerelease_versions_not_included_in_reverse_dependencies.snap

@@ -51,6 +51,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c3/1.0.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]

src/tests/routes/crates/snapshots/all__routes__crates__reverse_dependencies__reverse_dependencies.snap

@@ -51,6 +51,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c2/1.1.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]

src/tests/routes/crates/snapshots/all__routes__crates__reverse_dependencies__reverse_dependencies_includes_published_by_user_when_present.snap

@@ -63,6 +63,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c3/3.0.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     },
     {
@@ -89,6 +90,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c2/2.0.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]

src/tests/routes/crates/snapshots/all__routes__crates__reverse_dependencies__reverse_dependencies_query_supports_u64_version_number_parts.snap

@@ -51,6 +51,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c2/1.0.18446744073709551615/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]

src/tests/routes/crates/snapshots/all__routes__crates__reverse_dependencies__reverse_dependencies_when_old_version_doesnt_depend_but_new_does.snap

@@ -51,6 +51,7 @@ expression: response.json()
       "readme_path": "/api/v1/crates/c2/2.0.0/readme",
       "rust_version": null,
       "updated_at": "[datetime]",
+      "yank_message": null,
       "yanked": false
     }
   ]