in fixtures.go [998:1032]
func (t *TestWorkflow) appendCreateFirewallStep(firewallName, networkName, protocol string, ports, ranges []string) (*daisy.Step, *daisy.FirewallRule, error) {
if ranges == nil {
ranges = []string{DefaultSourceRange}
}
firewall := &daisy.FirewallRule{
Firewall: compute.Firewall{
Name: firewallName,
Network: networkName,
SourceRanges: ranges,
Allowed: []*compute.FirewallAllowed{
{
IPProtocol: protocol,
Ports: ports,
},
},
},
}
createFirewallRules := &daisy.CreateFirewallRules{}
*createFirewallRules = append(*createFirewallRules, firewall)
createFirewallStep, ok := t.wf.Steps[createFirewallStepName]
if ok {
// append to existing step.
*createFirewallStep.CreateFirewallRules = append(*createFirewallStep.CreateFirewallRules, firewall)
} else {
var err error
createFirewallStep, err = t.wf.NewStep(createFirewallStepName)
if err != nil {
return nil, nil, err
}
createFirewallStep.CreateFirewallRules = createFirewallRules
}
return createFirewallStep, firewall, nil
}