/*
 * 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 static org.testng.Assert.assertEquals;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;
import com.amazonaws.services.dynamodbv2.datamodeling.encryption.DynamoDBEncryptor;
import com.amazonaws.services.dynamodbv2.datamodeling.encryption.EncryptionContext;
import com.amazonaws.services.dynamodbv2.mapper.encryption.NumberSetAttributeTestClass;
import com.amazonaws.services.dynamodbv2.mapper.encryption.TestDynamoDBMapperFactory;
import com.amazonaws.services.dynamodbv2.mapper.encryption.TestEncryptionMaterialsProvider;
import com.amazonaws.services.dynamodbv2.model.AttributeValue;
import com.amazonaws.services.dynamodbv2.model.PutItemRequest;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

/** Tests string set attributes */
public class NumericSetAttributesITCase extends DynamoDBMapperCryptoIntegrationTestBase {

  private static final String INTEGER_ATTRIBUTE = "integerAttribute";
  private static final String FLOAT_OBJECT_ATTRIBUTE = "floatObjectAttribute";
  private static final String DOUBLE_OBJECT_ATTRIBUTE = "doubleObjectAttribute";
  private static final String BIG_INTEGER_ATTRIBUTE = "bigIntegerAttribute";
  private static final String BIG_DECIMAL_ATTRIBUTE = "bigDecimalAttribute";
  private static final String LONG_OBJECT_ATTRIBUTE = "longObjectAttribute";
  private static final String BYTE_OBJECT_ATTRIBUTE = "byteObjectAttribute";
  private static final String BOOLEAN_ATTRIBUTE = "booleanAttribute";

  // We don't start with the current system millis like other tests because
  // it's out of the range of some data types
  private static int start = 1;
  private static int byteStart = 1;

  private static final List<Map<String, AttributeValue>> attrs =
      new LinkedList<Map<String, AttributeValue>>();

  // Test data
  static {
    for (int i = 0; i < 5; i++) {
      Map<String, AttributeValue> attr = new HashMap<String, AttributeValue>();
      attr.put(KEY_NAME, new AttributeValue().withS("" + start++));
      attr.put(
          INTEGER_ATTRIBUTE, new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          FLOAT_OBJECT_ATTRIBUTE,
          new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          DOUBLE_OBJECT_ATTRIBUTE,
          new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          BIG_INTEGER_ATTRIBUTE,
          new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          BIG_DECIMAL_ATTRIBUTE,
          new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          LONG_OBJECT_ATTRIBUTE,
          new AttributeValue().withNS("" + start++, "" + start++, "" + start++));
      attr.put(
          BYTE_OBJECT_ATTRIBUTE,
          new AttributeValue().withNS("" + byteStart++, "" + byteStart++, "" + byteStart++));
      attr.put(BOOLEAN_ATTRIBUTE, new AttributeValue().withNS("0", "1"));
      attrs.add(attr);
    }
  }
  ;

  @BeforeClass
  public static void setUp() throws Exception {
    DynamoDBMapperCryptoIntegrationTestBase.setUp();
    DynamoDBEncryptor encryptor =
        DynamoDBEncryptor.getInstance(new TestEncryptionMaterialsProvider());
    EncryptionContext context =
        new EncryptionContext.Builder().withHashKeyName(KEY_NAME).withTableName(TABLE_NAME).build();
    // Insert the data
    for (Map<String, AttributeValue> attr : attrs) {
      attr = encryptor.encryptAllFieldsExcept(attr, context, KEY_NAME);
      dynamo.putItem(new PutItemRequest(TABLE_NAME, attr));
    }
  }

