class RackspaceListServersTest(unittest.TestCase): def setUp(self): self.api = mock.Mock(spec=RackspaceApi) self.command = RackspaceListServers(rackspace_api=self.api) def test_outputs_servers(self): progress = StringIO() server1 = {'id': '0', 'name': 'server1', 'public_ipv4': '50.50.50.50'} server2 = {'id': '1', 'name': 'server2', 'public_ipv4': '51.51.51.51'} self.api.list_servers.return_value = [server1, server2] self.command.execute(progress=progress) self.assertEquals([ '{0}{1}{2}'.format(server1['id'].ljust(36 + 5), server1['name'].ljust(20), server1['public_ipv4']), '{0}{1}{2}'.format(server2['id'].ljust(36 + 5), server2['name'].ljust(20), server2['public_ipv4']), ], progress.getvalue().splitlines())
def setUp(self): self.api = mock.Mock(spec=RackspaceApi) self.command = RackspaceListServers(rackspace_api=self.api)