From 07976e00c048330a37a9379225e10eb86343d434 Mon Sep 17 00:00:00 2001
From: hupe1980 <frankhuebner1980@gmail.com>
Date: Tue, 19 Dec 2023 00:30:36 +0100
Subject: [PATCH] Add testcases

---
 model/chatmodel/fake_test.go | 103 +++++++++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 model/chatmodel/fake_test.go

diff --git a/model/chatmodel/fake_test.go b/model/chatmodel/fake_test.go
new file mode 100644
index 0000000..22385f1
--- /dev/null
+++ b/model/chatmodel/fake_test.go
@@ -0,0 +1,103 @@
+package chatmodel
+
+import (
+	"context"
+	"testing"
+
+	"github.com/hupe1980/golc"
+	"github.com/hupe1980/golc/schema"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFake(t *testing.T) {
+	t.Run("Generate_Success", func(t *testing.T) {
+		// Arrange
+		fake := NewSimpleFake("response")
+
+		// Act
+		result, err := fake.Generate(context.Background(), schema.ChatMessages{})
+
+		// Assert
+		assert.NoError(t, err)
+		assert.NotNil(t, result)
+		assert.Equal(t, "response", result.Generations[0].Text)
+	})
+
+	t.Run("Generate_CustomResultFunc", func(t *testing.T) {
+		// Arrange
+		expectedResult := &schema.ModelResult{
+			Generations: []schema.Generation{{Text: "custom_response"}},
+		}
+		fake := NewFake(func(ctx context.Context, messages schema.ChatMessages) (*schema.ModelResult, error) {
+			return expectedResult, nil
+		})
+
+		// Act
+		result, err := fake.Generate(context.Background(), schema.ChatMessages{})
+
+		// Assert
+		assert.NoError(t, err)
+		assert.NotNil(t, result)
+		assert.Equal(t, "custom_response", result.Generations[0].Text)
+	})
+
+	t.Run("Generate_ErrorFromResultFunc", func(t *testing.T) {
+		// Arrange
+		expectedError := assert.AnError
+		fake := NewFake(func(ctx context.Context, messages schema.ChatMessages) (*schema.ModelResult, error) {
+			return nil, expectedError
+		})
+
+		// Act
+		result, err := fake.Generate(context.Background(), schema.ChatMessages{})
+
+		// Assert
+		assert.Error(t, err)
+		assert.Nil(t, result)
+		assert.EqualError(t, err, expectedError.Error())
+	})
+
+	t.Run("Type", func(t *testing.T) {
+		// Arrange
+		fake := NewSimpleFake("response")
+
+		// Act
+		result := fake.Type()
+
+		// Assert
+		assert.Equal(t, "chatmodel.Fake", result)
+	})
+
+	t.Run("Verbose", func(t *testing.T) {
+		// Arrange
+		fake := NewSimpleFake("response")
+
+		// Act
+		result := fake.Verbose()
+
+		// Assert
+		assert.Equal(t, golc.Verbose, result)
+	})
+
+	t.Run("Callbacks", func(t *testing.T) {
+		// Arrange
+		fake := NewSimpleFake("response")
+
+		// Act
+		result := fake.Callbacks()
+
+		// Assert
+		assert.Empty(t, result)
+	})
+
+	t.Run("InvocationParams", func(t *testing.T) {
+		// Arrange
+		fake := NewSimpleFake("response")
+
+		// Act
+		result := fake.InvocationParams()
+
+		// Assert
+		assert.Empty(t, result)
+	})
+}