  @Test
  public void testLoad() throws Exception {
    DynamoDBMapper util = TestDynamoDBMapperFactory.createDynamoDBMapper(dynamo);

    for (Map<String, AttributeValue> attr : attrs) {
      NumberSetAttributeTestClass x =
          util.load(NumberSetAttributeTestClass.class, attr.get(KEY_NAME).getS());
      assertEquals(x.getKey(), attr.get(KEY_NAME).getS());

      // Convert all numbers to the most inclusive type for easy comparison
      assertNumericSetsEquals(x.getBigDecimalAttribute(), attr.get(BIG_DECIMAL_ATTRIBUTE).getNS());
      assertNumericSetsEquals(x.getBigIntegerAttribute(), attr.get(BIG_INTEGER_ATTRIBUTE).getNS());
      assertNumericSetsEquals(
          x.getFloatObjectAttribute(), attr.get(FLOAT_OBJECT_ATTRIBUTE).getNS());
      assertNumericSetsEquals(
          x.getDoubleObjectAttribute(), attr.get(DOUBLE_OBJECT_ATTRIBUTE).getNS());
      assertNumericSetsEquals(x.getIntegerAttribute(), attr.get(INTEGER_ATTRIBUTE).getNS());
      assertNumericSetsEquals(x.getLongObjectAttribute(), attr.get(LONG_OBJECT_ATTRIBUTE).getNS());
      assertNumericSetsEquals(x.getByteObjectAttribute(), attr.get(BYTE_OBJECT_ATTRIBUTE).getNS());
      assertSetsEqual(toSet("0", "1"), attr.get(BOOLEAN_ATTRIBUTE).getNS());
    }
  }

  @Test
  public void testSave() throws Exception {
    List<NumberSetAttributeTestClass> objs = new ArrayList<NumberSetAttributeTestClass>();
    for (int i = 0; i < 5; i++) {
      NumberSetAttributeTestClass obj = getUniqueObject();
      objs.add(obj);
    }

    DynamoDBMapper util = TestDynamoDBMapperFactory.createDynamoDBMapper(dynamo);
    for (NumberSetAttributeTestClass obj : objs) {
      util.save(obj);
    }

    for (NumberSetAttributeTestClass obj : objs) {
      NumberSetAttributeTestClass loaded =
          util.load(NumberSetAttributeTestClass.class, obj.getKey());
      assertEquals(obj, loaded);
    }
  }

  @Test
  public void testUpdate() throws Exception {
    List<NumberSetAttributeTestClass> objs = new ArrayList<NumberSetAttributeTestClass>();
    for (int i = 0; i < 5; i++) {
      NumberSetAttributeTestClass obj = getUniqueObject();
      objs.add(obj);
    }

    DynamoDBMapper util = TestDynamoDBMapperFactory.createDynamoDBMapper(dynamo);
    for (NumberSetAttributeTestClass obj : objs) {
      util.save(obj);
    }

    for (NumberSetAttributeTestClass obj : objs) {
      NumberSetAttributeTestClass replacement = getUniqueObject();
      replacement.setKey(obj.getKey());
      util.save(replacement);
      assertEquals(replacement, util.load(NumberSetAttributeTestClass.class, obj.getKey()));
    }
  }

  private NumberSetAttributeTestClass getUniqueObject() {
    NumberSetAttributeTestClass obj = new NumberSetAttributeTestClass();
    obj.setKey(String.valueOf(startKey++));
    obj.setBigDecimalAttribute(
        toSet(new BigDecimal(startKey++), new BigDecimal(startKey++), new BigDecimal(startKey++)));
    obj.setBigIntegerAttribute(
        toSet(
            new BigInteger("" + startKey++),
            new BigInteger("" + startKey++),
            new BigInteger("" + startKey++)));
    obj.setByteObjectAttribute(
        toSet(new Byte("" + byteStart++), new Byte("" + byteStart++), new Byte("" + byteStart++)));
    obj.setDoubleObjectAttribute(
        toSet(new Double("" + start++), new Double("" + start++), new Double("" + start++)));
    obj.setFloatObjectAttribute(
        toSet(new Float("" + start++), new Float("" + start++), new Float("" + start++)));
    obj.setIntegerAttribute(
        toSet(new Integer("" + start++), new Integer("" + start++), new Integer("" + start++)));
    obj.setLongObjectAttribute(
        toSet(new Long("" + start++), new Long("" + start++), new Long("" + start++)));
    obj.setBooleanAttribute(toSet(true, false));
    obj.setDateAttribute(toSet(new Date(startKey++), new Date(startKey++), new Date(startKey++)));
    Set<Calendar> cals = new HashSet<Calendar>();
    for (Date d : obj.getDateAttribute()) {
      Calendar cal = GregorianCalendar.getInstance();
      cal.setTime(d);
      cals.add(cal);
    }
    obj.setCalendarAttribute(toSet(cals));
    return obj;
  }
}