From 12047cdc92b4d12f3d51b13395f749516caf9640 Mon Sep 17 00:00:00 2001 From: Ivan Habunek Date: Thu, 20 Apr 2017 10:58:49 +0200 Subject: [PATCH] Fix tests for follow/unfollow --- tests/test_console.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/tests/test_console.py b/tests/test_console.py index 6d19d7c..c9c60eb 100644 --- a/tests/test_console.py +++ b/tests/test_console.py @@ -3,7 +3,7 @@ import pytest import requests import re -from toot import console, User, App +from toot import console, User, App, ConsoleError from tests.utils import MockResponse @@ -218,10 +218,9 @@ def test_follow_not_found(monkeypatch, capsys): monkeypatch.setattr(requests, 'get', mock_get) - console.run_command(app, user, 'follow', ['blixa']) - - out, err = capsys.readouterr() - assert "Account not found" in err + with pytest.raises(ConsoleError) as ex: + console.run_command(app, user, 'follow', ['blixa']) + assert "Account not found" == str(ex.value) def test_unfollow(monkeypatch, capsys): @@ -265,10 +264,9 @@ def test_unfollow_not_found(monkeypatch, capsys): monkeypatch.setattr(requests, 'get', mock_get) - console.run_command(app, user, 'unfollow', ['blixa']) - - out, err = capsys.readouterr() - assert "Account not found" in err + with pytest.raises(ConsoleError) as ex: + console.run_command(app, user, 'unfollow', ['blixa']) + assert "Account not found" == str(ex.value) def test_whoami(monkeypatch, capsys):