Search in sources :

Example 1 with StringDataSchemaLocation

use of com.linkedin.data.schema.StringDataSchemaLocation in project rest.li by linkedin.

the class TestTemplateSpecGenerator method initMocks.

@BeforeMethod
public void initMocks() {
    MockitoAnnotations.initMocks(this);
    _uniqueNumberGenerator = new AtomicInteger();
    _location = new StringDataSchemaLocation("location");
    HashMap<String, DataSchemaLocation> schemaToLocation = new HashMap<>();
    schemaToLocation.put(INPUT_SCHEMA_NAME, _location);
    schemaToLocation.put(CUSTOM_TYPE_1.getFullName(), _location);
    schemaToLocation.put(CUSTOM_TYPE_2.getFullName(), _location);
    Mockito.when(_resolver.nameToDataSchemaLocations()).thenReturn(schemaToLocation);
}
Also used : AtomicInteger(java.util.concurrent.atomic.AtomicInteger) HashMap(java.util.HashMap) StringDataSchemaLocation(com.linkedin.data.schema.StringDataSchemaLocation) StringDataSchemaLocation(com.linkedin.data.schema.StringDataSchemaLocation) DataSchemaLocation(com.linkedin.data.schema.DataSchemaLocation) BeforeMethod(org.testng.annotations.BeforeMethod)

Aggregations

DataSchemaLocation (com.linkedin.data.schema.DataSchemaLocation)1 StringDataSchemaLocation (com.linkedin.data.schema.StringDataSchemaLocation)1 HashMap (java.util.HashMap)1 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)1 BeforeMethod (org.testng.annotations.BeforeMethod)1