Fix node wrapper tests

Signed-off-by: Miroslav Kovar <miroslavkovar@protonmail.com>
hyperledger · Sep 24, 2020 · c0c76f1 · c0c76f1
1 parent 496e4bd
commit c0c76f1
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/wrappers/node/src/api/common.ts b/wrappers/node/src/api/common.ts
@@ -138,5 +138,6 @@ export interface IPaymentOutput {
 
 export interface ISerializedData<T> {
   version: string,
-  data: T
+  data: T,
+  source_id: string
 }
diff --git a/wrappers/node/src/api/vcx-base.ts b/wrappers/node/src/api/vcx-base.ts
@@ -13,7 +13,7 @@ export abstract class VCXBase<SerializedData> extends GCWatcher {
     constructorParams?: P
   ): Promise<T> {
     try {
-      const obj = new VCXClass(objData.data.source_id, constructorParams)
+      const obj = new VCXClass(objData.data.source_id || objData.source_id, constructorParams)
       await obj._initFromData(objData)
       return obj
     } catch (err) {

diff --git a/wrappers/node/test/suite1/ariesvcx-connection.test.ts b/wrappers/node/test/suite1/ariesvcx-connection.test.ts
@@ -79,17 +79,16 @@ describe('Connection:', () => {
       assert.property(serialized, 'version')
       assert.property(serialized, 'data')
       assert.property(serialized, 'state')
-      assert.property(serialized.data, 'source_id')
+      assert.property(serialized, 'source_id')
       assert.property(serialized.data, 'pw_did')
-      assert.property(serialized.data, 'pw_verkey')
-      assert.property(serialized.data, 'endpoint')
-      assert.property(serialized.data, 'their_pw_did')
-      assert.property(serialized.data, 'their_pw_verkey')
-      const { data, version } = serialized
+      assert.property(serialized.data, 'pw_vk')
+      assert.property(serialized.data, 'agent_did')
+      assert.property(serialized.data, 'agent_vk')
+      const { data, version, source_id } = serialized
       assert.ok(data)
       assert.ok(version)
-      assert.equal(data.source_id, connection.sourceId)
-      assert.equal(data.state, StateType.Initialized)
+      assert.ok(source_id)
+      assert.equal(source_id, connection.sourceId)
     })
 
     // TODO: restore for aries