github.com/lfch/etcd-io/tests/v3@v3.0.0-20221004140520-eac99acd3e9d/e2e/ctl_v3_txn_test.go (about) 1 // Copyright 2016 The etcd Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package e2e 16 17 import ( 18 "github.com/lfch/etcd-io/tests/v3/framework/e2e" 19 ) 20 21 type txnRequests struct { 22 compare []string 23 ifSucess []string 24 ifFail []string 25 results []string 26 } 27 28 func ctlV3Txn(cx ctlCtx, rqs txnRequests) error { 29 // TODO: support non-interactive mode 30 cmdArgs := append(cx.PrefixArgs(), "txn") 31 if cx.interactive { 32 cmdArgs = append(cmdArgs, "--interactive") 33 } 34 proc, err := e2e.SpawnCmd(cmdArgs, cx.envMap) 35 if err != nil { 36 return err 37 } 38 _, err = proc.Expect("compares:") 39 if err != nil { 40 return err 41 } 42 for _, req := range rqs.compare { 43 if err = proc.Send(req + "\r"); err != nil { 44 return err 45 } 46 } 47 if err = proc.Send("\r"); err != nil { 48 return err 49 } 50 51 _, err = proc.Expect("success requests (get, put, del):") 52 if err != nil { 53 return err 54 } 55 for _, req := range rqs.ifSucess { 56 if err = proc.Send(req + "\r"); err != nil { 57 return err 58 } 59 } 60 if err = proc.Send("\r"); err != nil { 61 return err 62 } 63 64 _, err = proc.Expect("failure requests (get, put, del):") 65 if err != nil { 66 return err 67 } 68 for _, req := range rqs.ifFail { 69 if err = proc.Send(req + "\r"); err != nil { 70 return err 71 } 72 } 73 if err = proc.Send("\r"); err != nil { 74 return err 75 } 76 77 for _, line := range rqs.results { 78 _, err = proc.Expect(line) 79 if err != nil { 80 return err 81 } 82 } 83 return proc.Close() 84 }