table_with_hash_key_and_range_key_spec.rb

require 'aws-sdk-dynamodb'

RSpec.describe 'table with partition (hash) key and sort (range) key' do

  it 'saves multiple items by hash and range keys to DynamoDB' do
    items = [{k1: 'a1', k2: 'a2'}, {k1: 'a1', k2: 'y'}]

    recreate_table('bar', 'k1', 'S', 'k2', 'S')
    client = Aws::DynamoDB::Client.new(connection_info)
    items.each do |item|
      client.put_item({
                          table_name: 'bar',
                          item: item
                      })
    end

    response = client.query({
                                table_name: 'bar',
                                select: 'COUNT',
                                key_condition_expression: 'k1 = :v_k1',
                                expression_attribute_values: {
                                    ':v_k1': 'a1'
                                }
                            })
    expect(response.count).to eql(2)
  end

  it 'supports querying using hash key and comparison on range key using begins_with' do
    items = [
        {
            Id: 206,
            Brand: 'Reebok',
            Size: 8
        },
        {
            Id: 206,
            Brand: 'Nike'
        },
        {
            Id: 206,
            Brand: 'Ni Hao'
        }
    ]

    recreate_table('shoes', 'Id', 'N', 'Brand', 'S')
    client = Aws::DynamoDB::Client.new(connection_info)
    items.each do |item|
      client.put_item({
                          table_name: 'shoes',
                          item: item
                      })
    end

    response = client.query({
                                table_name: 'shoes',
                                key_condition_expression: 'Id = :v_id AND begins_with(Brand, :v_brand)',
                                expression_attribute_values: {
                                    ':v_id': 206,
                                    ':v_brand': 'Ni'
                                }
                            })

    expect(response.items).not_to be_nil
    expect(response.count).to eql(2)
  end

end