From 4b3e0383235ca4ca0127f24c2e05543bb45a56bb Mon Sep 17 00:00:00 2001
From: Amir Bandeali <abandeali1@gmail.com>
Date: Wed, 29 Nov 2017 22:02:43 -0800
Subject: Add contracts to packages, fix most linting errors

---
 .../contracts/test/ts/token_transfer_proxy/auth.ts | 83 ++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 packages/contracts/test/ts/token_transfer_proxy/auth.ts

(limited to 'packages/contracts/test/ts/token_transfer_proxy/auth.ts')

diff --git a/packages/contracts/test/ts/token_transfer_proxy/auth.ts b/packages/contracts/test/ts/token_transfer_proxy/auth.ts
new file mode 100644
index 000000000..d027fa9e7
--- /dev/null
+++ b/packages/contracts/test/ts/token_transfer_proxy/auth.ts
@@ -0,0 +1,83 @@
+import * as chai from 'chai';
+
+import {constants} from '../../../util/constants';
+import {ContractInstance} from '../../../util/types';
+import {chaiSetup} from '../utils/chai_setup';
+
+chaiSetup.configure();
+const expect = chai.expect;
+const TokenTransferProxy = artifacts.require('./db/TokenTransferProxy.sol');
+
+contract('TokenTransferProxy', (accounts: string[]) => {
+  const owner = accounts[0];
+  const notOwner = accounts[1];
+
+  let tokenTransferProxy: ContractInstance;
+  let authorized: string;
+  let notAuthorized = owner;
+
+  before(async () => {
+    tokenTransferProxy = await TokenTransferProxy.deployed();
+  });
+
+  describe('addAuthorizedAddress', () => {
+    it('should throw if not called by owner', async () => {
+      return expect(tokenTransferProxy.addAuthorizedAddress(notOwner, {from: notOwner}))
+        .to.be.rejectedWith(constants.INVALID_OPCODE);
+    });
+
+    it('should allow owner to add an authorized address', async () => {
+      await tokenTransferProxy.addAuthorizedAddress(notAuthorized, {from: owner});
+      authorized = notAuthorized;
+      notAuthorized = null;
+      const isAuthorized = await tokenTransferProxy.authorized.call(authorized);
+      expect(isAuthorized).to.be.true();
+    });
+
+    it('should throw if owner attempts to authorize a duplicate address', async () => {
+      return expect(tokenTransferProxy.addAuthorizedAddress(authorized, {from: owner}))
+        .to.be.rejectedWith(constants.INVALID_OPCODE);
+    });
+  });
+
+  describe('removeAuthorizedAddress', () => {
+    it('should throw if not called by owner', async () => {
+      return expect(tokenTransferProxy.removeAuthorizedAddress(authorized, {from: notOwner}))
+        .to.be.rejectedWith(constants.INVALID_OPCODE);
+    });
+
+    it('should allow owner to remove an authorized address', async () => {
+      await tokenTransferProxy.removeAuthorizedAddress(authorized, {from: owner});
+      notAuthorized = authorized;
+      authorized = null;
+
+      const isAuthorized = await tokenTransferProxy.authorized.call(notAuthorized);
+      expect(isAuthorized).to.be.false();
+    });
+
+    it('should throw if owner attempts to remove an address that is not authorized', async () => {
+      return expect(tokenTransferProxy.removeAuthorizedAddress(notAuthorized, {from: owner}))
+        .to.be.rejectedWith(constants.INVALID_OPCODE);
+    });
+  });
+
+  describe('getAuthorizedAddresses', () => {
+    it('should return all authorized addresses', async () => {
+      const initial = await tokenTransferProxy.getAuthorizedAddresses();
+      expect(initial).to.have.length(1);
+      await tokenTransferProxy.addAuthorizedAddress(notAuthorized, {from: owner});
+
+      authorized = notAuthorized;
+      notAuthorized = null;
+      const afterAdd = await tokenTransferProxy.getAuthorizedAddresses();
+      expect(afterAdd).to.have.length(2);
+      expect(afterAdd).to.include(authorized);
+
+      await tokenTransferProxy.removeAuthorizedAddress(authorized, {from: owner});
+      notAuthorized = authorized;
+      authorized = null;
+      const afterRemove = await tokenTransferProxy.getAuthorizedAddresses();
+      expect(afterRemove).to.have.length(1);
+    });
+  });
+});
-- 
cgit v1.2.3