Add Contract.address() method (#614)

stellar · May 31, 2023 · c8d3817 · c8d3817
1 parent 33dcef5
commit c8d3817
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/contract.js b/src/contract.js
@@ -1,3 +1,4 @@
+import { Address } from './address';
 import { Operation } from './operation';
 import xdr from './xdr';
 import { StrKey } from './strkey';
@@ -51,6 +52,14 @@ export class Contract {
     }
   }
 
+  /**
+   * Returns the address of this contract as an Address type.
+   * @returns {Address}
+   */
+  address() {
+    return Address.contract(this._id);
+  }
+
   /**
    * Returns an operation that will invoke this contract call.
    *

diff --git a/test/unit/contract_test.js b/test/unit/contract_test.js
@@ -19,6 +19,14 @@ describe('Contract', function () {
     });
   });
 
+  describe('address', function () {
+    it('returns the contract address', function () {
+      let contractId = '0'.repeat(63) + '1';
+      let contract = new StellarBase.Contract(contractId);
+      expect(contract.address().toBuffer().toString('hex')).to.equal(contractId);
+    });
+  });
+
   describe('getFootprint', function () {
     it('includes the correct contract code footprint', function () {
       let contractId = '0'.repeat(63) + '1';