fix(grpc): add more checks for tag validation

This commit is contained in:
Adrien Raffin-Caboisse 2022-07-25 11:25:20 +02:00
parent c46a34e6b8
commit c90e862460
No known key found for this signature in database
GPG key ID: 7FB60532DEBEAD6A
3 changed files with 60 additions and 6 deletions

42
grpcv1_test.go Normal file
View file

@ -0,0 +1,42 @@
package headscale
import "testing"
func Test_validateTag(t *testing.T) {
type args struct {
tag string
}
tests := []struct {
name string
args args
wantErr bool
}{
{
name: "valid tag",
args: args{tag: "tag:test"},
wantErr: false,
},
{
name: "tag without tag prefix",
args: args{tag: "test"},
wantErr: true,
},
{
name: "uppercase tag",
args: args{tag: "tag:tEST"},
wantErr: true,
},
{
name: "tag that contains space",
args: args{tag: "tag:this is a spaced tag"},
wantErr: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if err := validateTag(tt.args.tag); (err != nil) != tt.wantErr {
t.Errorf("validateTag() error = %v, wantErr %v", err, tt.wantErr)
}
})
}
}