From d1514a5b173deeb35a4748f9b9ba857c16d07753 Mon Sep 17 00:00:00 2001
From: Talha <tamjad91@gmail.com>
Date: Fri, 31 Jul 2026 12:57:42 +0500
Subject: [PATCH] test(desktop): cover stable package index

---
 desktop/src-tauri/src/lib.rs | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/desktop/src-tauri/src/lib.rs b/desktop/src-tauri/src/lib.rs
index 2a62b76..c2bb1e2 100644
--- a/desktop/src-tauri/src/lib.rs
+++ b/desktop/src-tauri/src/lib.rs
@@ -1630,10 +1630,11 @@ mod tests {
     }
 
     #[test]
-    fn prerelease_package_and_dependencies_use_separate_indexes() {
+    fn package_and_dependencies_use_channel_specific_indexes() {
         let manifest = manifest().expect("manifest");
         let python = Path::new("managed-python");
         let constraints = Path::new("runtime-constraints.txt");
+        let selected_package_index = package_index(&manifest.package_version);
         let acquisition = package_acquisition_arguments(&manifest, python);
         let dependencies = dependency_installation_arguments(
             &manifest,
@@ -1644,17 +1645,21 @@ mod tests {
             true,
         );
 
-        assert_eq!(
-            package_index(&manifest.package_version),
+        let expected_package_index = if manifest.package_version.contains('-') {
             "https://test.pypi.org/simple"
-        );
+        } else {
+            "https://pypi.org/simple"
+        };
+
+        assert_eq!(selected_package_index, expected_package_index);
+        assert_eq!(package_index("0.3.0-b.1"), "https://test.pypi.org/simple");
         assert_eq!(package_index("0.3.0"), "https://pypi.org/simple");
         assert_eq!(manifest.dependency_index, "https://pypi.org/simple");
         assert!(acquisition.iter().any(|item| item == "--no-deps"));
         assert!(
             acquisition
                 .iter()
-                .any(|item| item == "https://test.pypi.org/simple")
+                .any(|item| item == selected_package_index)
         );
         assert!(!dependencies.iter().any(|item| item == "--no-deps"));
         assert!(