de.hybris.platform.ycommercewebservicestest.storelocator.impl.MockedGeoServiceWrapper.java Source code

Introduction

Here is the source code for de.hybris.platform.ycommercewebservicestest.storelocator.impl.MockedGeoServiceWrapper.java

Source

/*
 * [y] hybris Platform
 *
 * Copyright (c) 2000-2014 hybris AG
 * All rights reserved.
 *
 * This software is the confidential and proprietary information of hybris
 * ("Confidential Information"). You shall not disclose such Confidential
 * Information and shall use it only in accordance with the terms of the
 * license agreement you entered into with hybris.
 */
package de.hybris.platform.ycommercewebservicestest.storelocator.impl;

import de.hybris.platform.storelocator.GPS;
import de.hybris.platform.storelocator.data.AddressData;
import de.hybris.platform.storelocator.exception.GeoServiceWrapperException;
import de.hybris.platform.storelocator.impl.CommerceMockGeoWebServiceWrapper;
import de.hybris.platform.storelocator.location.Location;

import java.util.Map;

import org.apache.commons.lang.StringUtils;

/**
 * Mock wrapper to avoid communication with google geo-location services.
 */
public class MockedGeoServiceWrapper extends CommerceMockGeoWebServiceWrapper {
    private final Map<String, GPS> geoMap;

    public MockedGeoServiceWrapper(final Map<String, GPS> geoMap) {
        super();
        this.geoMap = geoMap;
    }

    @Override
    public GPS geocodeAddress(final Location address) throws GeoServiceWrapperException {
        final GPS gpsAddress = geoMap.get(StringUtils.lowerCase(address.getAddressData().getCity()));

        if (gpsAddress == null) {
            throw new GeoServiceWrapperException();
        }

        return gpsAddress;
    }

    @Override
    public GPS geocodeAddress(final AddressData address) throws GeoServiceWrapperException {
        final GPS gpsAddress = geoMap.get(StringUtils.lowerCase(address.getCity()));

        if (gpsAddress == null) {
            throw new GeoServiceWrapperException();
        }

        return gpsAddress;
    }

}