/*
 * Copyright 2015 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except
 * in compliance with the License. A copy of the License is located at
 *
 * http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package com.amazonaws.services.dynamodbv2.mapper.integration;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBAttribute;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBHashKey;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBTable;

public class KeyOnlyPutITCase extends DynamoDBCryptoIntegrationTestBase {
  @DynamoDBTable(tableName = "aws-java-sdk-util-crypto")
  public static class HashAndAttribute {

    protected String key;
    protected String normalStringAttribute;

    @DynamoDBHashKey
    public String getKey() {
      return key;
    }

    public void setKey(String key) {
      this.key = key;
    }

    @DynamoDBAttribute
    public String getNormalStringAttribute() {
      return normalStringAttribute;
    }

    public void setNormalStringAttribute(String normalStringAttribute) {
      this.normalStringAttribute = normalStringAttribute;
    }

    @Override
    public int hashCode() {
      final int prime = 31;
      int result = 1;
      result = prime * result + ((key == null) ? 0 : key.hashCode());
      result =
          prime * result + ((normalStringAttribute == null) ? 0 : normalStringAttribute.hashCode());
      return result;
    }

    @Override
    public boolean equals(Object obj) {
      if (this == obj) return true;
      if (obj == null) return false;
      if (getClass() != obj.getClass()) return false;
      HashAndAttribute other = (HashAndAttribute) obj;
      if (key == null) {
        if (other.key != null) return false;
      } else if (!key.equals(other.key)) return false;
      if (normalStringAttribute == null) {
        if (other.normalStringAttribute != null) return false;
      } else if (!normalStringAttribute.equals(other.normalStringAttribute)) return false;
      return true;
    }
  }

  private <T extends HashAndAttribute> T getUniqueObject(T obj) {
    obj.setKey("" + startKey++);
    return obj;
  }
}