test/com/android/i18n/addressinput/FormControllerTest.java

/*
 * Copyright (C) 2010 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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.android.i18n.addressinput;

import com.android.i18n.addressinput.LookupKey.KeyType;
import com.android.i18n.addressinput.testing.AsyncTestCase;

import java.util.List;

/**
 * Basic Tests for {@link FormController}.
 */
public class FormControllerTest extends AsyncTestCase {

    private static final AddressData US_CA_ADDRESS;
    private static final AddressData US_ADDRESS;
    private ClientData clientData;

    static {
        US_CA_ADDRESS = new AddressData.Builder().setCountry("US")
                .setAdminArea("CA")
                .setLocality("Mt View")
                .setAddressLine1("1098 Alta Ave")
                .setPostalCode("94043")
                .build();
        US_ADDRESS = new AddressData.Builder().setCountry("US").build();
    }

    @Override
    public void setUp() {
        clientData = new ClientData(new CacheData());
    }

    public void testRequestDataForAddress() {
        final FormController controller = new FormController(clientData, "en", "US");

        delayTestFinish(15000);

        controller.requestDataForAddress(US_CA_ADDRESS, new DataLoadListener() {
            boolean beginCalled = false;
            public void dataLoadingBegin() {
                beginCalled = true;
            }

            public void dataLoadingEnd() {
                assertTrue("dataLoadingBegin should be called before dataLoadingEnd",
                           beginCalled);
                LookupKey usCaMtvKey = new LookupKey.Builder(KeyType.DATA)
                        .setAddressData(US_CA_ADDRESS).build();
                LookupKey usKey = usCaMtvKey.getKeyForUpperLevelField(
                        AddressField.COUNTRY);
                LookupKey usCaKey = usCaMtvKey.getKeyForUpperLevelField(
                        AddressField.ADMIN_AREA);
                assertNotNull("key should be data/US/CA", usCaKey);
                assertNotNull("key should be data/US/CA/Mt View", usCaMtvKey);
                assertNotNull(clientData.get(usKey.toString()));
                assertNotNull(clientData.get(usCaKey.toString()));
                assertNull(clientData.get(usCaMtvKey.toString()));
                finishTest();
            }
        });
    }

    public void testRequestDataForBadAddress() {
        final AddressData address = new AddressData.Builder(US_CA_ADDRESS)
                .setAdminArea("FOOBAR")
                .setLocality("KarKar")
                .build();

        final FormController controller = new FormController(clientData, "en", "US");

        delayTestFinish(15000);

        controller.requestDataForAddress(address, new DataLoadListener() {
            boolean beginCalled = false;
            public void dataLoadingBegin() {
                beginCalled = true;
            }

            public void dataLoadingEnd() {
                assertTrue("dataLoadingBegin should be called before dataLoadingEnd",
                           beginCalled);
                LookupKey badKey = new LookupKey.Builder(KeyType.DATA)
                        .setAddressData(address).build();
                LookupKey usKey = badKey.getKeyForUpperLevelField(AddressField.COUNTRY);

                List<RegionData> rdata = controller.getRegionData(usKey);
                assertTrue(rdata.size() > 0);
                String subkey = rdata.get(0).getKey();
                assertNotNull("Should be the first US state", subkey);
                LookupKey usFirstStateKey =
                        new LookupKey.Builder(usKey.toString() + "/" + subkey).build();

                assertNotNull(clientData.get(usKey.toString()));
                assertNotNull(clientData.get(usFirstStateKey.toString()));
                assertNull(clientData.get(badKey.toString()));
                finishTest();
            }
        });
    }

    public void testRequestDataForCountry() {
        final FormController controller = new FormController(clientData, "en", "US");

        delayTestFinish(15000);

        controller.requestDataForAddress(US_ADDRESS, new DataLoadListener() {
            boolean beginCalled = false;
            public void dataLoadingBegin() {
                beginCalled = true;
            }

            public void dataLoadingEnd() {
                assertTrue("dataLoadingBegin should be called before dataLoadingEnd",
                           beginCalled);
                LookupKey usKey = new LookupKey.Builder(KeyType.DATA)
                        .setAddressData(US_ADDRESS).build();
                assertNotNull("key should be data/US", usKey);
                List<RegionData> rdata = controller.getRegionData(usKey);
                assertTrue(rdata.size() > 0);
                String subkey = rdata.get(0).getKey();
                assertNotNull("Should be the first US state", subkey);
                LookupKey usFirstStateKey =
                        new LookupKey.Builder(usKey.toString() + "/" + subkey).build();
                assertNotNull(clientData.get(usKey.toString()));
                assertNotNull(clientData.get(usFirstStateKey.toString()));
                finishTest();
            }
        });
    }
}