From f3053c36287078aa144ff7ac259ac5d89de0a3e0 Mon Sep 17 00:00:00 2001
From: Siarhei Fedartsou <siarhei.fedartsou@gmail.com>
Date: Sun, 12 Sep 2021 20:58:13 +0300
Subject: [PATCH] fix tests

---
 pkg/cmd/pr/list/list_test.go | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/pkg/cmd/pr/list/list_test.go b/pkg/cmd/pr/list/list_test.go
index 1823175c7..2d3aa62aa 100644
--- a/pkg/cmd/pr/list/list_test.go
+++ b/pkg/cmd/pr/list/list_test.go
@@ -191,9 +191,8 @@ func TestPRList_bothNonDraftAndDraft(t *testing.T) {
 	defer http.Verify(t)
 
 	_, err := runCommand(http, true, `--draft --non-draft`)
-	if err == nil || err.Error() != "specify only one of `--draft` or `--non-draft`" {
-		t.Fatal(err)
-	}
+	assert.NotNil(t, err)
+	assert.Equal(t, err.Error(), "specify only one of `--draft` or `--non-draft`")
 }
 
 func TestPRList_filteringDraft(t *testing.T) {
@@ -238,9 +237,8 @@ func TestPRList_withInvalidLimitFlag(t *testing.T) {
 	defer http.Verify(t)
 
 	_, err := runCommand(http, true, `--limit=0`)
-	if err == nil && err.Error() != "invalid limit: 0" {
-		t.Errorf("error running command `issue list`: %v", err)
-	}
+	assert.NotNil(t, err)
+	assert.Equal(t, err.Error(), "invalid value for --limit: 0")
 }
 
 func TestPRList_web(t *testing.T) {