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 recieved: %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 recieved: %s", expected, buff.String())
		}
		total *= 1024
	}
}