-
Notifications
You must be signed in to change notification settings - Fork 34
/
ate.go
216 lines (195 loc) · 5.45 KB
/
ate.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
// Copyright 2019 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Package ate controls automated test equipment (ATE) for ONDATRA tests.
package ate
import (
"fmt"
"sync"
"time"
"golang.org/x/net/context"
"github.com/openconfig/ondatra/binding"
"github.com/openconfig/ondatra/internal/rawapis"
gpb "github.com/openconfig/gnmi/proto/gnmi"
opb "github.com/openconfig/ondatra/proto"
)
var (
mu sync.Mutex
ixias = make(map[binding.ATE]*ixATE)
)
// Topology is an ATE topology.
type Topology struct {
Interfaces []*opb.InterfaceConfig
LAGs []*opb.Lag
}
func ixiaForATE(ctx context.Context, ate binding.ATE) (*ixATE, error) {
mu.Lock()
defer mu.Unlock()
ix, ok := ixias[ate]
if !ok {
ixnet, err := rawapis.FetchIxNetwork(ctx, ate)
if err != nil {
return nil, err
}
ix, err = newIxATE(ctx, ate.Name(), ixnet)
if err != nil {
return nil, err
}
ixias[ate] = ix
}
return ix, nil
}
// PushTopology pushes a topology to an ATE.
func PushTopology(ctx context.Context, ate binding.ATE, top *Topology) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.PushTopology(ctx, top); err != nil {
return err
}
ix.FlushStats()
return nil
}
// UpdateTopology updates a topology on an ATE.
func UpdateTopology(ctx context.Context, ate binding.ATE, top *Topology, bgpPeerStateOnly bool) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
// TODO(team): Remove this branching once new Ixia config binding is used.
if bgpPeerStateOnly {
err = ix.UpdateBGPPeerStates(ctx, top.Interfaces)
} else {
err = ix.UpdateTopology(ctx, top)
}
if err != nil {
return err
}
ix.FlushStats()
return nil
}
// UpdateNetworks updates network groups in a topology on an ATE on the fly.
func UpdateNetworks(ctx context.Context, ate binding.ATE, top *Topology) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
return ix.UpdateNetworkGroups(ctx, top.Interfaces)
}
// StartProtocols starts control plane protocols on an ATE.
func StartProtocols(ctx context.Context, ate binding.ATE) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.StartProtocols(ctx); err != nil {
return fmt.Errorf("failed to start protocols: %w", err)
}
ix.FlushStats()
return nil
}
// StopProtocols stops control protocols on an ATE.
func StopProtocols(ctx context.Context, ate binding.ATE) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.StopProtocols(ctx); err != nil {
return err
}
ix.FlushStats()
return nil
}
// StartTraffic starts traffic flows on an ATE.
func StartTraffic(ctx context.Context, ate binding.ATE, flows []*opb.Flow) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.StartTraffic(ctx, flows); err != nil {
return err
}
ix.FlushStats()
return nil
}
// UpdateTraffic updates traffic flows an an ATE.
func UpdateTraffic(ctx context.Context, ate binding.ATE, flows []*opb.Flow) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.UpdateTraffic(ctx, flows); err != nil {
return err
}
ix.FlushStats()
return nil
}
// StopTraffic stops traffic flows on an ATE.
func StopTraffic(ctx context.Context, ate binding.ATE) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.StopAllTraffic(ctx); err != nil {
return err
}
ix.FlushStats()
return nil
}
// FetchGNMI returns the GNMI client for the Ixia.
func FetchGNMI(ctx context.Context, ate binding.ATE) (gpb.GNMIClient, error) {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return nil, err
}
return ix.FetchGNMI(ctx)
}
// SetPortState sets the state of a specified interface on the ATE.
func SetPortState(ctx context.Context, ate binding.ATE, port string, enabled *bool) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
return ix.SetPortState(ctx, port, enabled)
}
// SetLACPState sets the LACP state of a specified interface on the ATE.
func SetLACPState(ctx context.Context, ate binding.ATE, port string, enabled *bool) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
return ix.SetLACPState(ctx, port, enabled)
}
// SendBGPPeerNotification sends a notification from BGP peers.
func SendBGPPeerNotification(ctx context.Context, ate binding.ATE, peerIDs []uint32, code int, subCode int) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.SendBGPPeerNotification(ctx, peerIDs, code, subCode); err != nil {
return fmt.Errorf("failed to send notification: %w", err)
}
return nil
}
// SendBGPGracefulRestart sends a BGP graceful restart event to BGP peers.
func SendBGPGracefulRestart(ctx context.Context, ate binding.ATE, peerIDs []uint32, delay time.Duration) error {
ix, err := ixiaForATE(ctx, ate)
if err != nil {
return err
}
if err := ix.SendBGPGracefulRestart(ctx, peerIDs, delay); err != nil {
return fmt.Errorf("failed to send graceful restart: %w", err)
}
return nil
}