def test_add_console(self):
        instance = self._create_instance()
        self.console.add_console(self.context, instance["id"])
        instance = db.instance_get(self.context, instance["id"])
        pool = db.console_pool_get_by_host_type(
            self.context, instance["host"], self.console.host, self.console.driver.console_type
        )

        console_instances = [con["instance_uuid"] for con in pool["consoles"]]
        self.assert_(instance["uuid"] in console_instances)
        db.instance_destroy(self.context, instance["uuid"])

    def test_add_console(self):
        instance = self._create_instance()
        self.console.add_console(self.context, instance['id'])
        instance = db.instance_get(self.context, instance['id'])
        pool = db.console_pool_get_by_host_type(self.context,
                instance['host'], self.console.host,
                self.console.driver.console_type)

        console_instances = [con['instance_uuid'] for con in pool['consoles']]
        self.assert_(instance['uuid'] in console_instances)
        db.instance_destroy(self.context, instance['uuid'])

    def test_add_console(self, mock_get):
        mock_get.return_value = self.pool_info
        instance = self._create_instance()
        self.console.add_console(self.context, instance['id'])
        instance = db.instance_get(self.context, instance['id'])
        pool = db.console_pool_get_by_host_type(self.context,
                instance['host'], self.console.host,
                self.console.driver.console_type)

        console_instances = [con['instance_uuid'] for con in pool['consoles']]
        self.assertIn(instance['uuid'], console_instances)
        db.instance_destroy(self.context, instance['uuid'])

 def test_get_pool_creates_new_pool_if_needed(self):
     self.assertRaises(
         exception.NotFound,
         db.console_pool_get_by_host_type,
         self.context,
         self.host,
         self.console.host,
         self.console.driver.console_type,
     )
     pool = self.console.get_pool_for_instance_host(self.context, self.host)
     pool2 = db.console_pool_get_by_host_type(
         self.context, self.host, self.console.host, self.console.driver.console_type
     )
     self.assertEqual(pool["id"], pool2["id"])

 def test_get_pool_creates_new_pool_if_needed(self, mock_get):
     mock_get.return_value = self.pool_info
     self.assertRaises(exception.NotFound,
                       db.console_pool_get_by_host_type,
                       self.context,
                       self.host,
                       self.console.host,
                       self.console.driver.console_type)
     pool = self.console._get_pool_for_instance_host(self.context,
                                                        self.host)
     pool2 = db.console_pool_get_by_host_type(self.context,
                           self.host,
                           self.console.host,
                           self.console.driver.console_type)
     self.assertEqual(pool['id'], pool2['id'])