v0.5.2-pre.003

ks-wallet/tests/public_api.rs

@@ -1,12 +1,41 @@
 // file: ks-wallet/tests/public_api.rs
-// version: 1
+// version: 3
 
 //! External crate-root contract tests for the `ks-wallet` public API.
 
-fn write_native_identification_fixture(path: &std::path::Path) {
+fn write_native_wallet_fixture(
+    path: &std::path::Path,
+    alias: &str,
+    marker: u8,
+) -> std::string::String {
+    let public_key = solana_pubkey::Pubkey::new_from_array([marker; 32]);
+    let alias_bytes = alias.as_bytes();
+    let alias_length =
+        u8::try_from(alias_bytes.len()).unwrap_or_else(|_| panic!("fixture alias must fit u8"));
+    let header_length = 72_u16 + u16::from(alias_length);
+    let ciphertext = vec![marker.wrapping_add(3); 80];
     let mut bytes = std::vec::Vec::new();
     bytes.extend_from_slice(b"KSWALLET");
     bytes.extend_from_slice(&1_u16.to_le_bytes());
+    bytes.extend_from_slice(&header_length.to_le_bytes());
+    bytes.extend_from_slice(&0_u16.to_le_bytes());
+    bytes.push(1);
+    bytes.push(0x13);
+    bytes.push(1);
+    bytes.push(16);
+    bytes.push(24);
+    bytes.push(alias_length);
+    bytes.extend_from_slice(&[0_u8; 4]);
+    bytes.extend_from_slice(&65_536_u32.to_le_bytes());
+    bytes.extend_from_slice(&3_u32.to_le_bytes());
+    bytes.extend_from_slice(&4_u32.to_le_bytes());
+    bytes.extend_from_slice(&80_u32.to_le_bytes());
+    let public_key_bytes = public_key.to_bytes();
+    bytes.extend_from_slice(&public_key_bytes);
+    bytes.extend_from_slice(alias_bytes);
+    bytes.extend_from_slice(&[marker.wrapping_add(1); 16]);
+    bytes.extend_from_slice(&[marker.wrapping_add(2); 24]);
+    bytes.extend_from_slice(ciphertext.as_slice());
     std::fs::write(path, bytes)
         .unwrap_or_else(|error| panic!("native fixture must be writable: {error}"));
     #[cfg(unix)]
@@ -15,6 +44,7 @@ fn write_native_identification_fixture(path: &std::path::Path) {
         std::fs::set_permissions(path, std::fs::Permissions::from_mode(0o600))
             .unwrap_or_else(|error| panic!("native fixture permissions must be set: {error}"));
     }
+    return public_key.to_string();
 }
 
 #[tokio::test]
@@ -22,16 +52,27 @@ async fn crate_root_exposes_multi_wallet_discovery_without_path_disclosure() {
     assert_eq!(ks_wallet::KSWALLET_FILE_EXTENSION, "kswallet");
     let directory = tempfile::tempdir()
         .unwrap_or_else(|error| panic!("temporary directory must exist: {error}"));
+    #[cfg(unix)]
+    {
+        use std::os::unix::fs::PermissionsExt; // rust-rules: trait-import
+        std::fs::set_permissions(directory.path(), std::fs::Permissions::from_mode(0o700))
+            .unwrap_or_else(|error| {
+                panic!("wallet fixture directory permissions must be set: {error}")
+            });
+    }
     let manager = ks_wallet::WalletManager::new(directory.path())
         .unwrap_or_else(|error| panic!("unexpected manager error: {error}"));
     let path = directory.path().join("public-api.kswallet");
-    write_native_identification_fixture(&path);
+    let expected_public_key = write_native_wallet_fixture(&path, "public-api", 37);
     let handles = manager
         .scan()
         .await
         .unwrap_or_else(|error| panic!("scan must succeed: {error}"));
     assert_eq!(handles.len(), 1);
     assert_eq!(handles[0].alias().as_str(), "public-api");
+    assert_eq!(handles[0].public_key(), expected_public_key);
+    assert_eq!(handles[0].format_version(), 1);
+    assert_eq!(handles[0].identity().persistence, ks_wallet::WalletPersistence::Persistent);
     assert!(!format!("{:?}", handles[0]).contains(directory.path().to_string_lossy().as_ref()));
 }