package main

import (
	"testing"
)

var parsePathsTests = []struct {
	path          string
	expectedPaths []string
}{
	{"a.b", []string{"a", "b"}},
	{"a.b[0]", []string{"a", "b", "0"}},
}

func testParsePath(t *testing.T) {
	for _, tt := range parsePathsTests {
		assertResultWithContext(t, tt.expectedPaths, parsePath(tt.path), tt)
	}
}

var nextYamlPathTests = []struct {
	path              string
	expectedElement   string
	expectedRemaining string
}{
	{"a.b", "a", "b"},
	{"a", "a", ""},
	{"a.b.c", "a", "b.c"},
	{"\"a.b\".c", "a.b", "c"},
	{"a.\"b.c\".d", "a", "\"b.c\".d"},
	{"[1].a.d", "1", "a.d"},
	{"a[0].c", "a", "[0].c"},
	{"[0]", "0", ""},
}

func TestNextYamlPath(t *testing.T) {
	for _, tt := range nextYamlPathTests {
		var element, remaining = nextYamlPath(tt.path)
		assertResultWithContext(t, tt.expectedElement, element, tt)
		assertResultWithContext(t, tt.expectedRemaining, remaining, tt)
	}
}