From 1ffa924483060d6dbcb1e4f276055b51eed59d71 Mon Sep 17 00:00:00 2001 From: Ivan Kravets Date: Wed, 16 Mar 2022 18:17:21 +0200 Subject: [PATCH] Fix test --- tests/commands/pkg/test_uninstall.py | 18 +++++------------- 1 file changed, 5 insertions(+), 13 deletions(-) diff --git a/tests/commands/pkg/test_uninstall.py b/tests/commands/pkg/test_uninstall.py index b7519451..3fc8f3ec 100644 --- a/tests/commands/pkg/test_uninstall.py +++ b/tests/commands/pkg/test_uninstall.py @@ -16,11 +16,10 @@ import os -import pytest - from platformio import fs from platformio.package.commands.install import package_install_cmd from platformio.package.commands.uninstall import package_uninstall_cmd +from platformio.package.exception import UnknownPackageError from platformio.package.manager.library import LibraryPackageManager from platformio.package.manager.platform import PlatformPackageManager from platformio.package.manager.tool import ToolPackageManager @@ -306,8 +305,7 @@ def test_custom_project_libraries( result = clirunner.invoke( package_uninstall_cmd, ["-l", "platformio/unknown_library"] ) - with pytest.raises(AssertionError, match="UnknownPackageError"): - validate_cliresult(result) + assert isinstance(result.exception, UnknownPackageError) def test_custom_project_tools( @@ -371,8 +369,7 @@ def test_custom_project_tools( result = clirunner.invoke( package_uninstall_cmd, ["-t", "platformio/unknown_tool"] ) - with pytest.raises(AssertionError, match="UnknownPackageError"): - validate_cliresult(result) + assert isinstance(result.exception, UnknownPackageError) def test_custom_project_platforms( @@ -407,10 +404,5 @@ def test_custom_project_platforms( assert not pkgs_to_names(ToolPackageManager().get_installed()) # unknown platform - with pytest.raises( - AssertionError, - match="Could not find the package with 'unknown_platform' requirements", - ): - validate_cliresult( - clirunner.invoke(package_uninstall_cmd, ["-p", "unknown_platform"]) - ) + result = clirunner.invoke(package_uninstall_cmd, ["-p", "unknown_platform"]) + assert isinstance(result.exception, UnknownPackageError)