From 242b025c9a056065b2575e8fe55773dd1b7e8493 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Musab=20G=C3=BCltekin?= <gultekinmusab@gmail.com>
Date: Sun, 8 Aug 2021 22:08:13 +0300
Subject: [PATCH] Set cookie test

---
 geziyor_test.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/geziyor_test.go b/geziyor_test.go
index b4251e9..5d552b8 100644
--- a/geziyor_test.go
+++ b/geziyor_test.go
@@ -10,6 +10,7 @@ import (
 	"github.com/geziyor/geziyor/cache/diskcache"
 	"github.com/geziyor/geziyor/client"
 	"github.com/geziyor/geziyor/export"
+	"github.com/geziyor/geziyor/internal"
 	"github.com/geziyor/geziyor/metrics"
 	"github.com/stretchr/testify/assert"
 	"net/http"
@@ -132,6 +133,27 @@ func TestGetRendered(t *testing.T) {
 	}).Start()
 }
 
+func TestGetRenderedCookie(t *testing.T) {
+	testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_, _ = w.Write([]byte(r.Header.Get("Cookie")))
+	}))
+	geziyor.NewGeziyor(&geziyor.Options{
+		StartRequestsFunc: func(g *geziyor.Geziyor) {
+			req, err := client.NewRequest("GET", testServer.URL, nil)
+			if err != nil {
+				internal.Logger.Printf("Request creating error %v\n", err)
+				return
+			}
+			req.Header.Set("Cookie", "key=value")
+			req.Rendered = true
+			g.Do(req, g.Opt.ParseFunc)
+		},
+		ParseFunc: func(g *geziyor.Geziyor, r *client.Response) {
+			assert.Contains(t, string(r.Body), "key=value")
+		},
+	}).Start()
+}
+
 // Run chrome headless instance to test this
 //func TestGetRenderedRemoteAllocator(t *testing.T) {
 //	geziyor.NewGeziyor(&geziyor.Options{