add get account keys test

integration_tests/tests/wallet_ffi.rs

@@ -7,10 +7,12 @@ use std::{
 use anyhow::Result;
 use integration_tests::{ACC_SENDER, TestContext};
 use log::info;
-use nssa::{Account, AccountId, program::Program};
+use nssa::{Account, AccountId, PublicKey, program::Program};
 use tempfile::tempdir;
 use wallet::WalletCore;
-use wallet_ffi::{FfiAccount, FfiAccountList, FfiBytes32, WalletHandle, error};
+use wallet_ffi::{
+    FfiAccount, FfiAccountList, FfiBytes32, FfiPublicAccountKey, WalletHandle, error,
+};
 
 unsafe extern "C" {
     fn wallet_ffi_create_new(
@@ -50,6 +52,12 @@ unsafe extern "C" {
     ) -> error::WalletFfiError;
 
     fn wallet_ffi_free_account_data(account: *mut FfiAccount);
+
+    fn wallet_ffi_get_public_account_key(
+        handle: *mut WalletHandle,
+        account_id: *const FfiBytes32,
+        out_public_key: *mut FfiPublicAccountKey,
+    ) -> error::WalletFfiError;
 }
 
 fn new_wallet_ffi_with_test_context_config(ctx: &TestContext) -> *mut WalletHandle {
@@ -329,3 +337,35 @@ fn test_wallet_ffi_get_account_public() -> Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn test_wallet_ffi_get_public_account_keys() -> Result<()> {
+    let ctx = TestContext::new_blocking()?;
+    let account_id: AccountId = ACC_SENDER.parse().unwrap();
+    let wallet_ffi_handle = new_wallet_ffi_with_test_context_config(&ctx);
+    let mut out_key = FfiPublicAccountKey::default();
+
+    let key: PublicKey = unsafe {
+        let ffi_account_id = FfiBytes32::from(&account_id);
+        let _result = wallet_ffi_get_public_account_key(
+            wallet_ffi_handle,
+            (&ffi_account_id) as *const FfiBytes32,
+            (&mut out_key) as *mut FfiPublicAccountKey,
+        );
+        (&out_key).try_into().unwrap()
+    };
+
+    let expected_key = {
+        let private_key = ctx
+            .wallet()
+            .get_account_public_signing_key(&account_id)
+            .unwrap();
+        PublicKey::new_from_private_key(private_key)
+    };
+
+    assert_eq!(key, expected_key);
+
+    info!("Successfully retrieved account key");
+
+    Ok(())
+}