com.lordofthejars.nosqlunit.dynamodb.ExpectedDataSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nosqlunit-dynamodb Show documentation

The newest version!


package com.lordofthejars.nosqlunit.dynamodb;

import java.util.List;
import java.util.Map;
import java.util.Set;

import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.fasterxml.jackson.core.type.TypeReference;

public class ExpectedDataSet {

    public static final TypeReference>>> TYPE_REFERENCE = new TypeReference>>>() {
    };

    private final Map>> dataset;

    public ExpectedDataSet(Map>> dataset) {
        this.dataset = dataset;
    }

    public Map>> getDataset() {
        return dataset;
    }

    public Set getTables() {
        return dataset.keySet();
    }

    public List> getDataFor(String tableName) {
        return dataset.get(tableName);
    }

}