solidity-cron/test/periodicStages.js

[](https://travis-ci.com/Frontier-project/cron) [![codecov](https://codecov.io/gh/Frontier-project/cron/branch/master/graph/badge.svg?token=BGbU5Q6IRV

var advanceToBlock = require('./helpers/advanceToBlock')
var stagesFromBounds = require('./helpers/stages')
const { assertRevert } = require('./helpers/assertRevert')

const StackContract = artifacts.require('Stack')
const PeriodContract = artifacts.require('Period')
const PeriodicStageContract = artifacts.require('PeriodicStages')

let PeriodInstance
let Stack
let PeriodicStagesInstance
const T = 10
let initialOffset
let Registry

contract('PeriodicStages', function (accounts) {
  beforeEach(async () => {
    PeriodicStagesInstance = await PeriodicStageContract.new(T)
    let periodAddress = await PeriodicStagesInstance.period.call()
    PeriodInstance = await PeriodContract.at(periodAddress)
    let stackAddress = await PeriodicStagesInstance.stack.call()
    Stack = await StackContract.at(stackAddress)
  })
  describe('Calculating stages', async () => {
    it('Should start at stage 0', async () => {
      const currentStage = await PeriodicStagesInstance.currentStage.call()
      assert.strictEqual(0, currentStage.toNumber())
    })
    it('Should create a new Stage', async () => {
      const bounds = [T / 5]
      const totalCapacity = Stack.totalCapacity.call()
      stagesFromBounds.stagesFromBounds(PeriodicStagesInstance, bounds)
      assert.strictEqual(0, 0)
    })
    it('Should modify the Period Lentgth', async () => {
      await PeriodicStagesInstance.setPeriodLength(T - 1)
      const periodLength = await PeriodInstance.T.call()
      assert.strictEqual(T - 1, periodLength.toNumber())
    })
    it('Should be able to push a new stage', async () => {
      await PeriodicStagesInstance.setPeriodLength(T - 1)
      const newStageLength = 1
      await PeriodicStagesInstance.pushStage(newStageLength)
      const newHeight = await Stack.height.call()
      assert.strictEqual(1, newHeight.toNumber())
    })
    it('Should return 0 if the current block is inside of the first pushed stage', async () => {
      const bounds = [T / 5]
      const totalCapacity = await Stack.totalCapacity.call()
      await stagesFromBounds.stagesFromBounds(PeriodicStagesInstance, bounds)
      const indexInsideStage = await PeriodInstance.getRelativeIndex()
      const neededIndexInStage = 1
      const blocksToAdvance = T - indexInsideStage + neededIndexInStage
      await advanceToBlock.advanceToBlock(web3.eth.blockNumber + blocksToAdvance)
      const newIndexInsideStage = await PeriodInstance.getRelativeIndex()
      const currentStage = await PeriodicStagesInstance.currentStage.call()
      assert.strictEqual(0, currentStage.toNumber())
    })
    it('Should return 1 if the current block is inside of the second pushed stage', async () => {
      const bounds = [T / 5, 2 * T / 5]
      await stagesFromBounds.stagesFromBounds(PeriodicStagesInstance, bounds)
      const indexInsideStage = await PeriodInstance.getRelativeIndex()
      const neededIndexInStage = T / 5 + 1
      const blocksToAdvance = T - indexInsideStage + neededIndexInStage
      await advanceToBlock.advanceToBlock(web3.eth.blockNumber + blocksToAdvance)
      const newIndexInsideStage = await PeriodInstance.getRelativeIndex()
      const currentStage = await PeriodicStagesInstance.currentStage.call()
      assert.strictEqual(1, currentStage.toNumber())
    })
  })
})