Add platform and sensors for Vultr VPS (#9928)
* Initial commit of Vultr components Have a working Vultr hub and binary sensor which pulls down the following attributes of your VPS: - Date created - Subscription id (server id) - Cost per month (in US$) - Operating System installed - IPv4 address - label (human readable name) - region - number of vcpus - which storage package chosen - IPV6 address (if applicable) - RAM amount Working next on sensor and then testing / coverage. * Added Vultr sensor for pending charges and current bandwidth. Refactored binary_sensor and hub too * Corrected is_on bases * Added basic tests for Vultr binary & platform * Updated require files * Changing test fixture to highlight different cases * Written basic test for sensor.vultr * Resolved linting errors and broken test * Increase test coverage and corrected docs * Resolved hound issues * Revert back negative binary test * Another hound resolve * Refactoring and adding is switch, moving over to vultr branch * Made Vultr components more resiliant to invalid configs * Added negetive test for vultr binary sensor * Added better testing of vultr sensor * Resolved vultr platform test affecting subsequent vultr tests * Moving VULTR components to single use design * Added in sensor name config * Added missing sensors var * Resolved init data setting of sensors, added in name conf to switch * Made the Vultr component more resiliant to startup failure with better alerting * Various Vultr component changes - Refactored sensor, binary_sensor, and switch to reference one subscription - Renamed CURRENT_BANDWIDTH_GB monitored condition to CURRENT_BANDWIDTH_USED - Improved test coverage * Resolved local tox linting issue * Added more testing for Vultr switch * Improved test coverage for Vultr components * Made PR comment changes to vultr binary sensor * Made PR comment changes to Vultr sensor * resolved PR comments for Vultr Switch * Resolved vultr sensor name and improved tests * Improved Vultr switch testing (default name formatting) * Removed vultr hub failure checking
This commit is contained in:
Adam Cooper
Fabian Affolter
parent
a5d5f3f727
commit
72ce9ec321
13 changed files with 1139 additions and 1 deletions
165
tests/components/sensor/test_vultr.py
Normal file
165
tests/components/sensor/test_vultr.py
Normal file
|
|
@ -0,0 +1,165 @@
|
|||
"""The tests for the Vultr sensor platform."""
|
||||
import pytest
|
||||
import unittest
|
||||
import requests_mock
|
||||
import voluptuous as vol
|
||||
|
||||
from components.sensor import vultr
|
||||
from components import vultr as base_vultr
|
||||
from components.vultr import CONF_SUBSCRIPTION
|
||||
from homeassistant.const import (
|
||||
CONF_NAME, CONF_MONITORED_CONDITIONS, CONF_PLATFORM)
|
||||
|
||||
from tests.components.test_vultr import VALID_CONFIG
|
||||
from tests.common import (
|
||||
get_test_home_assistant, load_fixture)
|
||||
|
||||
|
||||
class TestVultrSensorSetup(unittest.TestCase):
|
||||
"""Test the Vultr platform."""
|
||||
|
||||
DEVICES = []
|
||||
|
||||
def add_devices(self, devices, action):
|
||||
"""Mock add devices."""
|
||||
for device in devices:
|
||||
self.DEVICES.append(device)
|
||||
|
||||
def setUp(self):
|
||||
"""Initialize values for this testcase class."""
|
||||
self.hass = get_test_home_assistant()
|
||||
self.configs = [
|
||||
{
|
||||
CONF_NAME: vultr.DEFAULT_NAME,
|
||||
CONF_SUBSCRIPTION: '576965',
|
||||
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'
|
||||
]
|
||||
}
|
||||
]
|
||||
|
||||
def tearDown(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
@requests_mock.Mocker()
|
||||
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'))
|
||||
|
||||
mock.get(
|
||||
'https://api.vultr.com/v1/server/list?api_key=ABCDEFG1234567',
|
||||
text=load_fixture('vultr_server_list.json'))
|
||||
|
||||
base_vultr.setup(self.hass, VALID_CONFIG)
|
||||
|
||||
for config in self.configs:
|
||||
setup = vultr.setup_platform(self.hass,
|
||||
config,
|
||||
self.add_devices,
|
||||
None)
|
||||
|
||||
self.assertIsNone(setup)
|
||||
|
||||
self.assertEqual(5, len(self.DEVICES))
|
||||
|
||||
tested = 0
|
||||
|
||||
for device in self.DEVICES:
|
||||
|
||||
# Test pre update
|
||||
if device.subscription == '576965':
|
||||
self.assertEqual(vultr.DEFAULT_NAME, device.name)
|
||||
|
||||
device.update()
|
||||
|
||||
if device.unit_of_measurement == 'GB': # Test Bandwidth Used
|
||||
if device.subscription == '576965':
|
||||
self.assertEqual(
|
||||
'Vultr my new server Current Bandwidth Used',
|
||||
device.name)
|
||||
self.assertEqual('mdi:chart-histogram', device.icon)
|
||||
self.assertEqual(131.51, device.state)
|
||||
self.assertEqual('mdi:chart-histogram', device.icon)
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '123456':
|
||||
self.assertEqual('Server Current Bandwidth Used',
|
||||
device.name)
|
||||
self.assertEqual(957.46, device.state)
|
||||
tested += 1
|
||||
|
||||
elif device.unit_of_measurement == 'US$': # Test Pending Charges
|
||||
|
||||
if device.subscription == '576965': # Default 'Vultr {} {}'
|
||||
self.assertEqual('Vultr my new server Pending Charges',
|
||||
device.name)
|
||||
self.assertEqual('mdi:currency-usd', device.icon)
|
||||
self.assertEqual(46.67, device.state)
|
||||
self.assertEqual('mdi:currency-usd', device.icon)
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '123456': # Custom name with 1 {}
|
||||
self.assertEqual('Server Pending Charges', device.name)
|
||||
self.assertEqual('not a number', device.state)
|
||||
tested += 1
|
||||
|
||||
elif device.subscription == '555555': # No {} in name
|
||||
self.assertEqual('VPS Charges', device.name)
|
||||
self.assertEqual(5.45, device.state)
|
||||
tested += 1
|
||||
|
||||
self.assertEqual(tested, 5)
|
||||
|
||||
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
|
||||
})
|
||||
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',
|
||||
]
|
||||
})
|
||||
|
||||
@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'))
|
||||
|
||||
mock.get(
|
||||
'https://api.vultr.com/v1/server/list?api_key=ABCDEFG1234567',
|
||||
text=load_fixture('vultr_server_list.json'))
|
||||
|
||||
base_vultr.setup(self.hass, VALID_CONFIG)
|
||||
|
||||
bad_conf = {
|
||||
CONF_MONITORED_CONDITIONS: vultr.MONITORED_CONDITIONS,
|
||||
} # No subs at all
|
||||
|
||||
no_sub_setup = vultr.setup_platform(self.hass,
|
||||
bad_conf,
|
||||
self.add_devices,
|
||||
None)
|
||||
|
||||
self.assertIsNotNone(no_sub_setup)
|
||||
self.assertEqual(0, len(self.DEVICES))
|
||||
Loading…
Add table
Add a link
Reference in a new issue