drone-yaml/yaml/port_test.go

50 lines
1001 B
Go
Raw Normal View History

2019-02-10 19:00:16 +00:00
// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
2019-02-10 19:00:16 +00:00
2019-01-22 23:44:17 +00:00
package yaml
import (
"testing"
2019-05-30 01:21:23 +00:00
"github.com/buildkite/yaml"
2019-01-22 23:44:17 +00:00
)
func TestPort(t *testing.T) {
tests := []struct {
yaml string
port int
host int
protocol string
}{
{
yaml: "80",
port: 80,
},
{
yaml: "{ port: 80, host: 8080, protocol: TCP }",
port: 80,
host: 8080,
protocol: "TCP",
},
}
for _, test := range tests {
in := []byte(test.yaml)
out := new(Port)
err := yaml.Unmarshal(in, out)
if err != nil {
t.Error(err)
return
}
if got, want := out.Port, test.port; got != want {
t.Errorf("Want Port %d, got %d", want, got)
}
if got, want := out.Host, test.host; got != want {
t.Errorf("Want Host %d, got %d", want, got)
}
if got, want := out.Protocol, test.protocol; got != want {
t.Errorf("Want Host %s, got %s", want, got)
}
}
}