Updating EC2 test mocks to match changes in 79c289d

This commit is contained in:
Andrew Phillips 2014-05-18 12:42:32 -04:00
parent 79c289da9d
commit 5239e0fea1
1 changed files with 2 additions and 2 deletions

View File

@ -113,7 +113,7 @@ public class EC2CreateNodesInGroupThenAddToSetTest {
expect(instance.getId()).andReturn(instanceCreatedId).atLeastOnce();
// simulate a lazy credentials fetch
LoginCredentials creds = LoginCredentials.builder().user("foo").privateKey("bar").build();
expect(strategy.instanceToCredentials.apply(instance)).andReturn(Optional.of(creds));
expect(strategy.instanceToCredentials.getUnchecked(instance)).andReturn(Optional.of(creds));
expect(instance.getRegion()).andReturn(region).atLeastOnce();
expect(strategy.credentialStore.put("node#" + region + "/" + instanceCreatedId, creds)).andReturn(null);
@ -212,7 +212,7 @@ public class EC2CreateNodesInGroupThenAddToSetTest {
expect(instance.getId()).andReturn(instanceCreatedId).atLeastOnce();
// simulate a lazy credentials fetch
LoginCredentials creds = LoginCredentials.builder().user("foo").privateKey("bar").build();
expect(strategy.instanceToCredentials.apply(instance)).andReturn(Optional.of(creds));
expect(strategy.instanceToCredentials.getUnchecked(instance)).andReturn(Optional.of(creds));
expect(instance.getRegion()).andReturn(region).atLeastOnce();
expect(strategy.credentialStore.put("node#" + region + "/" + instanceCreatedId, creds)).andReturn(null);