// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.

package runner

import (
	"testing"

	"github.com/drone/drone/core"
	"github.com/google/go-cmp/cmp"
)

func Test_systemEnviron(t *testing.T) {
	system := &core.System{
		Proto:   "https",
		Host:    "meta.drone.io",
		Link:    "https://meta.drone.io",
		Version: "v1.0.0",
	}
	got := systemEnviron(system)
	want := map[string]string{
		"CI":                    "true",
		"DRONE":                 "true",
		"DRONE_SYSTEM_PROTO":    "https",
		"DRONE_SYSTEM_HOST":     "meta.drone.io",
		"DRONE_SYSTEM_HOSTNAME": "meta.drone.io",
		"DRONE_SYSTEM_VERSION":  "v1.0.0",
	}
	if diff := cmp.Diff(got, want); diff != "" {
		t.Errorf(diff)
	}
}

func Test_runnerEnviron(t *testing.T) {
	runner := &Runner{
		Machine:  "ip-12-34-56-78.us-west-2.compute.internal",
		Platform: "linux/amd64",
	}
	got := agentEnviron(runner)
	want := map[string]string{
		"DRONE_MACHINE":         "ip-12-34-56-78.us-west-2.compute.internal",
		"DRONE_RUNNER_HOST":     "ip-12-34-56-78.us-west-2.compute.internal",
		"DRONE_RUNNER_HOSTNAME": "ip-12-34-56-78.us-west-2.compute.internal",
		"DRONE_RUNNER_PLATFORM": "linux/amd64",
	}
	if diff := cmp.Diff(got, want); diff != "" {
		t.Errorf(diff)
	}
}