From e94baa51530caf0ee27dfb4f12b222f6640befd3 Mon Sep 17 00:00:00 2001
From: YoussefAWasfy <youssef.w@affinidi.com>
Date: Thu, 26 Sep 2024 15:49:53 +0200
Subject: [PATCH] test: integration tests

---
 .../tests/integration_test.rs                 | 127 ++++++++++++++++++
 1 file changed, 127 insertions(+)
 create mode 100644 affinidi-did-resolver-cache-server/tests/integration_test.rs

diff --git a/affinidi-did-resolver-cache-server/tests/integration_test.rs b/affinidi-did-resolver-cache-server/tests/integration_test.rs
new file mode 100644
index 0000000..5c01d0f
--- /dev/null
+++ b/affinidi-did-resolver-cache-server/tests/integration_test.rs
@@ -0,0 +1,127 @@
+use affinidi_did_resolver_cache_sdk::{config::ClientConfigBuilder, DIDCacheClient};
+use affinidi_did_resolver_cache_server::server::start;
+use blake2::{Blake2s256, Digest};
+use did_peer::{
+    DIDPeer, DIDPeerCreateKeys, DIDPeerKeyType, DIDPeerKeys, DIDPeerService, PeerServiceEndPoint,
+    PeerServiceEndPointLong,
+};
+
+use ssi::{
+    dids::{DIDBuf, Document},
+    JWK,
+};
+
+const DID_ETHR: &str = "did:ethr:0x1:0xb9c5714089478a327f09197987f16f9e5d936e8a";
+const DID_JWK: &str= "did:jwk:eyJjcnYiOiJQLTI1NiIsImt0eSI6IkVDIiwieCI6ImFjYklRaXVNczNpOF91c3pFakoydHBUdFJNNEVVM3l6OTFQSDZDZEgyVjAiLCJ5IjoiX0tjeUxqOXZXTXB0bm1LdG00NkdxRHo4d2Y3NEk1TEtncmwyR3pIM25TRSJ9";
+const DID_KEY: &str = "did:key:z6MkiToqovww7vYtxm1xNM15u9JzqzUFZ1k7s7MazYJUyAxv";
+const DID_PKH: &str =
+    "did:pkh:solana:4sGjMW1sUnHzSxGspuhpqLDx6wiyjNtZ:CKg5d12Jhpej1JqtmxLJgaFqqeYjxgPqToJ4LBdvG9Ev";
+
+#[tokio::test]
+async fn test_server() {
+    //  Run cache server
+    tokio::spawn(async move { start().await });
+    println!("Server running");
+
+    // create did peer
+    let (e_did_key, v_did_key, keys) = _get_keys(DIDPeerKeyType::Secp256k1, true);
+    let services = vec![DIDPeerService {
+        _type: "dm".into(),
+        service_end_point: PeerServiceEndPoint::Long(PeerServiceEndPointLong {
+            uri: "https://localhost:7037".into(),
+            accept: vec!["didcomm/v2".into()],
+            routing_keys: vec![],
+        }),
+        id: None,
+    }];
+
+    let (did_peer, _) = DIDPeer::create_peer_did(&keys, Some(&services)).unwrap();
+    // Check did created
+    let parts: Vec<&str> = did_peer.split(":").collect();
+    let mut method_ids: Vec<&str> = parts[2].split(".").collect();
+    method_ids = method_ids[1..].to_vec();
+    let keys_multibase = [v_did_key[8..].to_string(), e_did_key[8..].to_string()];
+
+    for i in 0..2 {
+        assert!(keys_multibase.contains(&method_ids[i][1..].to_string()));
+    }
+    assert_eq!(parts.len(), 3);
+    assert_eq!(parts[1], "peer");
+
+    // Build config with network
+    let config = ClientConfigBuilder::default()
+        .with_network_mode("ws://127.0.0.1:8080/did/v1/ws")
+        .build();
+
+    // Resolve DIDs and add to cache
+    let client = DIDCacheClient::new(config).await.unwrap();
+    let dids: Vec<&str> = vec![&did_peer, DID_ETHR, DID_JWK, DID_KEY, DID_PKH];
+    let mut did_docs_vec: Vec<Document> = vec![];
+    for did in dids.clone() {
+        let res = client.resolve(did).await.unwrap();
+        let doc = res.doc;
+        did_docs_vec.push(doc);
+        assert!(!res.cache_hit)
+    }
+
+    // Check if it was a cache hit
+    assert!(client.resolve(DID_ETHR).await.unwrap().cache_hit);
+
+    // Match doc in cache with resolved doc
+    let cache = client.get_cache().clone();
+    for (i, did) in dids.clone().iter().enumerate() {
+        let in_cache_doc = cache.get(&_hash_did(did)).await.unwrap();
+        assert_eq!(in_cache_doc, did_docs_vec[i]);
+    }
+    client.remove(DID_PKH).await.unwrap();
+    assert!(!client.get_cache().contains_key(&_hash_did(DID_PKH)));
+}
+
+fn _hash_did(did: &str) -> String {
+    let mut hasher = Blake2s256::new();
+    hasher.update(did);
+    format!("{:x}", hasher.clone().finalize())
+}
+
+fn _get_keys(
+    key_type: DIDPeerKeyType,
+    with_pub_key: bool,
+) -> (DIDBuf, DIDBuf, Vec<DIDPeerCreateKeys>) {
+    let encryption_key = match key_type {
+        DIDPeerKeyType::Ed25519 => JWK::generate_ed25519().unwrap(),
+        DIDPeerKeyType::P256 => JWK::generate_p256(),
+        DIDPeerKeyType::Secp256k1 => JWK::generate_secp256k1(),
+    };
+    let verification_key = match key_type {
+        DIDPeerKeyType::Ed25519 => JWK::generate_ed25519().unwrap(),
+        DIDPeerKeyType::P256 => JWK::generate_p256(),
+        DIDPeerKeyType::Secp256k1 => JWK::generate_secp256k1(),
+    };
+    //  Create the did:key DID's for each key above
+    let e_did_key = ssi::dids::DIDKey::generate(&encryption_key).unwrap();
+    let v_did_key = ssi::dids::DIDKey::generate(&verification_key).unwrap();
+
+    // Put these keys in order and specify the type of each key (we strip the did:key: from the front)
+    let keys = vec![
+        DIDPeerCreateKeys {
+            purpose: DIDPeerKeys::Verification,
+            type_: Some(key_type.clone()),
+            public_key_multibase: if with_pub_key {
+                Some(v_did_key[8..].to_string())
+            } else {
+                None
+            },
+        },
+        DIDPeerCreateKeys {
+            purpose: DIDPeerKeys::Encryption,
+            type_: Some(key_type.clone()),
+            public_key_multibase: if with_pub_key {
+                Some(e_did_key[8..].to_string())
+            } else {
+                None
+            },
+        },
+    ];
+
+    (e_did_key, v_did_key, keys)
+}