dynamite/test/testHashKeyOnly.js

promise-based DynamoDB client

// Copyright 2013 The Obvious Corporation.
var utils = require('./utils/testUtils.js')
var nodeunitq = require('nodeunitq')
var builder = new nodeunitq.Builder(exports)

var onError = function (err) {
  console.error(err.stack)
}
var initialData = [{"userId": "userA", "column": "@", "age": "29"}]

// basic setup for the tests, creating record userA with range key @
exports.setUp = function (done) {
  var self = this
  this.db = utils.getMockDatabase()
  this.client = utils.getMockDatabaseClient()
  utils.ensureLocalDynamo()
  utils.createTable(self.db, "userRangeOnly", "userId")
    .thenBound(utils.initTable, null, {db: self.db, tableName: "userRangeOnly", data: initialData})
    .fail(onError)
    .fin(done)
}

exports.tearDown = function (done) {
  utils.deleteTable(this.db, 'userRangeOnly')
    .fin(done)
}

// check that an item exists
builder.add(function testItemExists(test) {
  return this.client.getItem('userRangeOnly')
    .setHashKey('userId', 'userA')
    .execute()
    .then(function (data) {
      test.equal(data.result.age, 29, 'Age should match the provided age')
    })
})

// put an item and check that it exists
builder.add(function testSimplePut(test) {
  var self = this
  return this.client.putItem("userRangeOnly", {
    userId: 'userB',
    column: '@',
    age: 30
  })
  .execute()
  .then(function () {
    return utils.getItemWithSDK(self.db, "userB", null, 'userRangeOnly')
  })
  .then(function (data) {
    test.equal(data['Item']['age'].N, "30", "Age should be set")
  })
})

// put a list of strings and check if they exist
builder.add(function testStringSetPut(test) {
  var self = this
  return this.client.putItem("userRangeOnly", {
    userId: 'userC',
    column: '@',
    postIds: ['3a', '3b', '3c']
  })
  .execute()
  .then(function () {
    return utils.getItemWithSDK(self.db, "userC", null, 'userRangeOnly')
  })
  .then(function (data) {
    test.deepEqual(data['Item']['postIds'].SS, ['3a', '3b', '3c'], "postIds should be ['3a', '3b', '3c']")
  })
})

// test putting an attribute for an existing record
builder.add(function testPutAttributeExisting(test) {
  var self = this

  return this.client.newUpdateBuilder('userRangeOnly')
    .setHashKey('userId', 'userA')
    .enableUpsert()
    .putAttribute('age', 30)
    .putAttribute('height', 72)
    .execute()
    .then(function (data) {
      test.equal(data.result.age, 30, 'result age should be 30')
      test.equal(data.result.height, 72, 'height should be 72')
      return utils.getItemWithSDK(self.db, "userA", null, 'userRangeOnly')
    })
    .then(function (data) {
      test.equal(data['Item']['age'].N, "30", "result age should be 30")
      test.equal(data['Item']['height'].N, "72", "height should be 72")
    })
})


builder.add(function testDeleteItem(test) {
  //AWS.config.logger = process.stdout

  var self = this
  return self.client.deleteItem('userRangeOnly')
    .setHashKey('userId', 'userA')
    .execute()
    .then(function () {
      return utils.getItemWithSDK(self.db, "userA", null, "userRangeOnly")
    })
    .then(function (data) {
      test.equal(data['Item'], undefined, "User should be deleted~~~" + JSON.stringify(data))
    })
})