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

tests/components/reddit/test_sensor.py

@@ -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')