bot18/node_modules/bitfinex-api-node/test/helpers/test_model.js

A high-frequency cryptocurrency trading bot by Zenbot creator @carlos8f

/* eslint-env mocha */
'use strict'

const assert = require('assert')

const testModel = ({ values = {}, model, orderedFields, boolFields = [] }) => {
  const Model = model
  const fields = orderedFields
  const fieldValues = fields.slice()

  // Apply overrides
  for (let i = 0; i < fields.length; i++) {
    if (values[fields[i]]) {
      fieldValues[i] = values[fields[i]]
    }

    if (boolFields.indexOf(fields[i]) !== -1) {
      fieldValues[i] = false
    }
  }

  const checkModelFields = (m) => {
    fields.forEach((f) => {
      checkField(m, f)
    })
  }

  const checkField = (m, f) => {
    if (f === null) return

    if (boolFields.indexOf(f) !== -1) {
      assert.equal(m[f], false)
    } else if (values[f]) {
      assert.equal(m[f], values[f])
    } else {
      assert.equal(m[f], f)
    }
  }

  it('constructs from an array source', () => {
    const m = new Model(fieldValues)
    checkModelFields(m)
  })

  it('constructs from an object source', () => {
    const data = {}
    fields.forEach(f => (f !== null) && (data[f] = f))
    boolFields.forEach((f) => { data[f] = false })
    Object.assign(data, values)

    const m = new Model(data)
    checkModelFields(m)
  })

  it('serializes correctly', () => {
    const data = {}
    fields.forEach(f => (f !== null) && (data[f] = f))
    boolFields.forEach((f) => { data[f] = false })
    Object.assign(data, values)

    const m = new Model(data)
    const arr = m.serialize()

    arr.forEach((v, i) => {
      checkField(m, fields[i])
    })
  })

  it('unserializes correctly', () => {
    const m = model.unserialize(fieldValues)
    checkModelFields(m)
  })
}

module.exports = testModel