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.
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
/*
|
|
Copyright 2020 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 proxy
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"k8s.io/kubernetes/test/e2e/framework"
|
|
"k8s.io/kubernetes/test/e2e/storage/drivers/csi-test/mock/service"
|
|
)
|
|
|
|
type PodDirIO struct {
|
|
F *framework.Framework
|
|
Namespace string
|
|
PodName string
|
|
ContainerName string
|
|
}
|
|
|
|
var _ service.DirIO = PodDirIO{}
|
|
|
|
func (p PodDirIO) DirExists(path string) (bool, error) {
|
|
stdout, stderr, err := p.execute([]string{
|
|
"sh",
|
|
"-c",
|
|
fmt.Sprintf("if ! [ -e '%s' ]; then echo notexist; elif [ -d '%s' ]; then echo dir; else echo nodir; fi", path, path),
|
|
})
|
|
if err != nil {
|
|
return false, fmt.Errorf("error executing dir test commands: stderr=%q, %v", stderr, err)
|
|
}
|
|
switch stdout {
|
|
case "notexist":
|
|
return false, nil
|
|
case "nodir":
|
|
return false, fmt.Errorf("%s: not a directory", path)
|
|
case "dir":
|
|
return true, nil
|
|
default:
|
|
return false, fmt.Errorf("unexpected output from dir test commands: %q", stdout)
|
|
}
|
|
}
|
|
|
|
func (p PodDirIO) Mkdir(path string) error {
|
|
_, stderr, err := p.execute([]string{"mkdir", path})
|
|
if err != nil {
|
|
return fmt.Errorf("mkdir %q: stderr=%q, %v", path, stderr, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p PodDirIO) RemoveAll(path string) error {
|
|
_, stderr, err := p.execute([]string{"rm", "-rf", path})
|
|
if err != nil {
|
|
return fmt.Errorf("rm -rf %q: stderr=%q, %v", path, stderr, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p PodDirIO) execute(command []string) (string, string, error) {
|
|
return p.F.ExecWithOptions(framework.ExecOptions{
|
|
Command: command,
|
|
Namespace: p.Namespace,
|
|
PodName: p.PodName,
|
|
ContainerName: p.ContainerName,
|
|
CaptureStdout: true,
|
|
CaptureStderr: true,
|
|
Quiet: true,
|
|
})
|
|
}
|