def test_pick_random_category():
with patch.multiple('site_parser.index_parser',
get_categories=DEFAULT,
random=DEFAULT) as values:
random_mock = values['random']
get_categories_mock = values['get_categories']
soup = Mock(BeautifulSoup)
random_mock.choice.return_value = 'c'
get_categories_mock.return_value = categories = ['a', 'b', 'c', 'd']
result = pick_random_category(soup)
get_categories_mock.assert_called_once_with(soup)
random_mock.choice.assert_called_once_with(categories)
assert result == 'c'
def extract_sample_category_from_site():
soup = get_soup_from_url('http://p30download.com/')
# soup = build_beautiful_soup_from_path('sample/sample_post.html')
return pick_random_category(soup)
