@Test public void testRetryableException() { ResourceManagerRpcFactory rpcFactoryMock = EasyMock.createMock(ResourceManagerRpcFactory.class); ResourceManagerRpc resourceManagerRpcMock = EasyMock.createMock(ResourceManagerRpc.class); EasyMock.expect(rpcFactoryMock.create(EasyMock.anyObject(ResourceManagerOptions.class))) .andReturn(resourceManagerRpcMock); EasyMock.replay(rpcFactoryMock); ResourceManager resourceManagerMock = ResourceManagerOptions.newBuilder() .setServiceRpcFactory(rpcFactoryMock) .build() .getService(); EasyMock.expect(resourceManagerRpcMock.get(PARTIAL_PROJECT.getProjectId(), EMPTY_RPC_OPTIONS)) .andThrow(new ResourceManagerException(500, "Internal Error")) .andReturn(PARTIAL_PROJECT.toPb()); EasyMock.replay(resourceManagerRpcMock); Project returnedProject = resourceManagerMock.get(PARTIAL_PROJECT.getProjectId()); assertEquals( new Project(resourceManagerMock, new ProjectInfo.BuilderImpl(PARTIAL_PROJECT)), returnedProject); }
public Builder toBuilder() { return new BuilderImpl(this); }
public static Builder builder(String id) { return new BuilderImpl(id); }
Builder(Project project) { this.resourceManager = project.resourceManager; this.infoBuilder = new ProjectInfo.BuilderImpl(project); }
static Project fromPb(ResourceManager resourceManager, com.google.api.services.cloudresourcemanager.model.Project answer) { ProjectInfo info = ProjectInfo.fromPb(answer); return new Project(resourceManager, new ProjectInfo.BuilderImpl(info)); } }
@Override public Builder projectId(String projectId) { infoBuilder.projectId(projectId); return this; }
@Override public Builder labels(Map<String, String> labels) { infoBuilder.labels(labels); return this; }
@Override Builder parent(ResourceId parent) { infoBuilder.parent(parent); return this; }
@Override Builder createTimeMillis(Long createTimeMillis) { infoBuilder.createTimeMillis(createTimeMillis); return this; }
@Override public Builder clearLabels() { infoBuilder.clearLabels(); return this; }
@Override public Project apply( com.google.api.services.cloudresourcemanager.model.Project projectPb) { return new Project( serviceOptions.service(), new ProjectInfo.BuilderImpl(ProjectInfo.fromPb(projectPb))); } });
@Override public Builder addLabel(String key, String value) { infoBuilder.addLabel(key, value); return this; }
@Override public Builder name(String name) { infoBuilder.name(name); return this; }
@Override public Builder removeLabel(String key) { infoBuilder.removeLabel(key); return this; }
replay(resourceManager); Project.Builder builder = new Project.Builder(new Project(resourceManager, new ProjectInfo.BuilderImpl("wrong-id"))); Project project = builder
private void initializeProject() { project = new Project(resourceManager, new ProjectInfo.BuilderImpl(PROJECT_INFO)); }
public static Builder newBuilder(String id) { return new BuilderImpl(id); }
public Builder toBuilder() { return new BuilderImpl(this); }
@Test public void testReplace() { initializeExpectedProject(2); Project expectedReplacedProject = expectedProject.toBuilder().addLabel("k3", "v3").build(); expect(resourceManager.getOptions()).andReturn(mockOptions).times(2); expect(resourceManager.replace(anyObject(Project.class))).andReturn(expectedReplacedProject); replay(resourceManager); initializeProject(); Project newProject = new Project(resourceManager, new ProjectInfo.BuilderImpl(expectedReplacedProject)); Project actualReplacedProject = newProject.replace(); compareProjectInfos(expectedReplacedProject, actualReplacedProject); }
@Test public void testReload() { initializeExpectedProject(2); ProjectInfo newInfo = PROJECT_INFO.toBuilder().addLabel("k3", "v3").build(); Project expectedProject = new Project(serviceMockReturnsOptions, new ProjectInfo.BuilderImpl(newInfo)); expect(resourceManager.getOptions()).andReturn(mockOptions); expect(resourceManager.get(PROJECT_INFO.getProjectId())).andReturn(expectedProject); replay(resourceManager); initializeProject(); Project newProject = project.reload(); assertEquals(expectedProject, newProject); }