#26 add: added new tests for client timeout.

Author: Vein05

requestHandler_test.go

@@ -0,0 +1,179 @@
+package deepseek_test
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/cohesion-org/deepseek-go"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestHandleSendChatCompletionRequest(t *testing.T) {
+	t.Run("successful request", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusOK)
+		}))
+		defer ts.Close()
+		c := deepseek.NewClient("test", ts.URL)
+
+		req, err := http.NewRequest("GET", ts.URL, nil)
+		require.NoError(t, err)
+
+		resp, err := deepseek.HandleSendChatCompletionRequest(*c, req)
+		require.NoError(t, err)
+		defer resp.Body.Close()
+
+		assert.Equal(t, http.StatusOK, resp.StatusCode)
+	})
+
+	t.Run("request error", func(t *testing.T) {
+		req, err := http.NewRequest("GET", "http://invalid-url", nil)
+		require.NoError(t, err)
+		c := deepseek.NewClient("test", "http://invalid-url")
+		resp, err := deepseek.HandleSendChatCompletionRequest(*c, req)
+		require.Error(t, err)
+		assert.Nil(t, resp)
+		assert.Contains(t, err.Error(), "error sending request:")
+	})
+}
+
+func TestHandleNormalRequest(t *testing.T) {
+	t.Run("successful request", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusOK)
+		}))
+		defer ts.Close()
+		c := deepseek.NewClient("test", ts.URL)
+		req, err := http.NewRequest("GET", ts.URL, nil)
+		require.NoError(t, err)
+
+		resp, err := deepseek.HandleNormalRequest(*c, req)
+		require.NoError(t, err)
+		defer resp.Body.Close()
+
+		assert.Equal(t, http.StatusOK, resp.StatusCode)
+	})
+
+	t.Run("request error", func(t *testing.T) {
+		req, err := http.NewRequest("GET", "http://invalid-url", nil)
+		require.NoError(t, err)
+
+		c := deepseek.NewClient("test", "http://invalid-url")
+		resp, err := deepseek.HandleNormalRequest(*c, req)
+		require.Error(t, err)
+		assert.Nil(t, resp)
+		assert.Contains(t, err.Error(), "error sending request:")
+	})
+}
+
+func TestTimeoutConfiguration(t *testing.T) {
+	t.Run("chat completion timeout", func(t *testing.T) {
+		start := time.Now()
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			time.Sleep(250 * time.Millisecond)
+			w.WriteHeader(http.StatusOK)
+		}))
+		defer ts.Close()
+
+		req, err := http.NewRequest("GET", ts.URL, nil)
+		require.NoError(t, err)
+		c := deepseek.NewClient("test", ts.URL)
+		resp, err := deepseek.HandleSendChatCompletionRequest(*c, req)
+		require.NoError(t, err)
+		defer resp.Body.Close()
+
+		// Verify timeout configuration isn't too short
+		assert.WithinDuration(t, start, time.Now(), 300*time.Millisecond)
+	})
+
+	t.Run("normal request timeout", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			time.Sleep(50 * time.Millisecond)
+			w.WriteHeader(http.StatusOK)
+		}))
+		defer ts.Close()
+
+		req, err := http.NewRequest("GET", ts.URL, nil)
+		require.NoError(t, err)
+
+		start := time.Now()
+		c := deepseek.NewClient("test", ts.URL)
+		resp, err := deepseek.HandleNormalRequest(*c, req)
+		require.NoError(t, err)
+		defer resp.Body.Close()
+
+		// Verify timeout allows successful completion
+		assert.WithinDuration(t, start, time.Now(), 100*time.Millisecond)
+	})
+}
+
+func TestErrorTimeout(t *testing.T) {
+	t.Run("client timeout preservation", func(t *testing.T) {
+		// Create test server that responds slower than client timeout
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			time.Sleep(250 * time.Millisecond) // Longer than client timeout
+		}))
+		defer ts.Close()
+
+		req, err := http.NewRequest("GET", ts.URL, nil)
+		require.NoError(t, err)
+
+		// Create client with short timeout
+		shortClient := &http.Client{
+			Timeout: 5 * time.Millisecond,
+		}
+
+		_, err = shortClient.Do(req)
+		require.Error(t, err)
+
+		assert.True(t, errors.Is(err, context.DeadlineExceeded) ||
+			(os.IsTimeout(err)), "should be timeout error")
+	})
+}
+
+func TestHandleTimeoutApplication(t *testing.T) {
+	t.Run("default timeout applied when no env var set", func(t *testing.T) {
+		// Ensure no timeout environment variable is set
+		_ = os.Unsetenv("DEEPSEEK_TIMEOUT")
+
+		expectedTimeout := 5 * time.Minute
+
+		timeout, err := deepseek.HandleTimeout()
+		require.NoError(t, err)
+		assert.Equal(t, expectedTimeout, timeout, "Expected default timeout when env var is missing")
+
+		client, err := deepseek.NewClientWithOptions("your-api-key", deepseek.WithTimeout(timeout))
+		require.NoError(t, err)
+		assert.Equal(t, expectedTimeout, client.Timeout, "Client timeout should match default")
+	})
+
+	t.Run("custom timeout from environment variable", func(t *testing.T) {
+		_ = os.Setenv("DEEPSEEK_TIMEOUT", "30s")
+		defer os.Unsetenv("DEEPSEEK_TIMEOUT")
+
+		expectedTimeout := 30 * time.Second
+
+		timeout, err := deepseek.HandleTimeout()
+		require.NoError(t, err)
+		assert.Equal(t, expectedTimeout, timeout, "Expected timeout from env variable")
+
+		client, err := deepseek.NewClientWithOptions("your-api-key", deepseek.WithTimeout(timeout))
+		require.NoError(t, err)
+		assert.Equal(t, expectedTimeout, client.Timeout, "Client timeout should match env var")
+	})
+
+	t.Run("invalid timeout format returns error", func(t *testing.T) {
+		_ = os.Setenv("DEEPSEEK_TIMEOUT", "invalid")
+		defer os.Unsetenv("DEEPSEEK_TIMEOUT")
+
+		timeout, err := deepseek.HandleTimeout()
+		assert.Error(t, err, "Expected error for invalid timeout format")
+		assert.Zero(t, timeout, "Invalid timeout should return zero duration")
+	})
+}