package wireguard

import "testing"

func TestSimplePropertyInInterface(
	t *testing.T,
) {
	sample := dedent(`
[Interface]

`)
	parser := createWireguardParser()
	parser.parseFromString(sample)

	completions, err := parser.Sections[0].getCompletionsForEmptyLine()

	if err != nil {
		t.Fatalf("getCompletionsForEmptyLine failed with error: %v", err)
	}

	if len(completions) != len(interfaceOptions) {
		t.Fatalf("getCompletionsForEmptyLine: Expected %v completions, but got %v", len(interfaceOptions), len(completions))
	}
}

func TestSimpleOneExistingPropertyInInterface(
	t *testing.T,
) {
	sample := dedent(`
[Interface]
PrivateKey = 1234567890

`)
	parser := createWireguardParser()
	parser.parseFromString(sample)

	completions, err := parser.Sections[0].getCompletionsForEmptyLine()

	if err != nil {
		t.Fatalf("getCompletionsForEmptyLine failed with error: %v", err)
	}

	expected := len(interfaceOptions) - 1
	if len(completions) != expected {
		t.Fatalf("getCompletionsForEmptyLine: Expected %v completions, but got %v", expected, len(completions))
	}
}

func TestEmptyRootCompletionsWork(
	t *testing.T,
) {
	sample := dedent(`
	`)

	parser := createWireguardParser()
	parser.parseFromString(sample)

	completions := parser.getRootCompletionsForEmptyLine()

	if len(completions) != 2 {
		t.Fatalf("getRootCompletionsForEmptyLine: Expected 2 completions, but got %v", len(completions))
	}
}

func TestInterfaceSectionRootCompletionsBeforeWork(
	t *testing.T,
) {
	sample := dedent(`

[Interface]
`)
	parser := createWireguardParser()
	parser.parseFromString(sample)

	completions := parser.getRootCompletionsForEmptyLine()

	if len(completions) != 1 {
		t.Fatalf("getRootCompletionsForEmptyLine: Expected 1 completions, but got %v", len(completions))
	}
}

func TestInterfaceAndPeerSectionRootCompletionsWork(
	t *testing.T,
) {
	sample := dedent(`
[Interface]

[Peer]
`)
	parser := createWireguardParser()
	parser.parseFromString(sample)

	completions := parser.getRootCompletionsForEmptyLine()

	if len(completions) != 1 {
		t.Fatalf("getRootCompletionsForEmptyLine: Expected 1 completions, but got %v", len(completions))
	}
}