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")