def test_page_error(mock_requests, mock_db):
from plugins import horoscope
sess = setup_db(mock_db)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
with pytest.raises(requests.RequestException):
horoscope.horoscope('aries', sess, bot, 'some_user', event)
event.reply.assert_called_once_with(
"Could not get horoscope: Connection refused by Responses: GET " +
URL.format(sign=1) + " doesn't match Responses Mock. URL Error")
def test_database_read(mock_requests, mock_db):
sess = setup_db(mock_db)
mock_requests.add(
"GET",
URL.format(sign=4),
headers={"User-Agent": "Some user agent"},
body="""
<div class="main-horoscope">
<p>Some horoscope text</p>
</div>
""",
match_querystring=True,
)
mock_db.add_row(horoscope.table, nick="some_user", sign="cancer")
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
response = horoscope.horoscope("", sess, bot, "some_user", event)
assert response is None
event.message.assert_called_once_with("\x02cancer\x02 Some horoscope text")
def test_horoscope(mock_requests, mock_db):
sess = setup_db(mock_db)
mock_requests.add(
"GET",
URL.format(sign=1),
headers={"User-Agent": "Some user agent"},
body="""
<div class="main-horoscope">
<p>Some horoscope text</p>
</div>
""",
match_querystring=True,
)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
response = horoscope.horoscope("aries", sess, bot, "some_user", event)
assert response is None
event.message.assert_called_once_with("\x02aries\x02 Some horoscope text")
assert mock_db.get_data(horoscope.table) == [("some_user", "aries")]
def test_database_read(mock_requests, mock_db):
from plugins import horoscope
sess = setup_db(mock_db)
mock_requests.add(
'GET',
URL.format(sign=4),
headers={'User-Agent': "Some user agent"},
body="""
<div class="main-horoscope">
<p>Some horoscope text</p>
</div>
""",
match_querystring=True,
)
mock_db.add_row(horoscope.table, nick='some_user', sign='cancer')
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
response = horoscope.horoscope('', sess, bot, 'some_user', event)
assert response is None
event.message.assert_called_once_with("\x02cancer\x02 Some horoscope text")
def test_page_error(mock_requests, mock_db):
sess = setup_db(mock_db)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
mock_requests.add("GET", URL.format(sign=1), status=404)
with pytest.raises(requests.RequestException):
horoscope.horoscope("aries", sess, bot, "some_user", event)
event.reply.assert_called_once_with(
"Could not get horoscope: 404 Client Error: Not Found for url: {}. URL Error".format(
URL.format(sign=1)
)
)
def test_invalid_syntax(mock_requests, mock_db):
sess = setup_db(mock_db)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
response = horoscope.horoscope("", sess, bot, "some_user", event)
assert response is None
assert event.notice_doc.call_count == 1
def test_parse_fail(mock_requests, mock_db):
sess = setup_db(mock_db)
mock_requests.add(
"GET",
URL.format(sign=4),
headers={"User-Agent": "Some user agent"},
body="""
<div class="main-horoscope">
</div>
""",
match_querystring=True,
)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
with pytest.raises(horoscope.HoroscopeParseError):
horoscope.horoscope("cancer", sess, bot, "some_user", event)
event.reply.assert_called_once_with("Unable to parse horoscope posting")
def test_invalid_syntax(mock_requests, mock_db):
from plugins import horoscope
sess = setup_db(mock_db)
event = MagicMock()
bot = MagicMock()
bot.user_agent = "Some user agent"
response = horoscope.horoscope('', sess, bot, 'some_user', event)
assert response is None
event.notice_doc.assert_called_once()