43 lines
883 B
Go
43 lines
883 B
Go
|
package initialize
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestFleetEnvironment(t *testing.T) {
|
||
|
|
cfg := make(FleetEnvironment, 0)
|
||
|
|
cfg["public-ip"] = "12.34.56.78"
|
||
|
|
|
||
|
|
env := cfg.String()
|
||
|
|
|
||
|
|
expect := `[Service]
|
||
|
|
Environment="FLEET_PUBLIC_IP=12.34.56.78"
|
||
|
|
`
|
||
|
|
|
||
|
|
if env != expect {
|
||
|
|
t.Errorf("Generated environment:\n%s\nExpected environment:\n%s", env, expect)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFleetUnit(t *testing.T) {
|
||
|
|
cfg := make(FleetEnvironment, 0)
|
||
|
|
u, err := cfg.Unit("/")
|
||
|
|
if u != nil {
|
||
|
|
t.Errorf("unexpectedly generated unit with empty FleetEnvironment")
|
||
|
|
}
|
||
|
|
|
||
|
|
cfg["public-ip"] = "12.34.56.78"
|
||
|
|
|
||
|
|
u, err = cfg.Unit("/")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("error generating fleet unit: %v", err)
|
||
|
|
}
|
||
|
|
if u == nil {
|
||
|
|
t.Fatalf("unexpectedly got nil unit generating fleet unit!")
|
||
|
|
}
|
||
|
|
if !u.Runtime {
|
||
|
|
t.Errorf("bad Runtime for generated fleet unit!")
|
||
|
|
}
|
||
|
|
if !u.DropIn {
|
||
|
|
t.Errorf("bad DropIn for generated fleet unit!")
|
||
|
|
}
|
||
|
|
}
|