package main

import (
	"fmt"
	"testing"
)

func TestParse(t *testing.T) {
	cases := []string{
		"[1,2]",
		"[[3,2],1]",
		"[[3,2],[4,5]]",
		"[[3,2],[[7,8],5]]",
		"[[1,2],3]",
		"[9,[8,7]]",
		"[[1,9],[8,5]]",
		"[[[[1,2],[3,4]],[[5,6],[7,8]]],9]",
		"[[[9,[3,8]],[[0,9],6]],[[[3,7],[4,9]],3]]",
		"[[[[1,3],[5,3]],[[1,3],[8,7]]],[[[4,9],[6,9]],[[8,2],[7,3]]]]",
	}
	for _, input := range cases {
		p := Parse(input)
		if input != p.String() {
			t.Log(p.String())
			t.Fail()
		}
	}
}

func TestMustExplode(t *testing.T) {
	cases := map[string]bool{
		"[1,2]":                             false,
		"[[[[2,4],4],[2,3]],1]":             false,
		"[[[[[4,3],4],4],1],3]":             true,
		"[2,[1,[4,[3,[4,3]]]]]":             true,
		"[[[[0,7],4],[7,[[8,4],9]]],[1,1]]": true,
	}
	for input, output := range cases {
		if MustExplode(0, Parse(input)) != output {
			t.Log(input)
			t.Fail()
		}
	}
}

func TestAdd(t *testing.T) {
	cases := map[string][]*pair{
		"[[1,2],[[3,4],5]]": {
			&pair{left: 1, right: 2},
			&pair{leftp: &pair{left: 3, right: 4}, right: 5},
		},
	}
	for answer, pairs := range cases {
		r := Add(pairs[0], pairs[1])
		if r.String() != answer {
			fmt.Printf("%s %s", r, answer)
			t.Fail()
		}
	}
}

func TestMagnitude(t *testing.T) {
	cases := map[int]*pair{
		29: {left: 9, right: 1},
		21: {left: 1, right: 9},
		129: {
			leftp:  &pair{left: 9, right: 1},
			rightp: &pair{left: 1, right: 9},
		},
		143: {
			leftp:  &pair{left: 1, right: 2},
			rightp: &pair{leftp: &pair{left: 3, right: 4}, right: 5},
		},
	}
	for answer, pair := range cases {
		r := pair.Magnitude()
		if r != answer {
			fmt.Printf("%d %d", r, answer)
			t.Fail()
		}
	}
}