def setUp(self):
        super(TestAggregateRemoveHost, self).setUp()

        self.aggregate_mock.get.return_value = self.fake_ag
        self.aggregate_mock.remove_host.return_value = self.fake_ag
        self.cmd = aggregate.RemoveAggregateHost(self.app, None)
    def setUp(self):
        super(TestAggregateRemoveHost, self).setUp()

        self.sdk_client.find_aggregate.return_value = self.fake_ag
        self.sdk_client.remove_host_from_aggregate.return_value = self.fake_ag
        self.cmd = aggregate.RemoveAggregateHost(self.app, None)