use of com.google.cloud.notebooks.v1beta1.Environment in project java-notebooks by googleapis.
the class ITNotebookServiceClientTest method testGetEnvironment.
@Test
public void testGetEnvironment() {
GetEnvironmentRequest environmentRequest = GetEnvironmentRequest.newBuilder().setName(ENVIRONMENT_NAME).build();
Environment expectedEnvironmentResponse = client.getEnvironment(environmentRequest);
assertEquals(expectedEnvironmentResponse.getName(), expectedEnvironmentResponse.getName());
assertEquals(expectedEnvironmentResponse.getContainerImage(), expectedEnvironmentResponse.getContainerImage());
}
use of com.google.cloud.notebooks.v1beta1.Environment in project java-notebooks by googleapis.
the class ITNotebookServiceClientTest method testListEnvironment.
@Test
public void testListEnvironment() {
ListEnvironmentsRequest request = ListEnvironmentsRequest.newBuilder().setParent(PARENT).build();
for (Environment element : client.listEnvironments(request).iterateAll()) {
if (element.getName().equals(ENVIRONMENT_ID)) {
assertEquals(expectedEnvironmentResponse.getName(), element.getName());
assertEquals(expectedEnvironmentResponse.getContainerImage(), element.getContainerImage());
}
}
}
use of com.google.cloud.notebooks.v1beta1.Environment in project java-notebooks by googleapis.
the class ITNotebookServiceClientTest method setUp.
@BeforeClass
public static void setUp() throws IOException, ExecutionException, InterruptedException {
// Create Test Notebook Instance
client = NotebookServiceClient.create();
ContainerImage containerImage = ContainerImage.newBuilder().setRepository(FieldBehavior.OPTIONAL.name()).build();
Environment environment = Environment.newBuilder().setName(ENVIRONMENT_NAME).setContainerImage(containerImage).build();
CreateEnvironmentRequest environmentRequest = CreateEnvironmentRequest.newBuilder().setParent(PARENT).setEnvironmentId(ENVIRONMENT_ID).setEnvironment(environment).build();
expectedEnvironmentResponse = client.createEnvironmentAsync(environmentRequest).get();
Instance notebookInstance = Instance.newBuilder().setContainerImage(containerImage).setMachineType(MACHINE_TYPE_A).build();
CreateInstanceRequest instanceRequest = CreateInstanceRequest.newBuilder().setParent(PARENT).setInstanceId(NOTEBOOK_INSTANCE_ID).setInstance(notebookInstance).build();
expectedNotebookInstance = client.createInstanceAsync(instanceRequest).get();
}
use of com.google.cloud.notebooks.v1beta1.Environment in project java-notebooks by googleapis.
the class ITNotebookServiceClientTest method tearDown.
@AfterClass
public static void tearDown() throws ExecutionException, InterruptedException {
// Delete Test Environment Instance
DeleteEnvironmentRequest deleteEnvironmentRequest = DeleteEnvironmentRequest.newBuilder().setName(ENVIRONMENT_NAME).build();
client.deleteEnvironmentAsync(deleteEnvironmentRequest).get();
// Delete Test Notebook Instance
DeleteInstanceRequest deleteInstanceRequest = DeleteInstanceRequest.newBuilder().setName(INSTANCE_NAME).build();
client.deleteInstanceAsync(deleteInstanceRequest).get();
client.close();
}
use of com.google.cloud.notebooks.v1beta1.Environment in project java-notebooks by googleapis.
the class NotebookServiceClientTest method listEnvironmentsTest.
@Test
public void listEnvironmentsTest() throws Exception {
Environment responsesElement = Environment.newBuilder().build();
ListEnvironmentsResponse expectedResponse = ListEnvironmentsResponse.newBuilder().setNextPageToken("").addAllEnvironments(Arrays.asList(responsesElement)).build();
mockNotebookService.addResponse(expectedResponse);
ListEnvironmentsRequest request = ListEnvironmentsRequest.newBuilder().setParent("parent-995424086").setPageSize(883849137).setPageToken("pageToken873572522").build();
ListEnvironmentsPagedResponse pagedListResponse = client.listEnvironments(request);
List<Environment> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getEnvironmentsList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockNotebookService.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListEnvironmentsRequest actualRequest = ((ListEnvironmentsRequest) actualRequests.get(0));
Assert.assertEquals(request.getParent(), actualRequest.getParent());
Assert.assertEquals(request.getPageSize(), actualRequest.getPageSize());
Assert.assertEquals(request.getPageToken(), actualRequest.getPageToken());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Aggregations