"""The tests for the Vultr component."""
import unittest
import requests_mock

from copy import deepcopy
from homeassistant import setup
import homeassistant.components.vultr as vultr

from tests.common import (
    get_test_home_assistant, load_fixture)

VALID_CONFIG = {
    'vultr': {
        'api_key': 'ABCDEFG1234567'
    }
}


class TestVultr(unittest.TestCase):
    """Tests the Vultr component."""

    def setUp(self):
        """Initialize values for this test case class."""
        self.hass = get_test_home_assistant()
        self.config = VALID_CONFIG

    def tearDown(self):  # pylint: disable=invalid-name
        """Stop everything that we started."""
        self.hass.stop()

    @requests_mock.Mocker()
    def test_setup(self, mock):
        """Test successful setup."""
        mock.get(
            'https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567',
            text=load_fixture('vultr_account_info.json'))
        mock.get(
            'https://api.vultr.com/v1/server/list?api_key=ABCDEFG1234567',
            text=load_fixture('vultr_server_list.json'))

        response = vultr.setup(self.hass, self.config)
        self.assertTrue(response)

    def test_setup_no_api_key(self):
        """Test failed setup with missing API Key."""
        conf = deepcopy(self.config)
        del conf['vultr']['api_key']
        assert not setup.setup_component(self.hass, vultr.DOMAIN, conf)