Examples with Request org.apache.atlas.catalog.Request used on opensource projects

Example 16 with Request

use of org.apache.atlas.catalog.Request in project incubator-atlas by apache.

the class EntityTagResourceDefinitionTest method testValidate_missingName.

@Test(expectedExceptions = InvalidPayloadException.class)
public void testValidate_missingName() throws Exception {
    Map<String, Object> properties = new HashMap<>();
    Request request = new InstanceRequest(properties);
    ResourceDefinition entityTagDefinition = new EntityTagResourceDefinition();
    entityTagDefinition.validateCreatePayload(request);
}

Example 17 with Request

use of org.apache.atlas.catalog.Request in project incubator-atlas by apache.

the class EntityTagResourceDefinitionTest method testValidate_invalidProperty.

@Test(expectedExceptions = InvalidPayloadException.class)
public void testValidate_invalidProperty() throws Exception {
    Map<String, Object> properties = new HashMap<>();
    properties.put("name", "foo");
    properties.put("description", "desc");
    Request request = new InstanceRequest(properties);
    ResourceDefinition entityTagDefinition = new EntityTagResourceDefinition();
    entityTagDefinition.validateCreatePayload(request);
}

Example 18 with Request

use of org.apache.atlas.catalog.Request in project incubator-atlas by apache.

the class QueryFactoryTest method testCreateEntityTagQuery.

@Test
public void testCreateEntityTagQuery() throws Exception {
    Map<String, Object> requestProps = new HashMap<>();
    requestProps.put("id", "entity_id");
    requestProps.put("name", "test_taxonomy.term1");
    Request request = new InstanceRequest(requestProps);
    QueryFactory factory = new QueryFactory();
    AtlasEntityTagQuery query = (AtlasEntityTagQuery) factory.createEntityTagQuery(request);
    QueryExpression queryExpression = query.getQueryExpression();
    assertEquals(queryExpression.getClass(), TermQueryExpression.class);
    assertEquals(queryExpression.getField(), "name");
    assertEquals(queryExpression.getExpectedValue(), "test_taxonomy.term1");
    assertEquals(query.getRequest(), request);
    assertEquals(query.getResourceDefinition().getClass(), EntityTagResourceDefinition.class);
}

Example 19 with Request

use of org.apache.atlas.catalog.Request in project incubator-atlas by apache.

the class QueryFactoryTest method testCollectionQuery_WildcardQuery.

@Test
public void testCollectionQuery_WildcardQuery() throws Exception {
    String queryString = "name:ta?onomy";
    Request request = new CollectionRequest(Collections.<String, Object>emptyMap(), queryString);
    QueryFactory factory = new QueryFactory();
    AtlasTaxonomyQuery query = (AtlasTaxonomyQuery) factory.createTaxonomyQuery(request);
    QueryExpression queryExpression = query.getQueryExpression();
    assertEquals(queryExpression.getClass(), WildcardQueryExpression.class);
    assertEquals(queryExpression.getField(), "name");
    assertEquals(queryExpression.getExpectedValue(), "ta?onomy");
    assertEquals(query.getRequest(), request);
    assertEquals(query.getResourceDefinition().getTypeName(), "Taxonomy");
}

Example 20 with Request

use of org.apache.atlas.catalog.Request in project incubator-atlas by apache.

the class TaxonomyServiceTest method testGetTaxonomy.

@Test
public void testGetTaxonomy() throws Exception {
    String taxonomyName = "testTaxonomy";
    MetadataService metadataService = createStrictMock(MetadataService.class);
    AtlasTypeDefStore typeDefStore = createStrictMock(AtlasTypeDefStore.class);
    ResourceProvider taxonomyResourceProvider = createStrictMock(ResourceProvider.class);
    ResourceProvider termResourceProvider = createStrictMock(ResourceProvider.class);
    UriInfo uriInfo = createNiceMock(UriInfo.class);
    JsonSerializer serializer = createStrictMock(JsonSerializer.class);
    Capture<Request> requestCapture = newCapture();
    Collection<Map<String, Object>> resultPropertyMaps = new ArrayList<>();
    Map<String, Object> propertyMap = new HashMap<>();
    propertyMap.put("name", "testTaxonomy");
    resultPropertyMaps.add(propertyMap);
    Result result = new Result(resultPropertyMaps);
    expect(taxonomyResourceProvider.getResourceById(capture(requestCapture))).andReturn(result);
    expect(metadataService.getTypeDefinition(TaxonomyResourceProvider.TAXONOMY_TERM_TYPE)).andReturn(TaxonomyResourceProvider.TAXONOMY_TERM_TYPE + "-definition");
    expect(serializer.serialize(result, uriInfo)).andReturn("Taxonomy Get Response");
    replay(uriInfo, metadataService, taxonomyResourceProvider, termResourceProvider, serializer);
    TestTaxonomyService service = new TestTaxonomyService(metadataService, typeDefStore, taxonomyResourceProvider, termResourceProvider, serializer);
    Response response = service.getTaxonomy(null, uriInfo, taxonomyName);
    Request request = requestCapture.getValue();
    Map<String, Object> requestProperties = request.getQueryProperties();
    assertEquals(requestProperties.size(), 1);
    assertEquals(requestProperties.get("name"), taxonomyName);
    assertEquals(response.getStatus(), 200);
    assertEquals(response.getEntity(), "Taxonomy Get Response");
    verify(uriInfo, taxonomyResourceProvider, termResourceProvider, serializer);
}