fx/pkg/dict/dict_test.go

package dict

import "testing"

func Test_dict(t *testing.T) {
	d := NewDict()
	d.Set("number", 3)
	v, _ := d.Get("number")
	if v.(int) != 3 {
		t.Error("Set number")
	}
	// string
	d.Set("string", "x")
	v, _ = d.Get("string")
	if v.(string) != "x" {
		t.Error("Set string")
	}
	// string slice
	d.Set("strings", []string{
		"t",
		"u",
	})
	v, _ = d.Get("strings")
	if v.([]string)[0] != "t" {
		t.Error("Set strings first index")
	}
	if v.([]string)[1] != "u" {
		t.Error("Set strings second index")
	}
	// mixed slice
	d.Set("mixed", []interface{}{
		1,
		"1",
	})
	v, _ = d.Get("mixed")
	if v.([]interface{})[0].(int) != 1 {
		t.Error("Set mixed int")
	}
	if v.([]interface{})[1].(string) != "1" {
		t.Error("Set mixed string")
	}
	// overriding existing key
	d.Set("number", 4)
	v, _ = d.Get("number")
	if v.(int) != 4 {
		t.Error("Override existing key")
	}
	// Keys
	expectedKeys := []string{
		"number",
		"string",
		"strings",
		"mixed",
	}
	for i, key := range d.Keys {
		if key != expectedKeys[i] {
			t.Error("Keys method", key, "!=", expectedKeys[i])
		}
	}
	for i, key := range expectedKeys {
		if key != expectedKeys[i] {
			t.Error("Keys method", key, "!=", expectedKeys[i])
		}
	}
}