CustomsItemTest.java

package com.shippo.model;

import static org.junit.Assert.*;

import java.util.HashMap;
import java.util.Map;

import org.junit.Test;

import com.shippo.Shippo;
import com.shippo.exception.APIConnectionException;
import com.shippo.exception.APIException;
import com.shippo.exception.AuthenticationException;
import com.shippo.exception.InvalidRequestException;
import com.shippo.exception.ShippoException;

public class CustomsItemTest extends ShippoTest {

    // test data
    private static Map<String, Object> objectMap;
    static {
        objectMap = new HashMap<String, Object>();
        objectMap.put("description", "T-Shirt");
        objectMap.put("quantity", "2");
        objectMap.put("net_weight", "400");
        objectMap.put("mass_unit", "g");
        objectMap.put("value_amount", "20");
        objectMap.put("value_currency", "USD");
        objectMap.put("origin_country", "US");
        objectMap.put("tariff_number", null);
        objectMap.put("sku_code", null);
        objectMap.put("eccn_ear99", "3A001");
        objectMap.put("metadata", "Order ID #123123");
    }

    @Test
    public void testValidCreate() {
        CustomsItem testObject = createCustomsItemFixture();
        assertEquals(Shippo.apiKeyIsTest, testObject.isTest());
        assertEquals(testObject.getObjectState(), "VALID");
        assertEquals(testObject.getDescription(), objectMap.get("description"));
        assertEquals(testObject.getQuantity(), Double.parseDouble((String) objectMap.get("quantity")));
        assertEquals(testObject.getNetWeight(), objectMap.get("net_weight"));
        assertEquals(testObject.getMassUnit(), objectMap.get("mass_unit"));
        assertEquals(testObject.getValueAmount(), objectMap.get("value_amount"));
        assertEquals(testObject.getValueCurrency(), objectMap.get("value_currency"));
        assertEquals(testObject.getOriginCountry(), objectMap.get("origin_country"));
        assertEquals(testObject.getTariffNumber(), nullToEmptyString(objectMap.get("tariff_number")));
        assertEquals(testObject.getSkuCode(), objectMap.get("sku_code"));
        assertEquals(testObject.getEccnEar99(), nullToEmptyString(objectMap.get("eccn_ear99")));
        assertEquals(testObject.getMetadata(), nullToEmptyString(objectMap.get("metadata")));
    }

    @Test(expected = InvalidRequestException.class)
    public void testInvalidCreate() throws AuthenticationException, InvalidRequestException, APIConnectionException,
            APIException {
        CustomsItem.create(getInvalidObjectMap());
    }

    @Test
    public void testRetrieve() throws AuthenticationException, InvalidRequestException, APIConnectionException,
            APIException {
        CustomsItem testObject = createCustomsItemFixture();
        CustomsItem retrievedObject;

        retrievedObject = CustomsItem.retrieve((String) testObject.objectId);
        assertEquals(testObject.objectId, retrievedObject.objectId);

    }

    @Test(expected = InvalidRequestException.class)
    public void testInvalidRetrieve() throws AuthenticationException, InvalidRequestException, APIConnectionException,
            APIException {
        CustomsItem.retrieve("invalid_id");
    }

    @Test
    public void testListAll() throws AuthenticationException, InvalidRequestException, APIConnectionException,
            APIException {
        CustomsItemCollection objectCollection = CustomsItem.all(null);
        assertNotNull(objectCollection.getData());
    }

    @Test
    public void testListPageSize() throws AuthenticationException, InvalidRequestException, APIConnectionException,
            APIException {
        Map<String, Object> objectMap = new HashMap<String, Object>();
        objectMap.put("results", "1"); // one result per page
        objectMap.put("page", "1"); // the first page of results
        CustomsItemCollection CustomsItemCollection = CustomsItem.all(objectMap);
        assertEquals(CustomsItemCollection.getData().size(), 1);
    }

    public static CustomsItem createCustomsItemFixture() {
        try {
            return CustomsItem.create(objectMap);
        } catch (ShippoException e) {
            e.printStackTrace();
        }
        return null;
    }
}