internal/version/version_test.go

// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package version

import (
	"bytes"
	"fmt"
	"reflect"
	"strings"
	"testing"
)

func TestDedupEnv(t *testing.T) {
	tests := []struct {
		noCase bool
		in     []string
		want   []string
	}{
		{
			noCase: true,
			in:     []string{"k1=v1", "k2=v2", "K1=v3"},
			want:   []string{"K1=v3", "k2=v2"},
		},
		{
			noCase: false,
			in:     []string{"k1=v1", "K1=V2", "k1=v3"},
			want:   []string{"k1=v3", "K1=V2"},
		},
	}
	for _, tt := range tests {
		got := dedupEnv(tt.noCase, tt.in)
		if !reflect.DeepEqual(got, tt.want) {
			t.Errorf("Dedup(%v, %q) = %q; want %q", tt.noCase, tt.in, got, tt.want)
		}
	}
}

func TestFormatted(t *testing.T) {
	var total int64 = 1
	var buff = new(bytes.Buffer)
	var units = []string{"B", "KB", "MB"}
	for i := 1; i < 4; i++ {
		pw := &progressWriter{w: nil, total: total, formatted: true, output: buff}
		pw.update()
		total *= 1024
		expected := fmt.Sprintf("%d %s", 1, units[i-1])
		if !strings.Contains(buff.String(), expected) {
			t.Errorf("expected: %s received: %s", expected, buff.String())
		}
	}
}

func TestUnFormatted(t *testing.T) {
	var total int64 = 1
	var buff = new(bytes.Buffer)
	for i := 1; i < 4; i++ {
		pw := &progressWriter{w: nil, total: total, formatted: false, output: buff}
		pw.update()
		expected := fmt.Sprintf("%d bytes", total)
		if !strings.Contains(buff.String(), expected) {
			t.Errorf("expected: %s received: %s", expected, buff.String())
		}
		total *= 1024
	}
}

func TestComputeEnv(t *testing.T) {
	tests := []struct {
		name        string
		baseEnv     []string
		gotoolchain string
		wantContain string
		wantAbsent  string
	}{
		{
			name:        "not set, should use default",
			baseEnv:     []string{},
			gotoolchain: "auto",
			wantContain: "GOTOOLCHAIN=auto",
		},
		{
			name:        "already set, should override",
			baseEnv:     []string{"GOTOOLCHAIN=user-value"},
			gotoolchain: "auto",
			wantContain: "GOTOOLCHAIN=auto",
			wantAbsent:  "GOTOOLCHAIN=user-value",
		},
		{
			name:        "gotoolchain empty, should not add",
			baseEnv:     []string{},
			gotoolchain: "",
			wantAbsent:  "GOTOOLCHAIN=",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			env := computeEnv("/tmp/goroot", tt.gotoolchain, tt.baseEnv)

			foundContain := false
			foundAbsent := false
			for _, e := range env {
				if tt.wantContain != "" && e == tt.wantContain {
					foundContain = true
				}
				if tt.wantAbsent != "" && e == tt.wantAbsent {
					foundAbsent = true
				}
			}

			if tt.wantContain != "" && !foundContain {
				t.Errorf("expected env to contain %q, but it didn't. Env: %v", tt.wantContain, env)
			}
			if tt.wantAbsent != "" && foundAbsent {
				t.Errorf("expected env to NOT contain %q, but it did. Env: %v", tt.wantAbsent, env)
			}
		})
	}
}