-
-
Notifications
You must be signed in to change notification settings - Fork 31.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add sort by config and tests for Reddit integration (#22081)
* add sort type to reddit sensor add sort type to reddit sensor automated commit 14/03/2019 15:53:15 automated commit 15/03/2019 09:43:39 reddit sensor add tests and sort by config reddit sensor add tests and sort by config reddit sensor add tests and sort by config reddit sensor add tests and sort by config * Add reddit sensor sort_by config and tests * Add reddit sensor sort_by config and tests * Add reddit sensor sort_by config and tests * Add reddit sensor sort_by config and tests * Add reddit sensor sort_by config and tests
- Loading branch information
Showing
3 changed files
with
204 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| """Tests for the the Reddit component.""" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| """The tests for the Reddit platform.""" | ||
| import copy | ||
| import unittest | ||
| from unittest.mock import patch | ||
|
|
||
| from homeassistant.components.reddit.sensor import ( | ||
| DOMAIN, ATTR_SUBREDDIT, ATTR_POSTS, CONF_SORT_BY, | ||
| ATTR_ID, ATTR_URL, ATTR_TITLE, ATTR_SCORE, ATTR_COMMENTS_NUMBER, | ||
| ATTR_CREATED, ATTR_BODY) | ||
| from homeassistant.const import (CONF_USERNAME, CONF_PASSWORD, CONF_MAXIMUM) | ||
| from homeassistant.setup import setup_component | ||
|
|
||
| from tests.common import (get_test_home_assistant, | ||
| MockDependency) | ||
|
|
||
|
|
||
| VALID_CONFIG = { | ||
| 'sensor': { | ||
| 'platform': DOMAIN, | ||
| 'client_id': 'test_client_id', | ||
| 'client_secret': 'test_client_secret', | ||
| CONF_USERNAME: 'test_username', | ||
| CONF_PASSWORD: 'test_password', | ||
| 'subreddits': ['worldnews', 'news'], | ||
|
|
||
| } | ||
| } | ||
|
|
||
| VALID_LIMITED_CONFIG = { | ||
| 'sensor': { | ||
| 'platform': DOMAIN, | ||
| 'client_id': 'test_client_id', | ||
| 'client_secret': 'test_client_secret', | ||
| CONF_USERNAME: 'test_username', | ||
| CONF_PASSWORD: 'test_password', | ||
| 'subreddits': ['worldnews', 'news'], | ||
| CONF_MAXIMUM: 1 | ||
| } | ||
| } | ||
|
|
||
|
|
||
| INVALID_SORT_BY_CONFIG = { | ||
| 'sensor': { | ||
| 'platform': DOMAIN, | ||
| 'client_id': 'test_client_id', | ||
| 'client_secret': 'test_client_secret', | ||
| CONF_USERNAME: 'test_username', | ||
| CONF_PASSWORD: 'test_password', | ||
| 'subreddits': ['worldnews', 'news'], | ||
| 'sort_by': 'invalid_sort_by' | ||
| } | ||
| } | ||
|
|
||
|
|
||
| class ObjectView(): | ||
| """Use dict properties as attributes.""" | ||
|
|
||
| def __init__(self, d): | ||
| """Set dict as internal dict.""" | ||
| self.__dict__ = d | ||
|
|
||
|
|
||
| MOCK_RESULTS = { | ||
| 'results': [ | ||
| ObjectView({ | ||
| 'id': 0, | ||
| 'url': 'http://example.com/1', | ||
| 'title': 'example1', | ||
| 'score': '1', | ||
| 'num_comments': '1', | ||
| 'created': '', | ||
| 'selftext': 'example1 selftext' | ||
| }), | ||
| ObjectView({ | ||
| 'id': 1, | ||
| 'url': 'http://example.com/2', | ||
| 'title': 'example2', | ||
| 'score': '2', | ||
| 'num_comments': '2', | ||
| 'created': '', | ||
| 'selftext': 'example2 selftext' | ||
| }) | ||
| ] | ||
| } | ||
|
|
||
| MOCK_RESULTS_LENGTH = len(MOCK_RESULTS['results']) | ||
|
|
||
|
|
||
| class MockPraw(): | ||
| """Mock class for tmdbsimple library.""" | ||
|
|
||
| def __init__(self, client_id: str, client_secret: | ||
| str, username: str, password: str, | ||
| user_agent: str): | ||
| """Add mock data for API return.""" | ||
| self._data = MOCK_RESULTS | ||
|
|
||
| def subreddit(self, subreddit: str): | ||
| """Return an instance of a sunbreddit.""" | ||
| return MockSubreddit(subreddit, self._data) | ||
|
|
||
|
|
||
| class MockSubreddit(): | ||
| """Mock class for a subreddit instance.""" | ||
|
|
||
| def __init__(self, subreddit: str, data): | ||
| """Add mock data for API return.""" | ||
| self._subreddit = subreddit | ||
| self._data = data | ||
|
|
||
| def top(self, limit): | ||
| """Return top posts for a subreddit.""" | ||
| return self._return_data(limit) | ||
|
|
||
| def controversial(self, limit): | ||
| """Return controversial posts for a subreddit.""" | ||
| return self._return_data(limit) | ||
|
|
||
| def hot(self, limit): | ||
| """Return hot posts for a subreddit.""" | ||
| return self._return_data(limit) | ||
|
|
||
| def new(self, limit): | ||
| """Return new posts for a subreddit.""" | ||
| return self._return_data(limit) | ||
|
|
||
| def _return_data(self, limit): | ||
| """Test method to return modified data.""" | ||
| data = copy.deepcopy(self._data) | ||
| return data['results'][:limit] | ||
|
|
||
|
|
||
| class TestRedditSetup(unittest.TestCase): | ||
| """Test the Reddit platform.""" | ||
|
|
||
| def setUp(self): | ||
| """Initialize values for this testcase class.""" | ||
| self.hass = get_test_home_assistant() | ||
|
|
||
| def tearDown(self): # pylint: disable=invalid-name | ||
| """Stop everything that was started.""" | ||
| self.hass.stop() | ||
|
|
||
| @MockDependency('praw') | ||
| @patch('praw.Reddit', new=MockPraw) | ||
| def test_setup_with_valid_config(self, mock_praw): | ||
| """Test the platform setup with movie configuration.""" | ||
| setup_component(self.hass, 'sensor', VALID_CONFIG) | ||
|
|
||
| state = self.hass.states.get('sensor.reddit_worldnews') | ||
| assert int(state.state) == MOCK_RESULTS_LENGTH | ||
|
|
||
| state = self.hass.states.get('sensor.reddit_news') | ||
| assert int(state.state) == MOCK_RESULTS_LENGTH | ||
|
|
||
| assert state.attributes[ATTR_SUBREDDIT] == 'news' | ||
|
|
||
| assert state.attributes[ATTR_POSTS][0] == { | ||
| ATTR_ID: 0, | ||
| ATTR_URL: 'http://example.com/1', | ||
| ATTR_TITLE: 'example1', | ||
| ATTR_SCORE: '1', | ||
| ATTR_COMMENTS_NUMBER: '1', | ||
| ATTR_CREATED: '', | ||
| ATTR_BODY: 'example1 selftext' | ||
| } | ||
|
|
||
| assert state.attributes[CONF_SORT_BY] == 'hot' | ||
|
|
||
| @MockDependency('praw') | ||
| @patch('praw.Reddit', new=MockPraw) | ||
| def test_setup_with_invalid_config(self, mock_praw): | ||
| """Test the platform setup with invalid movie configuration.""" | ||
| setup_component(self.hass, 'sensor', INVALID_SORT_BY_CONFIG) | ||
| assert not self.hass.states.get('sensor.reddit_worldnews') |