Fixes OZ Audit 2 issues M01 and L02

buidler.config.js

@@ -22,11 +22,13 @@ const testnetUser1 = '0x573bd3868b7672332c4D22076f55Cb0b597eb5Fd' // Account 3
 const testnetUser2 = '0x7Cfc5a12506d92F29D52EC7B8d1148f46e9296ED' // Account 4
 const testnetUser3 = '0x50D6d6195b102f9b58A29a57E3D71822881033a5' // Account 5
 
+const optimizerEnabled = !process.env.OPTIMIZER_DISABLED
+
 const config = {
   solc: {
     version: "0.6.12",
     optimizer: {
-      enabled: true,
+      enabled: optimizerEnabled,
       runs: 200
     },
     evmVersion: "istanbul"

package.json

@@ -11,7 +11,7 @@
     "reinstall": "rm -rf node_modules/ && rm -f yarn.lock && yarn",
     "verify": "yarn hint && yarn test",
     "hint": "solhint \"contracts/**/*.sol\"",
-    "test": "buidler test",
+    "test": "OPTIMIZER_DISABLED=true buidler test",
     "compile": "buidler --show-stack-traces --max-memory 8192 compile",
     "start": "buidler node --port $LOCAL_BUIDLEREVM_PORT",
     "coverage": "rm -rf build cache test-results.xml && buidler coverage --network coverage --temp build",

test/PrizePool.test.js

@@ -227,6 +227,26 @@ describe('PrizePool', function() {
     })
 
     describe('captureAwardBalance()', () => {
+      it('should handle when the balance is less than the collateral', async () => {
+        await ticket.mock.totalSupply.returns(toWei('100'))
+        await yieldSourceStub.mock.balance.returns(toWei('99.9999'))
+
+        await expect(prizePool.captureAwardBalance()).to.not.emit(prizePool, 'ReserveFeeCaptured');
+        expect(await prizePool.awardBalance()).to.equal(toWei('0'))
+      })
+
+      it('should handle the situ when the total accrued interest is less than the captured total', async () => {
+        await ticket.mock.totalSupply.returns(toWei('100'))
+        await yieldSourceStub.mock.balance.returns(toWei('110'))
+
+        await comptroller.mock.reserveRateMantissa.returns('0')
+
+        // first capture the 10 tokens
+        await prizePool.captureAwardBalance()
+
+        await yieldSourceStub.mock.balance.returns(toWei('109.999'))
+      })
+
       it('should track the yield less the total token supply', async () => {
         await ticket.mock.totalSupply.returns(toWei('100'))
         await yieldSourceStub.mock.balance.returns(toWei('110'))