distribusi-go/distribusi_test.go

package main

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
)

func TestTrimFinalNewline(t *testing.T) {
	trimmed := trimFinalNewline([]byte("foo\n"))
	if trimmed != "foo" {
		t.Fatalf("failed to trimmed new line from 'foo\\n'")
	}
}

func TestTrimAllNewlines(t *testing.T) {
	trimmed := trimAllNewlines("\nfoo\n")
	if trimmed != "foo" {
		t.Fatalf("failed to trimmed new line from '\\nfoo\\n'")
	}
}

func TestParseMtype(t *testing.T) {
	mtype := "os/directory"
	ftype, stype, err := parseMtype(mtype)
	if ftype != "os" || stype != "directory" {
		t.Fatalf("unable to parse %s", mtype)
	}

	mtype = "image/gif; charset=utf-8"
	ftype, stype, err = parseMtype(mtype)
	if err != nil {
		t.Fatalf("failed to parse %s", mtype)
	}
	if ftype != "image" || stype != "gif" {
		t.Fatalf("unable to parse %s", mtype)
	}

	mtype = "notamimetype"
	ftype, stype, err = parseMtype(mtype)
	if err == nil {
		t.Fatalf("failed to error out correctly parsing %s", mtype)
	}
}

func TestGetLogFile(t *testing.T) {
	f, err := getLogFile()
	if err != nil {
		t.Fatalf("failed to create log file, saw: %s", err)
	}

	if !strings.Contains(f.Name(), "distribusi-go") {
		t.Fatalf("log file named incorrectly: %s", f.Name())
	}

	absPath, err := filepath.Abs(f.Name())
	if err != nil {
		t.Fatalf("failed to read absoluate path of %s", f.Name())
	}

	if err := os.Remove(absPath); err != nil {
		t.Fatalf("unable to remove %s", absPath)
	}
}