def test_get_data_ok():
mock_response = Mock()
mock_response.status_code = 200
session = Mock()
session.get.return_value = mock_response
pb = PushBullet("apikey")
pb._session = session
pb._get_data("url")
def test_get_data_other_error():
mock_response = Mock()
mock_response.status_code = 500
session = Mock()
session.get.return_value = mock_response
pb = PushBullet("apikey")
pb._session = session
with pytest.raises(PushbulletError):
pb._get_data("url")
def test_get_data_invalid_key():
mock_response = Mock()
mock_response.status_code = 401
session = Mock()
session.get.return_value = mock_response
pb = PushBullet("apikey")
pb._session = session
with pytest.raises(InvalidKeyError):
pb._get_data("url")
