From d8c3c29ff8e9eb585504814bd85ba2c0139d4b51 Mon Sep 17 00:00:00 2001 From: Eugen Eisler Date: Fri, 27 Sep 2024 12:07:11 +0200 Subject: [PATCH] fix: cli test --- cli/cli_test.go | 5 ++++- core/fabric.go | 4 +++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/cli/cli_test.go b/cli/cli_test.go index f750f14..ef9d776 100644 --- a/cli/cli_test.go +++ b/cli/cli_test.go @@ -1,6 +1,7 @@ package cli import ( + "github.com/danielmiessler/fabric/core" "os" "testing" @@ -9,8 +10,10 @@ import ( ) func TestCli(t *testing.T) { + os.Args = os.Args[:1] message, err := Cli("test") - assert.NoError(t, err) + assert.Error(t, err) + assert.Equal(t, core.NoSessionPatternUserMessages, err.Error()) assert.Empty(t, message) } diff --git a/core/fabric.go b/core/fabric.go index a70ff37..b3bd76b 100644 --- a/core/fabric.go +++ b/core/fabric.go @@ -30,6 +30,8 @@ import ( const DefaultPatternsGitRepoUrl = "https://github.com/danielmiessler/fabric.git" const DefaultPatternsGitRepoFolder = "patterns" +const NoSessionPatternUserMessages = "no session, pattern or user messages provided" + func NewFabric(db *db.Db) (ret *Fabric, err error) { ret = NewFabricBase(db) err = ret.Configure() @@ -279,7 +281,7 @@ func (o *Chat) BuildChatSession(raw bool) (ret *db.Session, err error) { if ret.IsEmpty() { ret = nil - err = fmt.Errorf("no session, pattern or user messages provided") + err = fmt.Errorf(NoSessionPatternUserMessages) } return }