package tests

import (
	"testing"

	hrp "github.com/httprunner/httprunner/v5"
	"github.com/stretchr/testify/assert"
)

// reference extracted variables for validation in the same step
func TestCaseExtractStepSingle(t *testing.T) {
	testcase := &hrp.TestCase{
		Config: hrp.NewConfig("run request with variables").
			SetBaseURL("https://postman-echo.com").
			SetVerifySSL(false),
		TestSteps: []hrp.IStep{
			hrp.NewStep("get with params").
				WithVariables(map[string]interface{}{
					"var1":               "bar1",
					"agent":              "HttpRunnerPlus",
					"expectedStatusCode": 200,
					"jmespathFoo1":       "body.args.foo1",
				}).
				GET("/get").
				WithParams(map[string]interface{}{"foo1": "$var1", "foo2": "bar2"}).
				WithHeaders(map[string]string{"User-Agent": "$agent"}).
				Extract().
				WithJmesPath("status_code", "statusCode").
				WithJmesPath("headers.\"Content-Type\"", "contentType").
				WithJmesPath("$jmespathFoo1", "varFoo1").
				Validate().
				AssertEqual("$statusCode", "$expectedStatusCode", "check status code").                      // assert with extracted variable from current step
				AssertEqual("$contentType", "application/json; charset=utf-8", "check header Content-Type"). // assert with extracted variable from current step
				AssertEqual("$varFoo1", "bar1", "check args foo1").                                          // assert with extracted variable from current step
				AssertEqual("body.args.foo2", "bar2", "check args foo2").
				AssertEqual("body.headers.\"user-agent\"", "HttpRunnerPlus", "check header user agent"),
		},
	}

	err := hrp.NewRunner(t).Run(testcase)
	assert.Nil(t, err)
}

// reference extracted variables from previous step
func TestCaseExtractStepAssociation(t *testing.T) {
	testcase := &hrp.TestCase{
		Config: hrp.NewConfig("run request with variables").
			SetBaseURL("https://postman-echo.com").
			SetVerifySSL(false),
		TestSteps: []hrp.IStep{
			hrp.NewStep("get with params").
				WithVariables(map[string]interface{}{
					"var1":  "bar1",
					"agent": "HttpRunnerPlus",
				}).
				GET("/get").
				WithParams(map[string]interface{}{"foo1": "$var1", "foo2": "bar2"}).
				WithHeaders(map[string]string{"User-Agent": "$agent"}).
				Extract().
				WithJmesPath("status_code", "statusCode").
				WithJmesPath("headers.\"Content-Type\"", "contentType").
				WithJmesPath("body.args.foo1", "varFoo1").
				Validate().
				AssertEqual("$statusCode", 200, "check status code").
				AssertEqual("$contentType", "application/json; charset=utf-8", "check header Content-Type").
				AssertEqual("$varFoo1", "bar1", "check args foo1").
				AssertEqual("body.args.foo2", "bar2", "check args foo2").
				AssertEqual("body.headers.\"user-agent\"", "HttpRunnerPlus", "check header user agent"),
			hrp.NewStep("post json data").
				POST("/post").
				WithHeaders(map[string]string{"User-Agent": "HttpRunnerPlus"}).
				WithBody(map[string]interface{}{"foo1": "bar1", "foo2": "bar2"}).
				Validate().
				AssertEqual("status_code", "$statusCode", "check status code"). // assert with extracted variable from previous step
				AssertEqual("$varFoo1", "bar1", "check json foo1").             // assert with extracted variable from previous step
				AssertEqual("body.json.foo2", "bar2", "check json foo2"),
		},
	}

	err := hrp.NewRunner(t).Run(testcase)
	assert.Nil(t, err)
}