// Copyright 2018 Netflix, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package env

import (
	"errors"
	"fmt"
	"testing"
)

func TestEnvSetApply(t *testing.T) {
	t.Parallel()
	es := EnvSet{
		"HOME":      "/home/edgarl",
		"WORKSPACE": "/mnt/builds/slave/workspace/test",
	}

	workspace := ""
	cs := ChangeSet{
		"HOME":      nil,        // os.Unsetenv("HOME")
		"WORKSPACE": &workspace, // os.Setenv("WORKSPACE", "")
	}

	es.Apply(cs)

	v, ok := es["HOME"]
	if ok {
		t.Errorf("Expected field '%s' to not exist but got '%s'", "HOME", v)
	}

	v, ok = es["WORKSPACE"]
	if !ok {
		t.Errorf("Expected field '%s' to exist but missing", "WORKSPACE")
	} else if v != "" {
		t.Errorf("Expected field value to be '%s' but got '%s'", workspace, v)
	}
}

func TestEnvironToEnvSet(t *testing.T) {
	t.Parallel()
	environ := []string{"HOME=/home/edgarl", "WORKSPACE=/mnt/builds/slave/workspace/test"}

	m, err := EnvironToEnvSet(environ)
	if err != nil {
		t.Errorf("Expected no error but got '%s'", err)
	}

	if m["HOME"] != "/home/edgarl" {
		t.Errorf("Expected map value to be '%s' but got '%s'", "/home/edgarl", m["HOME"])
	}

	if m["WORKSPACE"] != "/mnt/builds/slave/workspace/test" {
		t.Errorf("Expected map value to be '%s' but got '%s'", "/mnt/builds/slave/workspace/test", m["WORKSPACE"])
	}
}

func TestEnvironToEnvSetInvalid(t *testing.T) {
	t.Parallel()
	environ := []string{"INVALID"}

	if _, err := EnvironToEnvSet(environ); !errors.Is(err, ErrInvalidEnviron) {
		t.Errorf("Expected 'ErrInvalidEnviron' but got '%s'", err)
	}
}

func TestEnvironToEnvSetSplitN(t *testing.T) {
	t.Parallel()
	environ := []string{"SPLIT=one=two"}

	m, err := EnvironToEnvSet(environ)
	if err != nil {
		t.Errorf("Expected no error but got '%s'", err)
	}

	if m["SPLIT"] != "one=two" {
		t.Errorf("Expected map value to be '%s' but got '%s'", "one=two", m["SPLIT"])
	}
}

func TestEnvSetToEnviron(t *testing.T) {
	t.Parallel()
	m := EnvSet{
		"HOME":      "/home/test",
		"WORKSPACE": "/mnt/builds/slave/workspace/test",
	}

	environ := EnvSetToEnviron(m)
	if len(environ) != 2 {
		t.Errorf("Expected environ to have %d items but instead got %d", 2, len(environ))
	}

	for k, v := range m {
		found := false
		envPair := fmt.Sprintf("%s=%s", k, v)
		for _, e := range environ {
			if e == envPair {
				found = true
				break
			}
		}

		if !found {
			t.Errorf("Expected environ to contain '%s'", envPair)
		}
	}
}