def setUp(self): super(TestEnvCommand, self).setUp() self.m_client.get_all.return_value = [fake_env.get_fake_env() for i in range(10)] self.m_client.get_by_id.return_value = fake_env.get_fake_env() self.m_client.create.return_value = fake_env.get_fake_env() self.m_client.update.return_value = fake_env.get_fake_env()
def test_env_delete_wo_force(self): args = 'env delete 42' env = fake_env.get_fake_env(status='operational') self.m_client.get_by_id.return_value = env with mock.patch('sys.stdout', new=moves.cStringIO()) as m_stdout: self.exec_command(args) self.assertIn('--force', m_stdout.getvalue())