vitalif
/
etcd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[e2e] Fix AuthEnable/Disable e2e test implementations 

Signed-off-by: Oleg Guba <oleg@dropbox.com>
dependabot/go_modules/go.uber.org/atomic-1.10.0
Oleg Guba 2022-10-20 19:15:31 -04:00
parent bf6a1d7264
commit a39374f33a
1 changed files with 32 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
tests/framework/e2e/etcdctl.go
View File

@ -466,15 +466,41 @@ func (ctl *EtcdctlV3) AlarmDisarm(ctx context.Context, _ *clientv3.AlarmMember)
}
func (ctl *EtcdctlV3) AuthEnable(ctx context.Context) (*clientv3.AuthEnableResponse, error) {
var resp clientv3.AuthEnableResponse
err := ctl.spawnJsonCmd(ctx, &resp, "auth", "enable")
return &resp, err
args := []string{"auth", "enable"}
cmd, err := SpawnCmd(append(ctl.cmdArgs(), args...), nil)
if err != nil {
return nil, err
}
err = cmd.Send(strings.Join(args, " "))
if err != nil {
return nil, err
}
defer cmd.Close()
_, err = cmd.ExpectWithContext(ctx, "Authentication Enabled")
if err != nil {
return nil, err
}
return &clientv3.AuthEnableResponse{}, nil
}
func (ctl *EtcdctlV3) AuthDisable(ctx context.Context) (*clientv3.AuthDisableResponse, error) {
var resp clientv3.AuthDisableResponse
err := ctl.spawnJsonCmd(ctx, &resp, "auth", "disable")
return &resp, err
args := []string{"auth", "disable"}
cmd, err := SpawnCmd(append(ctl.cmdArgs(), args...), nil)
if err != nil {
return nil, err
}
err = cmd.Send(strings.Join(args, " "))
if err != nil {
return nil, err
}
defer cmd.Close()
_, err = cmd.ExpectWithContext(ctx, "Authentication Disabled")
if err != nil {
return nil, err
}
return &clientv3.AuthDisableResponse{}, nil
}
func (ctl *EtcdctlV3) AuthStatus(ctx context.Context) (*clientv3.AuthStatusResponse, error) {