From 4e520a78af7f2fa047c9c5d68c36276248436208 Mon Sep 17 00:00:00 2001
From: "C.C." <fanweixiao@gmail.com>
Date: Fri, 22 Mar 2024 10:42:20 +0800
Subject: [PATCH] add test

---
 .../ai/provider/cfopenai/provider_test.go     | 50 +++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 pkg/bridge/ai/provider/cfopenai/provider_test.go

diff --git a/pkg/bridge/ai/provider/cfopenai/provider_test.go b/pkg/bridge/ai/provider/cfopenai/provider_test.go
new file mode 100644
index 000000000..547b1b342
--- /dev/null
+++ b/pkg/bridge/ai/provider/cfopenai/provider_test.go
@@ -0,0 +1,50 @@
+package cfopenai
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCloudflareOpenAIProvider_Name(t *testing.T) {
+	provider := &CloudflareOpenAIProvider{}
+
+	name := provider.Name()
+
+	assert.Equal(t, "cloudflare_openai", name)
+}
+
+func TestNewProvider(t *testing.T) {
+	t.Run("with parameters", func(t *testing.T) {
+		provider := NewProvider("test_endpoint", "test_api_key", "test_model")
+
+		assert.Equal(t, "test_endpoint", provider.CfEndpoint)
+		assert.Equal(t, "test_api_key", provider.APIKey)
+		assert.Equal(t, "test_model", provider.Model)
+	})
+
+	t.Run("with environment variables", func(t *testing.T) {
+		os.Setenv("OPENAI_API_KEY", "env_api_key")
+		os.Setenv("OPENAI_MODEL", "env_model")
+
+		provider := NewProvider("test_endpoint", "", "")
+
+		assert.Equal(t, "test_endpoint", provider.CfEndpoint)
+		assert.Equal(t, "env_api_key", provider.APIKey)
+		assert.Equal(t, "env_model", provider.Model)
+
+		os.Unsetenv("OPENAI_API_KEY")
+		os.Unsetenv("OPENAI_MODEL")
+	})
+
+	t.Run("without cfEndpoint", func(t *testing.T) {
+		if os.Getenv("CI") != "true" {
+			t.Skip("Skipping testing in CI environment")
+		}
+
+		assert.Panics(t, func() {
+			NewProvider("", "test_api_key", "test_model")
+		})
+	})
+}