com.cloudera.nav.sdk.client.MetadataExtractorTest.java Source code

Introduction

Here is the source code for com.cloudera.nav.sdk.client.MetadataExtractorTest.java

Source

/*
 * Copyright (c) 2015 Cloudera, 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.cloudera.nav.sdk.client;

import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

import com.cloudera.nav.sdk.model.Source;
import com.cloudera.nav.sdk.model.SourceType;
import com.google.common.collect.Lists;

import org.apache.commons.lang.StringUtils;
import org.junit.*;
import org.junit.runner.*;
import org.mockito.runners.*;

/**
 * Unit tests for MetadataExtractor
 */
@RunWith(MockitoJUnitRunner.class)
public class MetadataExtractorTest {

    private MetadataExtractor extractor;
    private String marker1Rep;
    private NavApiCient client;

    @Before
    @SuppressWarnings("unchecked")
    public void setUp() {
        client = mock(NavApiCient.class);
        extractor = new MetadataExtractor(client, null);
        marker1Rep = "{\"identityString\":100}";
        Source source1 = new Source("source1", SourceType.HDFS, "cluster1", "foo/bar", "identityString", 100L);
        when(client.getAllSources()).thenReturn(Lists.newArrayList(source1));
    }

    @Test
    public void testExtractAll() {
        MetadataResultSet res = extractor.extractMetadata();
        assertNotNull(res);
        assertTrue(StringUtils.isNotEmpty(res.getMarker()));
    }

    @Test
    public void testIncrementalExtract() {
        MetadataResultSet res = extractor.extractMetadata(marker1Rep);
        assertNotNull(res);
        assertEquals(res.getMarker(), marker1Rep);
    }

    @Test
    public void testIncrementalExtractQuery() {
        String entityQuery = "sourceType:HDFS";
        String relationQuery = "type: PARENT_CHILD";
        MetadataResultSet res = extractor.extractMetadata(marker1Rep, null, entityQuery, relationQuery);
        assertNotNull(res);
        assertEquals(res.getMarker(), marker1Rep);
    }

    @Test
    public void testCurrentMarker() {
        String res = extractor.getMarker();
        assertEquals(res, marker1Rep);
    }
}