fog--fog/tests/aws/requests/dynamodb/table_tests.rb

Shindo.tests('Fog::AWS[:dynamodb] | table requests', ['aws']) do

  @table_format = {
    'CreationDateTime'  => Float,
    'KeySchema'             => {
      'HashKeyElement' => {
        'AttributeName' => String,
        'AttributeType' => String
      }
    },
    'ProvisionedThroughput' => {
      'ReadCapacityUnits'   => Integer,
      'WriteCapacityUnits'  => Integer
    },
    'TableName'             => String,
    'TableStatus'           => String
  }

  @table_name = "fog_table_#{Time.now.to_f.to_s.gsub('.','')}"

  tests('success') do

    tests("#create_table(#{@table_name}, {'HashKeyElement' => {'AttributeName' => 'id', 'AttributeType' => 'S'}, {'ReadCapacityUnits' => 5, 'WriteCapacityUnits' => 5})").formats('TableDescription' => @table_format) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].create_table(@table_name, {'HashKeyElement' => {'AttributeName' => 'id', 'AttributeType' => 'S'}}, {'ReadCapacityUnits' => 5, 'WriteCapacityUnits' => 5}).body
    end

    tests("#describe_table(#{@table_name})").formats('Table' => @table_format) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].describe_table(@table_name).body
    end

    tests("#list_tables").formats({'LastEvaluatedTableName' => Fog::Nullable::String, 'TableNames' => [String]}) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].list_tables.body
    end

    unless Fog.mocking?
      Fog.wait_for { Fog::AWS[:dynamodb].describe_table(@table_name).body['Table']['TableStatus'] == 'ACTIVE' }
    end

    @update_table_format = {
      'TableDescription' => @table_format.merge({
        'ItemCount'             => Integer,
        'ProvisionedThroughput' => {
          'LastIncreaseDateTime'  => Float,
          'ReadCapacityUnits'     => Integer,
          'WriteCapacityUnits'    => Integer
        },
        'TableSizeBytes'        => Integer
      })
    }

    tests("#update_table(#{@table_name}, {'ReadCapacityUnits' => 10, 'WriteCapacityUnits' => 10})").formats(@update_table_format) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].update_table(@table_name, {'ReadCapacityUnits' => 10, 'WriteCapacityUnits' => 10}).body
    end

    unless Fog.mocking?
      Fog.wait_for { Fog::AWS[:dynamodb].describe_table(@table_name).body['Table']['TableStatus'] == 'ACTIVE' }
    end

    @delete_table_format = {
      'TableDescription' => {
        'ProvisionedThroughput' => {
          'ReadCapacityUnits'   => Integer,
          'WriteCapacityUnits'  => Integer
        },
        'TableName'      => String,
        'TableStatus'    => String
      }
    }

    tests("#delete_table(#{@table_name}").formats(@delete_table_format) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].delete_table(@table_name).body
    end

  end

  tests('failure') do

    tests("#delete_table('notatable')").raises(Excon::Errors::BadRequest) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].delete_table('notatable')
    end

    tests("#describe_table('notatable')").raises(Excon::Errors::BadRequest) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].describe_table('notatable')
    end

    tests("#update_table('notatable', {'ReadCapacityUnits' => 10, 'WriteCapacityUnits' => 10})").raises(Excon::Errors::BadRequest) do
      pending if Fog.mocking?
      Fog::AWS[:dynamodb].update_table('notatable', {'ReadCapacityUnits' => 10, 'WriteCapacityUnits' => 10}).body
    end

  end
end