mirror of
https://github.com/Myzel394/config-lsp.git
synced 2025-06-18 23:15:26 +02:00
360 lines
12 KiB
Go
360 lines
12 KiB
Go
package ast
|
|
|
|
import (
|
|
"config-lsp/utils"
|
|
"testing"
|
|
)
|
|
|
|
func TestSSHConfigParserExample(
|
|
t *testing.T,
|
|
) {
|
|
input := utils.Dedent(`
|
|
HostName 1.2.3.4
|
|
User root
|
|
`)
|
|
p := NewSSHConfig()
|
|
errors := p.Parse(input)
|
|
|
|
if len(errors) != 0 {
|
|
t.Fatalf("Expected no errors, got %v", errors)
|
|
}
|
|
|
|
if !(p.Options.Size() == 2 &&
|
|
len(utils.KeysOfMap(p.CommentLines)) == 0) {
|
|
t.Errorf("Expected 2 options and no comment lines, but got: %v, %v", p.Options, p.CommentLines)
|
|
}
|
|
|
|
rawFirstEntry, _ := p.Options.Get(uint32(0))
|
|
firstEntry := rawFirstEntry.(*SSHOption)
|
|
|
|
if !(firstEntry.Value.Value == "HostName 1.2.3.4" &&
|
|
firstEntry.LocationRange.Start.Line == 0 &&
|
|
firstEntry.LocationRange.End.Line == 0 &&
|
|
firstEntry.LocationRange.Start.Character == 0 &&
|
|
firstEntry.LocationRange.End.Character == 16 &&
|
|
firstEntry.Key.Value.Value == "HostName" &&
|
|
firstEntry.Key.LocationRange.Start.Character == 0 &&
|
|
firstEntry.Key.LocationRange.End.Character == 8 &&
|
|
firstEntry.OptionValue.Value.Value == "1.2.3.4" &&
|
|
firstEntry.OptionValue.LocationRange.Start.Character == 9 &&
|
|
firstEntry.OptionValue.LocationRange.End.Character == 16) {
|
|
t.Errorf("Expected first entry to be HostName 1.2.3.4, but got: %v", firstEntry)
|
|
}
|
|
|
|
rawSecondEntry, _ := p.Options.Get(uint32(1))
|
|
secondEntry := rawSecondEntry.(*SSHOption)
|
|
|
|
if !(secondEntry.Value.Value == "User root" &&
|
|
secondEntry.LocationRange.Start.Line == 1 &&
|
|
secondEntry.LocationRange.End.Line == 1 &&
|
|
secondEntry.LocationRange.Start.Character == 0 &&
|
|
secondEntry.LocationRange.End.Character == 9 &&
|
|
secondEntry.Key.Value.Value == "User" &&
|
|
secondEntry.Key.LocationRange.Start.Character == 0 &&
|
|
secondEntry.Key.LocationRange.End.Character == 4 &&
|
|
secondEntry.OptionValue.Value.Value == "root" &&
|
|
secondEntry.OptionValue.LocationRange.Start.Character == 5 &&
|
|
secondEntry.OptionValue.LocationRange.End.Character == 9) {
|
|
t.Errorf("Expected second entry to be User root, but got: %v", secondEntry)
|
|
}
|
|
}
|
|
|
|
func TestMatchSimpleBlock(
|
|
t *testing.T,
|
|
) {
|
|
input := utils.Dedent(`
|
|
Hostname 1.2.3.4
|
|
|
|
Match originalhost "192.168.0.1"
|
|
User root
|
|
`)
|
|
p := NewSSHConfig()
|
|
errors := p.Parse(input)
|
|
|
|
if len(errors) != 0 {
|
|
t.Fatalf("Expected no errors, got %v", errors)
|
|
}
|
|
|
|
if !(p.Options.Size() == 2 &&
|
|
len(utils.KeysOfMap(p.CommentLines)) == 0) {
|
|
t.Errorf("Expected 2 option and no comment lines, but got: %v entries, %v comment lines", p.Options.Size(), len(p.CommentLines))
|
|
}
|
|
|
|
rawFirstEntry, _ := p.Options.Get(uint32(0))
|
|
firstEntry := rawFirstEntry.(*SSHOption)
|
|
|
|
if !(firstEntry.Value.Value == "Hostname 1.2.3.4" &&
|
|
firstEntry.LocationRange.Start.Line == 0 &&
|
|
firstEntry.LocationRange.End.Line == 0 &&
|
|
firstEntry.LocationRange.Start.Character == 0 &&
|
|
firstEntry.LocationRange.End.Character == 16 &&
|
|
firstEntry.Key.Value.Value == "Hostname" &&
|
|
firstEntry.Key.LocationRange.Start.Character == 0 &&
|
|
firstEntry.Key.LocationRange.End.Character == 8 &&
|
|
firstEntry.OptionValue.Value.Value == "1.2.3.4" &&
|
|
firstEntry.OptionValue.LocationRange.Start.Character == 9 &&
|
|
firstEntry.OptionValue.LocationRange.End.Character == 16) {
|
|
t.Errorf("Expected first entry to be Hostname 1.2.3.4, but got: %v", firstEntry)
|
|
}
|
|
|
|
rawSecondEntry, _ := p.Options.Get(uint32(2))
|
|
secondEntry := rawSecondEntry.(*SSHMatchBlock)
|
|
|
|
if !(secondEntry.Options.Size() == 1 &&
|
|
secondEntry.LocationRange.Start.Line == 2 &&
|
|
secondEntry.LocationRange.End.Line == 3 &&
|
|
secondEntry.LocationRange.Start.Character == 0 &&
|
|
secondEntry.LocationRange.End.Character == 10 &&
|
|
secondEntry.MatchOption.OptionValue.Value.Raw == "originalhost \"192.168.0.1\"" &&
|
|
secondEntry.MatchOption.OptionValue.LocationRange.Start.Character == 6 &&
|
|
secondEntry.MatchOption.OptionValue.LocationRange.End.Character == 32) {
|
|
t.Errorf("Expected second entry to be Match originalhost \"192.168.0.1\", but got: %v; options amount: %d", secondEntry, secondEntry.Options.Size())
|
|
}
|
|
|
|
rawThirdEntry, _ := secondEntry.Options.Get(uint32(3))
|
|
thirdEntry := rawThirdEntry.(*SSHOption)
|
|
if !(thirdEntry.Value.Raw == "\tUser root" &&
|
|
thirdEntry.LocationRange.Start.Line == 3 &&
|
|
thirdEntry.LocationRange.End.Line == 3 &&
|
|
thirdEntry.LocationRange.Start.Character == 0 &&
|
|
thirdEntry.LocationRange.End.Character == 10 &&
|
|
thirdEntry.Key.Value.Value == "User" &&
|
|
thirdEntry.Key.LocationRange.Start.Character == 1 &&
|
|
thirdEntry.Key.LocationRange.End.Character == 5 &&
|
|
thirdEntry.OptionValue.Value.Value == "root" &&
|
|
thirdEntry.OptionValue.LocationRange.Start.Character == 6 &&
|
|
thirdEntry.OptionValue.LocationRange.End.Character == 10) {
|
|
t.Errorf("Expected third entry to be User root, but got: %v", thirdEntry)
|
|
}
|
|
}
|
|
|
|
func TestSimpleHostBlock(
|
|
t *testing.T,
|
|
) {
|
|
input := utils.Dedent(`
|
|
Ciphers 3des-cbc
|
|
|
|
Host example.com
|
|
Port 22
|
|
`)
|
|
p := NewSSHConfig()
|
|
errors := p.Parse(input)
|
|
|
|
if len(errors) != 0 {
|
|
t.Fatalf("Expected no errors, got %v", errors)
|
|
}
|
|
|
|
if !(p.Options.Size() == 2 &&
|
|
len(utils.KeysOfMap(p.CommentLines)) == 0) {
|
|
t.Errorf("Expected 2 option and no comment lines, but got: %v entries, %v comment lines", p.Options.Size(), len(p.CommentLines))
|
|
}
|
|
|
|
rawFirstEntry, _ := p.Options.Get(uint32(0))
|
|
firstEntry := rawFirstEntry.(*SSHOption)
|
|
if !(firstEntry.Value.Value == "Ciphers 3des-cbc") {
|
|
t.Errorf("Expected first entry to be Ciphers 3des-cbc, but got: %v", firstEntry)
|
|
}
|
|
|
|
rawSecondEntry, _ := p.Options.Get(uint32(2))
|
|
secondEntry := rawSecondEntry.(*SSHHostBlock)
|
|
if !(secondEntry.Options.Size() == 1 &&
|
|
secondEntry.LocationRange.Start.Line == 2 &&
|
|
secondEntry.LocationRange.Start.Character == 0 &&
|
|
secondEntry.LocationRange.End.Line == 3 &&
|
|
secondEntry.LocationRange.End.Character == 8) {
|
|
t.Errorf("Expected second entry to be Host example.com, but got: %v", secondEntry)
|
|
}
|
|
|
|
rawThirdEntry, _ := secondEntry.Options.Get(uint32(3))
|
|
thirdEntry := rawThirdEntry.(*SSHOption)
|
|
if !(thirdEntry.Value.Raw == "\tPort 22" &&
|
|
thirdEntry.Key.Value.Raw == "Port" &&
|
|
thirdEntry.OptionValue.Value.Raw == "22" &&
|
|
thirdEntry.LocationRange.Start.Line == 3 &&
|
|
thirdEntry.LocationRange.Start.Character == 0 &&
|
|
thirdEntry.LocationRange.End.Line == 3 &&
|
|
thirdEntry.LocationRange.End.Character == 8) {
|
|
t.Errorf("Expected third entry to be Port 22, but got: %v", thirdEntry)
|
|
}
|
|
|
|
rawFourthEntry, _ := p.Options.Get(uint32(3))
|
|
|
|
if !(rawFourthEntry == nil) {
|
|
t.Errorf("Expected fourth entry to be nil, but got: %v", rawFourthEntry)
|
|
}
|
|
}
|
|
|
|
func TestComplexExample(
|
|
t *testing.T,
|
|
) {
|
|
input := utils.Dedent(`
|
|
Host laptop
|
|
HostName laptop.lan
|
|
|
|
Match originalhost laptop exec "[[ $(/usr/bin/dig +short laptop.lan) == '' ]]"
|
|
HostName laptop.sdn
|
|
`)
|
|
p := NewSSHConfig()
|
|
errors := p.Parse(input)
|
|
|
|
if len(errors) != 0 {
|
|
t.Fatalf("Expected no errors, got %v", errors)
|
|
}
|
|
|
|
if !(p.Options.Size() == 2 &&
|
|
len(utils.KeysOfMap(p.CommentLines)) == 0) {
|
|
t.Errorf("Expected 2 option and no comment lines, but got: %v entries, %v comment lines", p.Options.Size(), len(p.CommentLines))
|
|
}
|
|
|
|
rawFirstEntry, _ := p.Options.Get(uint32(0))
|
|
firstBlock := rawFirstEntry.(*SSHHostBlock)
|
|
if !(firstBlock.Options.Size() == 1 &&
|
|
firstBlock.LocationRange.Start.Line == 0 &&
|
|
firstBlock.LocationRange.Start.Character == 0 &&
|
|
firstBlock.LocationRange.End.Line == 1 &&
|
|
firstBlock.LocationRange.End.Character == 23) {
|
|
t.Errorf("Expected first entry to be Host laptop, but got: %v", firstBlock)
|
|
}
|
|
|
|
rawSecondEntry, _ := firstBlock.Options.Get(uint32(1))
|
|
secondOption := rawSecondEntry.(*SSHOption)
|
|
if !(secondOption.Value.Raw == " HostName laptop.lan" &&
|
|
secondOption.Key.Value.Raw == "HostName" &&
|
|
secondOption.OptionValue.Value.Raw == "laptop.lan" &&
|
|
secondOption.LocationRange.Start.Line == 1 &&
|
|
secondOption.LocationRange.Start.Character == 0 &&
|
|
secondOption.Key.LocationRange.Start.Character == 4 &&
|
|
secondOption.LocationRange.End.Line == 1 &&
|
|
secondOption.LocationRange.End.Character == 23) {
|
|
t.Errorf("Expected second entry to be HostName laptop.lan, but got: %v", secondOption)
|
|
}
|
|
|
|
rawThirdEntry, _ := p.Options.Get(uint32(3))
|
|
secondBlock := rawThirdEntry.(*SSHMatchBlock)
|
|
if !(secondBlock.Options.Size() == 1 &&
|
|
secondBlock.LocationRange.Start.Line == 3 &&
|
|
secondBlock.LocationRange.Start.Character == 0 &&
|
|
secondBlock.LocationRange.End.Line == 4 &&
|
|
secondBlock.LocationRange.End.Character == 23) {
|
|
t.Errorf("Expected second entry to be Match originalhost laptop exec \"[[ $(/usr/bin/dig +short laptop.lan) == '' ]]\", but got: %v", secondBlock)
|
|
}
|
|
|
|
if !(secondBlock.MatchOption.LocationRange.End.Character == 78) {
|
|
t.Errorf("Expected second entry to be Match originalhost laptop exec \"[[ $(/usr/bin/dig +short laptop.lan) == '' ]]\", but got: %v", secondBlock)
|
|
}
|
|
|
|
rawFourthEntry, _ := secondBlock.Options.Get(uint32(4))
|
|
thirdOption := rawFourthEntry.(*SSHOption)
|
|
if !(thirdOption.Value.Raw == " HostName laptop.sdn" &&
|
|
thirdOption.Key.Value.Raw == "HostName" &&
|
|
thirdOption.OptionValue.Value.Raw == "laptop.sdn" &&
|
|
thirdOption.LocationRange.Start.Line == 4 &&
|
|
thirdOption.LocationRange.Start.Character == 0 &&
|
|
thirdOption.Key.LocationRange.Start.Character == 4 &&
|
|
thirdOption.LocationRange.End.Line == 4 &&
|
|
thirdOption.LocationRange.End.Character == 23) {
|
|
t.Errorf("Expected third entry to be HostName laptop.sdn, but got: %v", thirdOption)
|
|
}
|
|
}
|
|
|
|
func TestComplexBigExample(
|
|
t *testing.T,
|
|
) {
|
|
// From https://gist.github.com/zeloc/b9455b793b07898025db
|
|
input := utils.Dedent(`
|
|
### default for all ##
|
|
Host *
|
|
ForwardAgent no
|
|
ForwardX11 no
|
|
ForwardX11Trusted yes
|
|
User nixcraft
|
|
Port 22
|
|
Protocol 2
|
|
ServerAliveInterval 60
|
|
ServerAliveCountMax 30
|
|
|
|
## override as per host ##
|
|
Host server1
|
|
HostName server1.cyberciti.biz
|
|
User nixcraft
|
|
Port 4242
|
|
IdentityFile /nfs/shared/users/nixcraft/keys/server1/id_rsa
|
|
|
|
## Home nas server ##
|
|
Host nas01
|
|
HostName 192.168.1.100
|
|
User root
|
|
IdentityFile ~/.ssh/nas01.key
|
|
|
|
## Login AWS Cloud ##
|
|
Host aws.apache
|
|
HostName 1.2.3.4
|
|
User wwwdata
|
|
IdentityFile ~/.ssh/aws.apache.key
|
|
|
|
## Login to internal lan server at 192.168.0.251 via our public uk office ssh based gateway using ##
|
|
## $ ssh uk.gw.lan ##
|
|
Host uk.gw.lan uk.lan
|
|
HostName 192.168.0.251
|
|
User nixcraft
|
|
ProxyCommand ssh nixcraft@gateway.uk.cyberciti.biz nc %h %p 2> /dev/null
|
|
|
|
## Our Us Proxy Server ##
|
|
## Forward all local port 3128 traffic to port 3128 on the remote vps1.cyberciti.biz server ##
|
|
## $ ssh -f -N proxyus ##
|
|
Host proxyus
|
|
HostName vps1.cyberciti.biz
|
|
User breakfree
|
|
IdentityFile ~/.ssh/vps1.cyberciti.biz.key
|
|
LocalForward 3128 127.0.0.1:3128
|
|
`)
|
|
p := NewSSHConfig()
|
|
|
|
errors := p.Parse(input)
|
|
|
|
if len(errors) != 0 {
|
|
t.Fatalf("Expected no errors, got %v", errors)
|
|
}
|
|
|
|
if !(p.Options.Size() == 6 && len(utils.KeysOfMap(p.CommentLines)) == 9) {
|
|
t.Errorf("Expected 6 options and no comment lines, but got: %v options, %v comment lines", p.Options.Size(), len(p.CommentLines))
|
|
}
|
|
|
|
// Validate each Host block and its options
|
|
rawFirstEntry, _ := p.Options.Get(uint32(1))
|
|
firstBlock := rawFirstEntry.(*SSHHostBlock)
|
|
if !(firstBlock.Options.Size() == 8) {
|
|
t.Errorf("Expected 8 options for Host *, but got: %v", firstBlock.Options.Size())
|
|
}
|
|
|
|
rawSecondEntry, _ := p.Options.Get(uint32(12))
|
|
secondBlock := rawSecondEntry.(*SSHHostBlock)
|
|
if !(secondBlock.Options.Size() == 4) {
|
|
t.Errorf("Expected 4 options for Host server1, but got: %v", secondBlock.Options.Size())
|
|
}
|
|
|
|
rawThirdEntry, _ := p.Options.Get(uint32(19))
|
|
thirdBlock := rawThirdEntry.(*SSHHostBlock)
|
|
if !(thirdBlock.Options.Size() == 3) {
|
|
t.Errorf("Expected 2 options for Host nas01, but got: %v", thirdBlock.Options.Size())
|
|
}
|
|
|
|
rawFourthEntry, _ := p.Options.Get(uint32(25))
|
|
fourthBlock := rawFourthEntry.(*SSHHostBlock)
|
|
if !(fourthBlock.Options.Size() == 3) {
|
|
t.Errorf("Expected 2 options for Host aws.apache, but got: %v", fourthBlock.Options.Size())
|
|
}
|
|
|
|
rawFifthEntry, _ := p.Options.Get(uint32(32))
|
|
fifthBlock := rawFifthEntry.(*SSHHostBlock)
|
|
if !(fifthBlock.Options.Size() == 3) {
|
|
t.Errorf("Expected 3 options for Host uk.gw.lan uk.lan, but got: %v", fifthBlock.Options.Size())
|
|
}
|
|
|
|
rawSixthEntry, _ := p.Options.Get(uint32(40))
|
|
sixthBlock := rawSixthEntry.(*SSHHostBlock)
|
|
if !(sixthBlock.Options.Size() == 4) {
|
|
t.Errorf("Expected 4 options for Host proxyus, but got: %v", sixthBlock.Options.Size())
|
|
}
|
|
}
|