mirror of
https://github.com/home-assistant/core.git
synced 2025-12-25 13:38:04 +00:00
Black
This commit is contained in:
Paulus Schoutsen
@@ -10,12 +10,10 @@ import voluptuous as vol
|
||||
import homeassistant.components.vultr.sensor as vultr
|
||||
from homeassistant.components import vultr as base_vultr
|
||||
from homeassistant.components.vultr import CONF_SUBSCRIPTION
|
||||
from homeassistant.const import (
|
||||
CONF_NAME, CONF_MONITORED_CONDITIONS, CONF_PLATFORM)
|
||||
from homeassistant.const import CONF_NAME, CONF_MONITORED_CONDITIONS, CONF_PLATFORM
|
||||
|
||||
from tests.components.vultr.test_init import VALID_CONFIG
|
||||
from tests.common import (
|
||||
get_test_home_assistant, load_fixture)
|
||||
from tests.common import get_test_home_assistant, load_fixture
|
||||
|
||||
|
||||
class TestVultrSensorSetup(unittest.TestCase):
|
||||
@@ -34,21 +32,19 @@ class TestVultrSensorSetup(unittest.TestCase):
|
||||
self.configs = [
|
||||
{
|
||||
CONF_NAME: vultr.DEFAULT_NAME,
|
||||
CONF_SUBSCRIPTION: '576965',
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS
|
||||
CONF_SUBSCRIPTION: "576965",
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS,
|
||||
},
|
||||
{
|
||||
CONF_NAME: 'Server {}',
|
||||
CONF_SUBSCRIPTION: '123456',
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS
|
||||
CONF_NAME: "Server {}",
|
||||
CONF_SUBSCRIPTION: "123456",
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS,
|
||||
},
|
||||
{
|
||||
CONF_NAME: 'VPS Charges',
|
||||
CONF_SUBSCRIPTION: '555555',
|
||||
CONF_MONITORED_CONDITIONS: [
|
||||
'pending_charges'
|
||||
]
|
||||
}
|
||||
CONF_NAME: "VPS Charges",
|
||||
CONF_SUBSCRIPTION: "555555",
|
||||
CONF_MONITORED_CONDITIONS: ["pending_charges"],
|
||||
},
|
||||
]
|
||||
|
||||
def tearDown(self):
|
||||
@@ -59,19 +55,19 @@ class TestVultrSensorSetup(unittest.TestCase):
|
||||
def test_sensor(self, mock):
|
||||
"""Test the Vultr sensor class and methods."""
|
||||
mock.get(
|
||||
'https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567',
|
||||
text=load_fixture('vultr_account_info.json'))
|
||||
"https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567",
|
||||
text=load_fixture("vultr_account_info.json"),
|
||||
)
|
||||
|
||||
with patch(
|
||||
'vultr.Vultr.server_list',
|
||||
return_value=json.loads(
|
||||
load_fixture('vultr_server_list.json'))):
|
||||
"vultr.Vultr.server_list",
|
||||
return_value=json.loads(load_fixture("vultr_server_list.json")),
|
||||
):
|
||||
# Setup hub
|
||||
base_vultr.setup(self.hass, VALID_CONFIG)
|
||||
|
||||
for config in self.configs:
|
||||
setup = vultr.setup_platform(
|
||||
self.hass, config, self.add_entities, None)
|
||||
setup = vultr.setup_platform(self.hass, config, self.add_entities, None)
|
||||
|
||||
assert setup is None
|
||||
|
||||
@@ -82,43 +78,40 @@ class TestVultrSensorSetup(unittest.TestCase):
|
||||
for device in self.DEVICES:
|
||||
|
||||
# Test pre update
|
||||
if device.subscription == '576965':
|
||||
if device.subscription == "576965":
|
||||
assert vultr.DEFAULT_NAME == device.name
|
||||
|
||||
device.update()
|
||||
|
||||
if device.unit_of_measurement == 'GB': # Test Bandwidth Used
|
||||
if device.subscription == '576965':
|
||||
assert 'Vultr my new server Current Bandwidth Used' == \
|
||||
device.name
|
||||
assert 'mdi:chart-histogram' == device.icon
|
||||
if device.unit_of_measurement == "GB": # Test Bandwidth Used
|
||||
if device.subscription == "576965":
|
||||
assert "Vultr my new server Current Bandwidth Used" == device.name
|
||||
assert "mdi:chart-histogram" == device.icon
|
||||
assert 131.51 == device.state
|
||||
assert 'mdi:chart-histogram' == device.icon
|
||||
assert "mdi:chart-histogram" == device.icon
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '123456':
|
||||
assert 'Server Current Bandwidth Used' == \
|
||||
device.name
|
||||
elif device.subscription == "123456":
|
||||
assert "Server Current Bandwidth Used" == device.name
|
||||
assert 957.46 == device.state
|
||||
tested += 1
|
||||
|
||||
elif device.unit_of_measurement == 'US$': # Test Pending Charges
|
||||
elif device.unit_of_measurement == "US$": # Test Pending Charges
|
||||
|
||||
if device.subscription == '576965': # Default 'Vultr {} {}'
|
||||
assert 'Vultr my new server Pending Charges' == \
|
||||
device.name
|
||||
assert 'mdi:currency-usd' == device.icon
|
||||
if device.subscription == "576965": # Default 'Vultr {} {}'
|
||||
assert "Vultr my new server Pending Charges" == device.name
|
||||
assert "mdi:currency-usd" == device.icon
|
||||
assert 46.67 == device.state
|
||||
assert 'mdi:currency-usd' == device.icon
|
||||
assert "mdi:currency-usd" == device.icon
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '123456': # Custom name with 1 {}
|
||||
assert 'Server Pending Charges' == device.name
|
||||
assert 'not a number' == device.state
|
||||
elif device.subscription == "123456": # Custom name with 1 {}
|
||||
assert "Server Pending Charges" == device.name
|
||||
assert "not a number" == device.state
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '555555': # No {} in name
|
||||
assert 'VPS Charges' == device.name
|
||||
elif device.subscription == "555555": # No {} in name
|
||||
assert "VPS Charges" == device.name
|
||||
assert 5.45 == device.state
|
||||
tested += 1
|
||||
|
||||
@@ -127,39 +120,43 @@ class TestVultrSensorSetup(unittest.TestCase):
|
||||
def test_invalid_sensor_config(self):
|
||||
"""Test config type failures."""
|
||||
with pytest.raises(vol.Invalid): # No subscription
|
||||
vultr.PLATFORM_SCHEMA({
|
||||
CONF_PLATFORM: base_vultr.DOMAIN,
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS
|
||||
})
|
||||
vultr.PLATFORM_SCHEMA(
|
||||
{
|
||||
CONF_PLATFORM: base_vultr.DOMAIN,
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS,
|
||||
}
|
||||
)
|
||||
with pytest.raises(vol.Invalid): # Bad monitored_conditions
|
||||
vultr.PLATFORM_SCHEMA({
|
||||
CONF_PLATFORM: base_vultr.DOMAIN,
|
||||
CONF_SUBSCRIPTION: '123456',
|
||||
CONF_MONITORED_CONDITIONS: [
|
||||
'non-existent-condition',
|
||||
]
|
||||
})
|
||||
vultr.PLATFORM_SCHEMA(
|
||||
{
|
||||
CONF_PLATFORM: base_vultr.DOMAIN,
|
||||
CONF_SUBSCRIPTION: "123456",
|
||||
CONF_MONITORED_CONDITIONS: ["non-existent-condition"],
|
||||
}
|
||||
)
|
||||
|
||||
@requests_mock.Mocker()
|
||||
def test_invalid_sensors(self, mock):
|
||||
"""Test the VultrSensor fails."""
|
||||
mock.get(
|
||||
'https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567',
|
||||
text=load_fixture('vultr_account_info.json'))
|
||||
"https://api.vultr.com/v1/account/info?api_key=ABCDEFG1234567",
|
||||
text=load_fixture("vultr_account_info.json"),
|
||||
)
|
||||
|
||||
with patch(
|
||||
'vultr.Vultr.server_list',
|
||||
return_value=json.loads(
|
||||
load_fixture('vultr_server_list.json'))):
|
||||
"vultr.Vultr.server_list",
|
||||
return_value=json.loads(load_fixture("vultr_server_list.json")),
|
||||
):
|
||||
# Setup hub
|
||||
base_vultr.setup(self.hass, VALID_CONFIG)
|
||||
|
||||
bad_conf = {
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS,
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS
|
||||
} # No subs at all
|
||||
|
||||
no_sub_setup = vultr.setup_platform(
|
||||
self.hass, bad_conf, self.add_entities, None)
|
||||
self.hass, bad_conf, self.add_entities, None
|
||||
)
|
||||
|
||||
assert no_sub_setup is None
|
||||
assert 0 == len(self.DEVICES)
|
||||
|
||||
Reference in New Issue
Block a user