@Test public void testNodeSuspendedReturnsTrueWhenSuspended() { expect(node.getStatus()).andReturn(Status.SUSPENDED).atLeastOnce(); expect(node.getBackendStatus()).andReturn(null).atLeastOnce(); replay(node, computeService); AtomicNodeSuspended nodeSuspended = new AtomicNodeSuspended(computeService); AtomicReference<NodeMetadata> reference = Atomics.newReference(node); Assert.assertTrue(nodeSuspended.apply(reference)); Assert.assertEquals(reference.get(), node); }