SimpleFeatureTest.java

package com.mapzen.pelias;

import com.mapzen.pelias.gson.Feature;
import com.mapzen.pelias.gson.Geometry;
import com.mapzen.pelias.gson.Properties;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.annotation.Config;

import android.os.Parcel;

import static org.fest.assertions.api.Assertions.assertThat;

@RunWith(RobolectricTestRunner.class) @Config(constants = BuildConfig.class, sdk = 21)
public class SimpleFeatureTest {
  @Test public void shouldReadAndWriteParcel() throws Exception {
    SimpleFeature before = getTestSimpleFeature();
    Parcel parcel = before.toParcel();
    SimpleFeature after = SimpleFeature.readFromParcel(parcel);
    assertThat(before).isEqualTo(after);
  }

  @Test public void getAddress_shouldReturnLabelTextWithoutName() throws Exception {
    assertThat(getTestSimpleFeature().address()).isEqualTo("Manhattan, NY");
  }

  @Test public void fromFeature_shouldConstructFromFeatureObject() throws Exception {
    assertThat(SimpleFeature.fromFeature(getTestFeature())).isEqualTo(getTestSimpleFeature());
  }

  @Test public void isAddress_shouldReturnTrueIfResultIsFromAddressLayer() throws Exception {
    Feature feature = getTestFeature();
    feature.properties.layer = "address";
    SimpleFeature simpleFeature = SimpleFeature.fromFeature(feature);
    assertThat(simpleFeature.isAddress()).isTrue();
  }

  @Test public void isAddress_shouldReturnFalseIfResultIsFromAnotherLayer() throws Exception {
    Feature feature = getTestFeature();
    feature.properties.layer = "locality";
    SimpleFeature simpleFeature = SimpleFeature.fromFeature(feature);
    assertThat(simpleFeature.isAddress()).isFalse();
  }

  public static SimpleFeature getTestSimpleFeature() {
    return getTestSimpleFeature("Test SimpleFeature");
  }

  public static SimpleFeature getTestSimpleFeature(String name) {
    return SimpleFeature.builder()
        .lat(1.0)
        .lng(1.0)
        .id("123")
        .gid("osm:venue:123")
        .name(name)
        .country("United States")
        .countryAbbr("USA")
        .region("New York")
        .regionAbbr("NY")
        .county("New York County")
        .localAdmin("Manhattan")
        .locality("New York")
        .neighborhood("Koreatown")
        .confidence(1.0)
        .label("Test SimpleFeature, Manhattan, NY")
        .layer("locality")
        .build();
  }

  public static Feature getTestFeature() {
    final Properties properties = new Properties();
    properties.id = "123";
    properties.gid = "osm:venue:123";
    properties.name = "Test SimpleFeature";
    properties.country = "United States";
    properties.country_a = "USA";
    properties.region = "New York";
    properties.region_a = "NY";
    properties.county = "New York County";
    properties.localadmin = "Manhattan";
    properties.locality = "New York";
    properties.neighbourhood = "Koreatown";
    properties.confidence = 1.0;
    properties.label = "Test SimpleFeature, Manhattan, NY";
    properties.layer = "locality";

    final Geometry geometry = new Geometry();
    geometry.coordinates.add(1.0);
    geometry.coordinates.add(1.0);

    final Feature feature = new Feature();
    feature.properties = properties;
    feature.geometry = geometry;
    return feature;
  }
}