Fixed test when context is nil

Pixboost · Feb 2, 2025 · 9e144e8 · 9e144e8
1 parent 92921a1
commit 9e144e8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/img/service.go b/img/service.go
@@ -409,6 +409,10 @@ func NewContextWithHeaders(ctx context.Context, headers *http.Header) context.Co
 }
 
 func HeaderFromContext(ctx context.Context) (*http.Header, bool) {
+	if ctx == nil {
+		return nil, false
+	}
+
 	header, ok := ctx.Value(headersKey(0)).(*http.Header)
 	return header, ok
 }
diff --git a/img/service_test.go b/img/service_test.go
@@ -697,7 +697,7 @@ func FuzzService_ResizeUrl(f *testing.F) {
 	})
 }
 
-func TestHeaderFromContext(t *testing.T) {
+func TestHeaderFromContext_NoHeader(t *testing.T) {
 	header, ok := img.HeaderFromContext(context.Background())
 	if header != nil {
 		t.Errorf("expected nil header")
@@ -707,6 +707,16 @@ func TestHeaderFromContext(t *testing.T) {
 	}
 }
 
+func TestHeaderFromContext_NoContext(t *testing.T) {
+	header, ok := img.HeaderFromContext(nil)
+	if header != nil {
+		t.Errorf("expected nil header")
+	}
+	if ok {
+		t.Errorf("expected ok to be false")
+	}
+}
+
 func createService(t *testing.T) *img.Service {
 	img.CacheTTL = 86400
 	s, err := img.NewService(&loaderMock{}, &resizerMock{}, 1)