context_test.go

package juliet

import (
	"testing"
)

func TestGet(t *testing.T) {
	ctx := NewContext()
	ctx.values["foo"] = "bar"
	ctx.values["faa"] = nil

	if val, ok := ctx.Get("foo"); ok {
		if val != "bar" {
			t.Fatalf("Invalid value for key foo. Expected 'bar' but was '%v'", val)
		}
	} else {
		t.Fatalf("Missing value for key foo")
	}

	if val, ok := ctx.Get("faa"); ok {
		if val != nil {
			t.Fatalf("Invalid value for key faa. Expected nil but was '%v'", val)
		}
	} else {
		t.Fatalf("Missing value for key faa")
	}

	if val, ok := ctx.Get("xyz"); ok {
		t.Fatalf("Invalid value '%v' for key xyz", val)
	}
}

func TestSet(t *testing.T) {
	ctx := NewContext()

	ctx.Set("foo", "bar")
	if val, ok := ctx.values["foo"]; ok {
		if val != "bar" {
			t.Fatalf("Invalid value for key foo. Expected 'bar' but was '%v'", val)
		}
	} else {
		t.Fatalf("Missing value for key foo")
	}
}

func TestDelete(t *testing.T) {
	ctx := NewContext()
	ctx.values["foo"] = "bar"

	ctx.Delete("foo")
	if val, ok := ctx.Get("foo"); ok {
		t.Fatalf("Invalid value '%v' for key foo", val)
	}

	ctx.Delete("faa")
}

func TestClear(t *testing.T) {
	ctx := NewContext()
	ctx.values["foo"] = "bar"
	ctx.values["plip"] = "plop"

	ctx.Clear()
	if len(ctx.values) > 0 {
		t.Fatalf("Invalid value count %d, Expected 0", len(ctx.values))
	}

	ctx.Delete("faa")
}

func TestCopy(t *testing.T) {
	ctx := NewContext()
	ctx.Set("foo", "bar")
	copy := ctx.Copy()
	copy.Set("foo", "baz")

	if val, ok := ctx.Get("foo"); ok {
		if val != "bar" {
			t.Fatalf("Invalid value for key foo. Expected 'bar' but was '%v'", val)
		}
	} else {
		t.Fatalf("Missing value for key foo")
	}

	if val, ok := copy.Get("foo"); ok {
		if val != "baz" {
			t.Fatalf("Invalid value for key foo. Expected 'baz' but was '%v'", val)
		}
	} else {
		t.Fatalf("Missing value for key plop")
	}
}

func TestString(t *testing.T) {
	ctx := NewContext()
	ctx.Set("foo", "bar")
	expected := "foo => bar\n"
	str := ctx.String()
	if str != expected {
		t.Fatalf("Invalid context string representation %s, expected %s", str, expected)
	}
}