433 lines
15 KiB
Python
433 lines
15 KiB
Python
"""The tests for the Radarr platform."""
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
import homeassistant.components.radarr.sensor as radarr
|
|
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
def mocked_exception(*args, **kwargs):
|
|
"""Mock exception thrown by requests.get."""
|
|
raise OSError
|
|
|
|
|
|
def mocked_requests_get(*args, **kwargs):
|
|
"""Mock requests.get invocations."""
|
|
class MockResponse:
|
|
"""Class to represent a mocked response."""
|
|
|
|
def __init__(self, json_data, status_code):
|
|
"""Initialize the mock response class."""
|
|
self.json_data = json_data
|
|
self.status_code = status_code
|
|
|
|
def json(self):
|
|
"""Return the json of the response."""
|
|
return self.json_data
|
|
|
|
url = str(args[0])
|
|
if 'api/calendar' in url:
|
|
return MockResponse([
|
|
{
|
|
"title": "Resident Evil",
|
|
"sortTitle": "resident evil final chapter",
|
|
"sizeOnDisk": 0,
|
|
"status": "announced",
|
|
"overview": "Alice, Jill, Claire, Chris, Leon, Ada, and...",
|
|
"inCinemas": "2017-01-25T00:00:00Z",
|
|
"physicalRelease": "2017-01-27T00:00:00Z",
|
|
"images": [
|
|
{
|
|
"coverType": "poster",
|
|
"url": ("/radarr/MediaCover/12/poster.jpg"
|
|
"?lastWrite=636208663600000000")
|
|
},
|
|
{
|
|
"coverType": "banner",
|
|
"url": ("/radarr/MediaCover/12/banner.jpg"
|
|
"?lastWrite=636208663600000000")
|
|
}
|
|
],
|
|
"website": "",
|
|
"downloaded": "false",
|
|
"year": 2017,
|
|
"hasFile": "false",
|
|
"youTubeTrailerId": "B5yxr7lmxhg",
|
|
"studio": "Impact Pictures",
|
|
"path": "/path/to/Resident Evil The Final Chapter (2017)",
|
|
"profileId": 3,
|
|
"monitored": "false",
|
|
"runtime": 106,
|
|
"lastInfoSync": "2017-01-24T14:52:40.315434Z",
|
|
"cleanTitle": "residentevilfinalchapter",
|
|
"imdbId": "tt2592614",
|
|
"tmdbId": 173897,
|
|
"titleSlug": "resident-evil-the-final-chapter-2017",
|
|
"genres": [
|
|
"Action",
|
|
"Horror",
|
|
"Science Fiction"
|
|
],
|
|
"tags": [],
|
|
"added": "2017-01-24T14:52:39.989964Z",
|
|
"ratings": {
|
|
"votes": 363,
|
|
"value": 4.3
|
|
},
|
|
"alternativeTitles": [
|
|
"Resident Evil: Rising"
|
|
],
|
|
"qualityProfileId": 3,
|
|
"id": 12
|
|
}
|
|
], 200)
|
|
if 'api/command' in url:
|
|
return MockResponse([
|
|
{
|
|
"name": "RescanMovie",
|
|
"startedOn": "0001-01-01T00:00:00Z",
|
|
"stateChangeTime": "2014-02-05T05:09:09.2366139Z",
|
|
"sendUpdatesToClient": "true",
|
|
"state": "pending",
|
|
"id": 24
|
|
}
|
|
], 200)
|
|
if 'api/movie' in url:
|
|
return MockResponse([
|
|
{
|
|
"title": "Assassin's Creed",
|
|
"sortTitle": "assassins creed",
|
|
"sizeOnDisk": 0,
|
|
"status": "released",
|
|
"overview": "Lynch discovers he is a descendant of...",
|
|
"inCinemas": "2016-12-21T00:00:00Z",
|
|
"images": [
|
|
{
|
|
"coverType": "poster",
|
|
"url": ("/radarr/MediaCover/1/poster.jpg"
|
|
"?lastWrite=636200219330000000")
|
|
},
|
|
{
|
|
"coverType": "banner",
|
|
"url": ("/radarr/MediaCover/1/banner.jpg"
|
|
"?lastWrite=636200219340000000")
|
|
}
|
|
],
|
|
"website": "https://www.ubisoft.com/en-US/",
|
|
"downloaded": "false",
|
|
"year": 2016,
|
|
"hasFile": "false",
|
|
"youTubeTrailerId": "pgALJgMjXN4",
|
|
"studio": "20th Century Fox",
|
|
"path": "/path/to/Assassin's Creed (2016)",
|
|
"profileId": 6,
|
|
"monitored": "true",
|
|
"runtime": 115,
|
|
"lastInfoSync": "2017-01-23T22:05:32.365337Z",
|
|
"cleanTitle": "assassinscreed",
|
|
"imdbId": "tt2094766",
|
|
"tmdbId": 121856,
|
|
"titleSlug": "assassins-creed-121856",
|
|
"genres": [
|
|
"Action",
|
|
"Adventure",
|
|
"Fantasy",
|
|
"Science Fiction"
|
|
],
|
|
"tags": [],
|
|
"added": "2017-01-14T20:18:52.938244Z",
|
|
"ratings": {
|
|
"votes": 711,
|
|
"value": 5.2
|
|
},
|
|
"alternativeTitles": [
|
|
"Assassin's Creed: The IMAX Experience"
|
|
],
|
|
"qualityProfileId": 6,
|
|
"id": 1
|
|
}
|
|
], 200)
|
|
if 'api/diskspace' in url:
|
|
return MockResponse([
|
|
{
|
|
"path": "/data",
|
|
"label": "",
|
|
"freeSpace": 282500067328,
|
|
"totalSpace": 499738734592
|
|
}
|
|
], 200)
|
|
if 'api/system/status' in url:
|
|
return MockResponse({
|
|
"version": "0.2.0.210",
|
|
"buildTime": "2017-01-22T23:12:49Z",
|
|
"isDebug": "false",
|
|
"isProduction": "true",
|
|
"isAdmin": "false",
|
|
"isUserInteractive": "false",
|
|
"startupPath": "/path/to/radarr",
|
|
"appData": "/path/to/radarr/data",
|
|
"osVersion": "4.8.13.1",
|
|
"isMonoRuntime": "true",
|
|
"isMono": "true",
|
|
"isLinux": "true",
|
|
"isOsx": "false",
|
|
"isWindows": "false",
|
|
"branch": "develop",
|
|
"authentication": "forms",
|
|
"sqliteVersion": "3.16.2",
|
|
"urlBase": "",
|
|
"runtimeVersion": ("4.6.1 "
|
|
"(Stable 4.6.1.3/abb06f1 "
|
|
"Mon Oct 3 07:57:59 UTC 2016)")
|
|
}, 200)
|
|
return MockResponse({
|
|
"error": "Unauthorized"
|
|
}, 401)
|
|
|
|
|
|
class TestRadarrSetup(unittest.TestCase):
|
|
"""Test the Radarr platform."""
|
|
|
|
# pylint: disable=invalid-name
|
|
DEVICES = []
|
|
|
|
def add_entities(self, devices, update):
|
|
"""Mock add devices."""
|
|
for device in devices:
|
|
self.DEVICES.append(device)
|
|
|
|
def setUp(self):
|
|
"""Initialize values for this testcase class."""
|
|
self.DEVICES = []
|
|
self.hass = get_test_home_assistant()
|
|
self.hass.config.time_zone = 'America/Los_Angeles'
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_diskspace_no_paths(self, req_mock):
|
|
"""Test getting all disk space."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [],
|
|
'monitored_conditions': [
|
|
'diskspace'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert '263.10' == device.state
|
|
assert 'mdi:harddisk' == device.icon
|
|
assert 'GB' == device.unit_of_measurement
|
|
assert 'Radarr Disk Space' == device.name
|
|
assert '263.10/465.42GB (56.53%)' == \
|
|
device.device_state_attributes["/data"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_diskspace_paths(self, req_mock):
|
|
"""Test getting diskspace for included paths."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'diskspace'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert '263.10' == device.state
|
|
assert 'mdi:harddisk' == device.icon
|
|
assert 'GB' == device.unit_of_measurement
|
|
assert 'Radarr Disk Space' == device.name
|
|
assert '263.10/465.42GB (56.53%)' == \
|
|
device.device_state_attributes["/data"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_commands(self, req_mock):
|
|
"""Test getting running commands."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'commands'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert 1 == device.state
|
|
assert 'mdi:code-braces' == device.icon
|
|
assert 'Commands' == device.unit_of_measurement
|
|
assert 'Radarr Commands' == device.name
|
|
assert 'pending' == \
|
|
device.device_state_attributes["RescanMovie"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_movies(self, req_mock):
|
|
"""Test getting the number of movies."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'movies'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert 1 == device.state
|
|
assert 'mdi:television' == device.icon
|
|
assert 'Movies' == device.unit_of_measurement
|
|
assert 'Radarr Movies' == device.name
|
|
assert 'false' == \
|
|
device.device_state_attributes["Assassin's Creed (2016)"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_upcoming_multiple_days(self, req_mock):
|
|
"""Test the upcoming movies for multiple days."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'upcoming'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert 1 == device.state
|
|
assert 'mdi:television' == device.icon
|
|
assert 'Movies' == device.unit_of_measurement
|
|
assert 'Radarr Upcoming' == device.name
|
|
assert '2017-01-27T00:00:00Z' == \
|
|
device.device_state_attributes["Resident Evil (2017)"]
|
|
|
|
@pytest.mark.skip
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_upcoming_today(self, req_mock):
|
|
"""Test filtering for a single day.
|
|
|
|
Radarr needs to respond with at least 2 days.
|
|
"""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '1',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'upcoming'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert 1 == device.state
|
|
assert 'mdi:television' == device.icon
|
|
assert 'Movies' == device.unit_of_measurement
|
|
assert 'Radarr Upcoming' == device.name
|
|
assert '2017-01-27T00:00:00Z' == \
|
|
device.device_state_attributes["Resident Evil (2017)"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_system_status(self, req_mock):
|
|
"""Test the getting of the system status."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '2',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'status'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert '0.2.0.210' == device.state
|
|
assert 'mdi:information' == device.icon
|
|
assert 'Radarr Status' == device.name
|
|
assert '4.8.13.1' == device.device_state_attributes['osVersion']
|
|
|
|
@pytest.mark.skip
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
|
|
def test_ssl(self, req_mock):
|
|
"""Test SSL being enabled."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '1',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'upcoming'
|
|
],
|
|
"ssl": "true"
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert 1 == device.state
|
|
assert 's' == device.ssl
|
|
assert 'mdi:television' == device.icon
|
|
assert 'Movies' == device.unit_of_measurement
|
|
assert 'Radarr Upcoming' == device.name
|
|
assert '2017-01-27T00:00:00Z' == \
|
|
device.device_state_attributes["Resident Evil (2017)"]
|
|
|
|
@unittest.mock.patch('requests.get', side_effect=mocked_exception)
|
|
def test_exception_handling(self, req_mock):
|
|
"""Test exception being handled."""
|
|
config = {
|
|
'platform': 'radarr',
|
|
'api_key': 'foo',
|
|
'days': '1',
|
|
'unit': 'GB',
|
|
"include_paths": [
|
|
'/data'
|
|
],
|
|
'monitored_conditions': [
|
|
'upcoming'
|
|
]
|
|
}
|
|
radarr.setup_platform(self.hass, config, self.add_entities, None)
|
|
for device in self.DEVICES:
|
|
device.update()
|
|
assert device.state is None
|