tree_test.go

package main

import (
	"bytes"
	"testing"
)

// TODO(gulyasm): write multi lvl draw
func TestDrawTree(t *testing.T) {
	t.SkipNow()
	var json = []byte(`
	{
		"alma": 1, 
		"barack": {
			"barack_1": 12, 
			"barack_2": "sdfsf", 
			"barack_3": [1,2,3],
			"wazz_barack_4": "sdsfe"
		},
		"wazz": 1 
	}`)
	var expected = `
root
├─ alma
├── barack
│   ├─ barack_1
│   ├─ barack_2
│   ├─ barack_3
│   │  ├─ [0]
│   │  ├─ [1]
│   │  └─ [2]
│   └─ wazz_barack_4
└── wazz
`
	tree, err := fromBytes(json)
	if err != nil {
		t.Fatalf("failed to convert JSON to tree")
	}
	var result bytes.Buffer
	tree.draw(&result, 2, 0)
	if expected != result.String() {
		t.Fatalf("tree drawing failed. Result tree:\n%s", result.String())
	}
}

func TestListTree(t *testing.T) {
	raw := []byte(`[1,234,35]`)
	tree, err := fromBytes(raw)
	if err != nil {
		t.Fatalf("failed to convert JSON to tree")
	}
	v, ok := tree.(*listNode)
	if !ok {
		t.Fatalf("root element should be a listNode")
	}
	if len(v.data) != 3 {
		t.Fatalf("root element should have 3 children")
	}
}

func TestEmpty(t *testing.T) {
	var tp treePosition = []string{"alma", "barack", "banan"}
	if tp.empty() {
		t.Fatalf("non-empty position returned true for Empty")
	}
}

func TestEmpty2(t *testing.T) {
	var tp treePosition = []string{}
	if !tp.empty() {
		t.Fatalf("empty position returned false for Empty")
	}
}

// TODO(gulyasm): nicer fail messages
func TestShift(t *testing.T) {
	var tp treePosition = []string{"alma", "barack", "banan"}
	expected := treePosition([]string{"barack", "banan"})
	new := tp.shift()
	if len(new) != 2 {
		t.Fatalf("length is not 2 after Shift")
	}
	for i := 0; i < len(expected); i++ {
		if new[i] != expected[i] {
			t.Fatalf("shift did not return what was expected")
		}
	}
}