tests: Optimize checking failed writes

Signed-off-by: Marek Siarkowicz <siarkowicz@google.com>
etcd-io · Nov 6, 2022 · 622fd91 · 622fd91
1 parent 6bfaf8d
commit 622fd91
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/tests/linearizability/model.go b/tests/linearizability/model.go
@@ -101,12 +101,11 @@ func step(state EtcdState, request etcdRequest, response etcdResponse) (bool, Et
 		if state.Value == response.getData {
 			return true, state
 		}
-		for write := range state.FailedWrites {
-			if write == response.getData {
-				state.Value = response.getData
-				delete(state.FailedWrites, write)
-				return true, state
-			}
+		_, ok := state.FailedWrites[response.getData]
+		if ok {
+			state.Value = response.getData
+			delete(state.FailedWrites, response.getData)
+			return true, state
 		}
 	case Put:
 		if response.err == nil {