Examples with Assets com.google.api.services.dataplex.v1.CloudDataplex.Projects.Locations.Lakes.Zones.Assets used on opensource projects

Example 1 with Assets

use of com.google.api.services.dataplex.v1.CloudDataplex.Projects.Locations.Lakes.Zones.Assets in project DataflowTemplates by GoogleCloudPlatform.

the class DataplexFileFormatConversionTest method setup.

@Before
public void setup() {
    final String tempDir = temporaryFolder.getRoot().getAbsolutePath();
    outputAsset = new GoogleCloudDataplexV1Asset().setName("projects/p1/locations/l1/lakes/l1/zones/z1/assets/aout");
    outputAsset.setResourceSpec(new GoogleCloudDataplexV1AssetResourceSpec().setType(DataplexAssetResourceSpec.STORAGE_BUCKET.name()).setName(tempDir));
}

Example 2 with Assets

use of com.google.api.services.dataplex.v1.CloudDataplex.Projects.Locations.Lakes.Zones.Assets in project DataflowTemplates by GoogleCloudPlatform.

the class DefaultDataplexClientTest method testCreateEntitiesWhenDiscoveryEnabled.

@Test
public void testCreateEntitiesWhenDiscoveryEnabled() throws IOException {
    CloudDataplex dataplex = mock(CloudDataplex.class, Answers.RETURNS_DEEP_STUBS);
    Zones zones = getZones(dataplex);
    Assets assets = getAssets(dataplex);
    // Don't care about the order they're checked, so set up for two calls.
    when(assets.get(any()).execute()).thenReturn(createAsset(ENABLED_ASSET_DISCOVERY_SPEC)).thenReturn(createAsset(DISABLED_ASSET_DISCOVERY_SPEC));
    when(zones.get(any()).execute()).thenReturn(createZone(ENABLED_ZONE_DISCOVERY_SPEC)).thenReturn(createZone(DISABLED_ZONE_DISCOVERY_SPEC));
    DataplexClient client = DefaultDataplexClient.withClient(dataplex);
    client.createMetadata(ASSET_NAME1, ImmutableList.of(UNUSED_METADATA), CreateBehavior.UPDATE_IF_EXISTS);
    client.createMetadata(ASSET_NAME1, ImmutableList.of(UNUSED_METADATA), CreateBehavior.UPDATE_IF_EXISTS);
    verify(assets, atLeastOnce()).get(ASSET_NAME1);
    verify(zones, atLeastOnce()).get(ZONE_NAME);
    // Enough to know nothing was done with partitions
    verify(zones, never()).entities();
}