public static DatabaseName parse(String formattedString) { if (formattedString.isEmpty()) { return null; } Map<String, String> matchMap = PATH_TEMPLATE.validatedMatch( formattedString, "DatabaseName.parse: formattedString not in valid format"); return of(matchMap.get("project"), matchMap.get("instance"), matchMap.get("database")); }
public static List<DatabaseName> parseList(List<String> formattedStrings) { List<DatabaseName> list = new ArrayList<>(formattedStrings.size()); for (String formattedString : formattedStrings) { list.add(parse(formattedString)); } return list; }
public static String format(String project, String instance, String database) { return newBuilder() .setProject(project) .setInstance(instance) .setDatabase(database) .build() .toString(); }
@Test @SuppressWarnings("all") public void createSessionTest() { SessionName name = SessionName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]", "[SESSION]"); Session expectedResponse = Session.newBuilder().setName(name.toString()).build(); mockSpanner.addResponse(expectedResponse); DatabaseName database = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); Session actualResponse = client.createSession(database); Assert.assertEquals(expectedResponse, actualResponse); List<GeneratedMessageV3> actualRequests = mockSpanner.getRequests(); Assert.assertEquals(1, actualRequests.size()); CreateSessionRequest actualRequest = (CreateSessionRequest) actualRequests.get(0); Assert.assertEquals(database, DatabaseName.parse(actualRequest.getDatabase())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
public static List<String> toStringList(List<DatabaseName> values) { List<String> list = new ArrayList<String>(values.size()); for (DatabaseName value : values) { if (value == null) { list.add(""); } else { list.add(value.toString()); } } return list; }
@Test @SuppressWarnings("all") public void listSessionsExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockSpanner.addException(exception); try { String formattedDatabase = DatabaseName.format("[PROJECT]", "[INSTANCE]", "[DATABASE]"); client.listSessions(formattedDatabase); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception } }
public DatabaseName build() { return new DatabaseName(this); } }
public String getFieldValue(String fieldName) { return getFieldValuesMap().get(fieldName); }
public static DatabaseName of(String project, String instance, String database) { return newBuilder().setProject(project).setInstance(instance).setDatabase(database).build(); }
.setDatabase(database == null ? null : database.toString()) .build(); return createSession(request);
@Test @SuppressWarnings("all") public void listSessionsTest() { String nextPageToken = ""; Session sessionsElement = Session.newBuilder().build(); List<Session> sessions = Arrays.asList(sessionsElement); ListSessionsResponse expectedResponse = ListSessionsResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllSessions(sessions) .build(); mockSpanner.addResponse(expectedResponse); String formattedDatabase = DatabaseName.format("[PROJECT]", "[INSTANCE]", "[DATABASE]"); ListSessionsPagedResponse pagedListResponse = client.listSessions(formattedDatabase); List<Session> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getSessionsList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockSpanner.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListSessionsRequest actualRequest = (ListSessionsRequest) actualRequests.get(0); Assert.assertEquals(formattedDatabase, actualRequest.getDatabase()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
.setDatabase(database == null ? null : database.toString()) .build(); return createSession(request);
@Test @SuppressWarnings("all") public void createSessionExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(Status.INVALID_ARGUMENT); mockSpanner.addException(exception); try { DatabaseName database = DatabaseName.of("[PROJECT]", "[INSTANCE]", "[DATABASE]"); client.createSession(database); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception } }