3adcf11b45
This replaces embedding of JavaScript code into the mock driver that runs inside the cluster with Go callbacks which run inside the e2e.test suite itself. In contrast to the JavaScript hooks, they have direct access to all parameters and can fabricate arbitrary responses, not just error codes. Because the callbacks run in the same process as the test itself, it is possible to set up two-way communication via shared variables or channels. This opens the door for writing better tests. Some of the existing tests that poll mock driver output could be simplified, but that can be addressed later. For now, only tests using hooks use embedding. How gRPC calls are retrieved is abstracted behind the CSIMockTestDriver interface, so tests don't need to be modified when switching between embedding and remote mock driver.
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
/*
|
|
Copyright 2021 The Kubernetes Authors.
|
|
|
|
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
|
|
|
|
http://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 driver
|
|
|
|
import (
|
|
"net"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type MockCSIDriverServers struct {
|
|
Controller *MockControllerServer
|
|
Identity *MockIdentityServer
|
|
Node *MockNodeServer
|
|
}
|
|
|
|
type MockCSIDriver struct {
|
|
CSIDriver
|
|
conn *grpc.ClientConn
|
|
interceptor grpc.UnaryServerInterceptor
|
|
}
|
|
|
|
func NewMockCSIDriver(servers *MockCSIDriverServers, interceptor grpc.UnaryServerInterceptor) *MockCSIDriver {
|
|
return &MockCSIDriver{
|
|
CSIDriver: CSIDriver{
|
|
servers: &CSIDriverServers{
|
|
Controller: servers.Controller,
|
|
Node: servers.Node,
|
|
Identity: servers.Identity,
|
|
},
|
|
},
|
|
interceptor: interceptor,
|
|
}
|
|
}
|
|
|
|
// StartOnAddress starts a new gRPC server listening on given address.
|
|
func (m *MockCSIDriver) StartOnAddress(network, address string) error {
|
|
l, err := net.Listen(network, address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := m.CSIDriver.Start(l, m.interceptor); err != nil {
|
|
l.Close()
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Start starts a new gRPC server listening on a random TCP loopback port.
|
|
func (m *MockCSIDriver) Start() error {
|
|
// Listen on a port assigned by the net package
|
|
return m.StartOnAddress("tcp", "127.0.0.1:0")
|
|
}
|
|
|
|
func (m *MockCSIDriver) Close() {
|
|
m.conn.Close()
|
|
m.server.Stop()
|
|
}
|