// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
//go:build script
// +build script

package js_test

import (
	"github.com/TeaOSLab/EdgeNode/internal/js"
	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
	"github.com/iwind/TeaGo/types"
	"runtime"
	"runtime/debug"
	"testing"
	"time"
)

func TestIsolatePool_GetContext(t *testing.T) {
	pool, err := js.NewIsolatePool(1)
	if err != nil {
		t.Fatal(err)
	}
	ctx, err := pool.GetContext()
	if err != nil {
		t.Fatal(err)
	}
	defer pool.PutContext(ctx)
	v, err := ctx.Run("1+1", "")
	if err != nil {
		t.Fatal(err)
	}
	t.Log("result:", v)
}

func TestIsolatePool_Memory(t *testing.T) {
	if !testutils.IsSingleTesting() {
		return
	}

	t.Log("creating pool")
	var req = &FakeRequest{}
	var before = time.Now()

	pool, err := js.NewAutoIsolatePool()
	if err != nil {
		t.Fatal(err)
	}
	t.Log("created pool", time.Since(before).Seconds()*1000, "ms")
	time.Sleep(10 * time.Second)

	t.Log("starting")
	for i := 0; i < 10_000_000; i++ {
		if i > 0 && i%50000 == 0 {
			t.Log(i)
			time.Sleep(1 * time.Second)
		}

		ctx, err := pool.GetContext()
		if err != nil {
			t.Fatal(err)
		}
		var requestObjectId = ctx.AddGoRequestObject(req)

		_, err = ctx.Run(`(function () {
let req = new gojs.net.http.Request()
req.setGoObject(`+types.String(requestObjectId)+`)

let resp = new gojs.net.http.Response()

if (req.path.match(/webhook/)) {
	req.deleteHeader("Cookie")
	req.setHeader("User-Agent", "gojs/1.0")

	resp.send(200, "Hello, World")
}

if (req.path == "/favicon.ico") {
	req.done()	
}
})()`, "")
		if err != nil {
			t.Fatal(err)
		}

		pool.PutContext(ctx)
	}
	t.Log("created", time.Since(before).Seconds(), "s")

	if testutils.IsSingleTesting() {
		time.Sleep(5 * time.Second)
		t.Log("gc")
		runtime.GC()
		debug.FreeOSMemory()

		time.Sleep(120 * time.Second)
		t.Log("gc again")
		runtime.GC()
		debug.FreeOSMemory()

		time.Sleep(5 * time.Hour)
		t.Log("2 minutes")
	}
}

func TestIsolatePool_Memory2(t *testing.T) {
	if !testutils.IsSingleTesting() {
		return
	}

	pool, err := js.NewIsolatePool(256)
	if err != nil {
		t.Fatal(err)
	}

	time.Sleep(4 * time.Second)
	var gcPause = func() {
		var before = time.Now()
		runtime.GC()
		var costSeconds = time.Since(before).Seconds()
		var stats = &debug.GCStats{}
		debug.ReadGCStats(stats)
		t.Log("GC pause:", stats.PauseTotal.Seconds()*1000, "ms", "cost:", costSeconds*1000, "ms")
	}

	gcPause()

	time.Sleep(10 * time.Second)

	_ = pool
	t.Log(pool.MaxSize())
}

func BenchmarkIsolatePool_GetContext(b *testing.B) {
	pool, err := js.NewIsolatePool(32)
	if err != nil {
		b.Fatal(err)
	}

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			func() {
				ctx, err := pool.GetContext()
				if err != nil {
					return
				}
				defer pool.PutContext(ctx)

				// TODO 需要实现测试用例
				/**err = ctx.AddGoObject(&FakeRequest{
					host:       "example.com",
					remoteAddr: "127.0.0.1",
				})
				if err != nil {
					b.Log(err)
					return
				}

				_, err = ctx.Run("req.host() + ' @ ' + req.remoteAddr() + ' @ ' + req.proto() + ' @ ' + req.url()", "")
				if err != nil {
					b.Log(err)
					return
				}**/
			}()
		}
	})
}