@@ -168,42 +168,32 @@ public void testUndelete() {
168168 }
169169
170170 @ Test
171- public void testReplaceWithProjectInfo () {
171+ public void testReplace () {
172172 initializeExpectedProject (2 );
173- ProjectInfo newInfo = PROJECT_INFO .toBuilder ().addLabel ("k3" , "v3" ).build ();
174- Project expectedProject =
175- new Project (serviceMockReturnsOptions , new ProjectInfo .BuilderImpl (newInfo ));
176- expect (resourceManager .options ()).andReturn (mockOptions );
177- expect (resourceManager .replace (newInfo )).andReturn (expectedProject );
178- replay (resourceManager );
179- initializeProject ();
180- Project newProject = project .replace (newInfo );
181- assertEquals (expectedProject , newProject );
182- }
183-
184- @ Test
185- public void testReplaceWithoutProjectInfo () {
186- initializeExpectedProject (2 );
187- ProjectInfo newInfo = PROJECT_INFO .toBuilder ().addLabel ("k3" , "v3" ).build ();
188- Project expectedProject =
189- new Project (serviceMockReturnsOptions , new ProjectInfo .BuilderImpl (newInfo ));
190- expect (resourceManager .options ()).andReturn (mockOptions );
191- expect (resourceManager .replace (anyObject (Project .class ))).andReturn (expectedProject );
173+ Project expectedReplacedProject = expectedProject .toBuilder ().addLabel ("k3" , "v3" ).build ();
174+ expect (resourceManager .options ()).andReturn (mockOptions ).times (2 );
175+ expect (resourceManager .replace (anyObject (Project .class ))).andReturn (expectedReplacedProject );
192176 replay (resourceManager );
193177 initializeProject ();
194- Project newProject = project .replace (newInfo );
195- assertEquals (expectedProject , newProject );
178+ Project newProject =
179+ new Project (resourceManager , new ProjectInfo .BuilderImpl (expectedReplacedProject ));
180+ Project actualReplacedProject = newProject .replace ();
181+ compareProjectInfos (expectedReplacedProject , actualReplacedProject );
196182 }
197183
198184 private void compareProjects (Project expected , Project value ) {
199185 assertEquals (expected , value );
186+ compareProjectInfos (expected , value );
187+ assertEquals (expected .resourceManager ().options (), value .resourceManager ().options ());
188+ }
189+
190+ private void compareProjectInfos (ProjectInfo expected , ProjectInfo value ) {
200191 assertEquals (expected .projectId (), value .projectId ());
201192 assertEquals (expected .name (), value .name ());
202193 assertEquals (expected .labels (), value .labels ());
203194 assertEquals (expected .projectNumber (), value .projectNumber ());
204195 assertEquals (expected .createTimeMillis (), value .createTimeMillis ());
205196 assertEquals (expected .state (), value .state ());
206197 assertEquals (expected .parent (), value .parent ());
207- assertEquals (expected .resourceManager ().options (), value .resourceManager ().options ());
208198 }
209199}
0 commit comments