zdravko/pkg/k6/k6_test.go

package k6

import (
	"context"
	"log/slog"
	"os"
	"testing"

	"github.com/mentos1386/zdravko/pkg/k6/zdravko"
)

func getLogger() *slog.Logger {
	opts := &slog.HandlerOptions{
		Level: slog.LevelDebug,
	}

	handler := slog.NewTextHandler(os.Stdout, opts)

	return slog.New(handler)
}

func TestK6Success(t *testing.T) {
	logger := getLogger()

	script := `
import http from 'k6/http';
import { sleep } from 'k6';
import { getTarget } from 'k6/x/zdravko';

export const options = {
  vus: 10,
  duration: '5s',
};

export default function () {
  const target = getTarget();
  console.log('Target:', target);
  http.get('https://test.k6.io');
  sleep(1);
}
`

	execution := NewExecution(logger, script)

	ctx := zdravko.WithZdravkoContext(context.Background(), zdravko.Context{Target: zdravko.Target{
		Name:  "Test",
		Group: "Test",
		Metadata: map[string]interface{}{
			"Kind": "Test",
		},
	}})

	result, err := execution.Run(ctx)
	if err != nil {
		t.Errorf("Error starting execution: %v", err)
	}
	if result != nil {
		t.Logf("Result: %v", result)
	}
}

func TestK6Fail(t *testing.T) {
	ctx := context.Background()
	logger := getLogger()

	script := `
import http from 'k6/http';
import { sleep } from 'k6';

export const options = {
  thresholds: {
    http_req_failed: ['rate<0.01'], // http errors should be less than 1%
  },
};

export default function () {
  http.get('https://fail.broken.example');
  sleep(1);
}
`

	execution := NewExecution(logger, script)

	result, err := execution.Run(ctx)
	if err != nil {
		t.Errorf("Error starting execution: %v", err)
	}
	if result != nil {
		t.Logf("Result: %v", result)
	}
}