fixed unit test

This commit is contained in:
Adrian Cole 2010-05-03 01:12:17 -07:00
parent a1c7ccfd3f
commit 98cae35ec6
1 changed files with 4 additions and 2 deletions

View File

@ -39,12 +39,14 @@ public class ServerToNodeMetadataTest {
expect(server.getId()).andReturn(10000).atLeastOnce(); expect(server.getId()).andReturn(10000).atLeastOnce();
expect(server.getName()).andReturn("adriancole-cloudservers-ea3").atLeastOnce(); expect(server.getName()).andReturn("adriancole-cloudservers-ea3").atLeastOnce();
expect(server.getHostId()).andReturn("AHOST").atLeastOnce();
expect(server.getMetadata()).andReturn(ImmutableMap.<String, String> of()).atLeastOnce(); expect(server.getMetadata()).andReturn(ImmutableMap.<String, String> of()).atLeastOnce();
expect(server.getStatus()).andReturn(ServerStatus.ACTIVE).atLeastOnce(); expect(server.getStatus()).andReturn(ServerStatus.ACTIVE).atLeastOnce();
expect(serverStateToNodeState.get(ServerStatus.ACTIVE)).andReturn(NodeState.RUNNING); expect(serverStateToNodeState.get(ServerStatus.ACTIVE)).andReturn(NodeState.RUNNING);
Location location = new LocationImpl(LocationScope.ZONE, "dallas", "description", null); Location provider = new LocationImpl(LocationScope.ZONE, "dallas", "description", null);
Location location = new LocationImpl(LocationScope.HOST, "AHOST", "AHOST", provider);
Addresses addresses = createMock(Addresses.class); Addresses addresses = createMock(Addresses.class);
expect(server.getAddresses()).andReturn(addresses).atLeastOnce(); expect(server.getAddresses()).andReturn(addresses).atLeastOnce();
@ -68,7 +70,7 @@ public class ServerToNodeMetadataTest {
replay(images); replay(images);
ServerToNodeMetadata parser = new ServerToNodeMetadata(serverStateToNodeState, images, ServerToNodeMetadata parser = new ServerToNodeMetadata(serverStateToNodeState, images,
location); provider);
NodeMetadata metadata = parser.apply(server); NodeMetadata metadata = parser.apply(server);
assertEquals(metadata.getLocation(), location); assertEquals(metadata.getLocation(), location);