mirror of
https://github.com/developersIndia/deviras.git
synced 2024-11-08 13:42:34 +05:30
981bd089b5
Added docstring for main.py script and updated it to only invoke update_titles function if the file was run directly. Added initial tests for titles updater script.
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
# patch out the environ dictionary
|
|
# used to instantiate global variables in the titles_updater script
|
|
# maybe move the call to os.environ in the script to the update_titles method?
|
|
environ_patcher = patch.dict('os.environ', {
|
|
'REDDIT_CLIENT_ID': '',
|
|
'REDDIT_CLIENT_SECRET': '',
|
|
'REDDIT_PASSWORD': ''
|
|
})
|
|
environ_patcher.start()
|
|
|
|
from main import get_titles
|
|
|
|
# TODO: write tests for update_titles method after figuring out
|
|
# which objects to mock and with what
|
|
class TestTitlesUpdater(unittest.TestCase):
|
|
@patch('main.json.load', return_value={'titles': ['foo', 'bar', 'baz']})
|
|
def test_get_titles_returns_a_list(self, _):
|
|
titles = get_titles()
|
|
self.assertIsInstance(titles, list)
|
|
|
|
@patch('main.json.load', return_value={'titles': ['foo', 'bar', 'baz']})
|
|
def test_get_titles_contains_titles_from_dataset_file(self, mock_json_load):
|
|
titles = get_titles()
|
|
|
|
self.assertTrue(all(map(lambda title: title in mock_json_load.return_value['titles'], titles)))
|
|
|
|
environ_patcher.stop()
|