Merge pull request #181 from zeppelinos/extend_app_wrapper_functionality

Extend App wrapper functionality

src/app/App.js

@@ -9,6 +9,7 @@ import AppProvider from './AppProvider'
 import AppDeployer from './AppDeployer'
 
 const log = new Logger('App')
+const ZERO_ADDRESS = '0x0000000000000000000000000000000000000000'
 
 export default class App {
 
@@ -49,6 +50,10 @@ export default class App {
     return this.currentDirectory().stdlib()
   }
 
+  async hasStdlib() {
+    return (await this.currentStdlib()) === ZERO_ADDRESS
+  }
+
   async getImplementation(contractName) {
     const directory = this.currentDirectory()
     return directory.getImplementation(contractName)
@@ -67,6 +72,12 @@ export default class App {
     return implementation
   }
 
+  async unsetImplementation(contractName) {
+    log.info(`Unsetting implementation of ${contractName} in directory...`)
+    await this.currentDirectory().unsetImplementation(contractName, this.txParams)
+    log.info(`Implementation unset`)
+  }
+
   async setStdlib(stdlibAddress = 0x0) {
     log.info(`Setting stdlib ${stdlibAddress}...`)
     await this.currentDirectory().setStdlib(stdlibAddress, this.txParams)

test/src/app/App.test.js

@@ -49,6 +49,10 @@ contract('App', function ([_, owner]) {
       const stdlib = await this.app.currentStdlib();
       stdlib.should.be.eq(currentStdlib)
     });
+
+    it('should tell whether current stdlib is zero', async function () {
+      (await this.app.hasStdlib()).should.be.false
+    })
   };
 
   describe('without stdlib', function () {
@@ -58,6 +62,10 @@ contract('App', function ([_, owner]) {
 
     describe('deploy', function () {
       shouldInitialize();
+
+      it('should not have an stdlib initially', async function () {
+        (await this.app.hasStdlib()).should.be.true
+      })
     });
 
     describe('connect', function () {
@@ -66,6 +74,10 @@ contract('App', function ([_, owner]) {
       });
 
       shouldInitialize();
+
+      it('should not have an stdlib initially', async function () {
+        (await this.app.hasStdlib()).should.be.true
+      })
     });
 
     const newVersion = '2.0';
@@ -108,6 +120,16 @@ contract('App', function ([_, owner]) {
       });
     });
 
+    describe('unsetImplementation', function () {
+      beforeEach('setting implementation', setImplementation)
+
+      it('should unset implementation on directory', async function () {
+        await this.app.unsetImplementation(contractName)
+        const implementation = await this.app.currentDirectory().getImplementation(contractName)
+        implementation.should.be.zeroAddress
+      })
+    })
+
     const createProxy = async function () {
       this.proxy = await this.app.createProxy(ImplV1, contractName);
     };