From 6aac35dd3216e7965c233ff1376f9181d3c1e9e2 Mon Sep 17 00:00:00 2001
From: AdamKorcz <adam@adalogics.com>
Date: Thu, 30 Dec 2021 15:48:05 +0000
Subject: [PATCH] v2auth: fix out of bounds in test

---
 server/etcdserver/api/v2auth/auth_test.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server/etcdserver/api/v2auth/auth_test.go b/server/etcdserver/api/v2auth/auth_test.go
index 1b664e0e8..cd9c66667 100644
--- a/server/etcdserver/api/v2auth/auth_test.go
+++ b/server/etcdserver/api/v2auth/auth_test.go
@@ -176,6 +176,11 @@ func (td *testDoer) Do(_ context.Context, req etcdserverpb.Request) (etcdserver.
 		}, nil
 	}
 	if (req.Method == "GET" || req.Method == "QGET") && td.get != nil {
+		if td.getindex >= len(td.get) {
+			return etcdserver.Response{}, &v2error.Error{
+				ErrorCode: v2error.EcodeKeyNotFound,
+			}
+		}
 		res := td.get[td.getindex]
 		if res.Event == nil {
 			td.getindex++
@@ -187,6 +192,11 @@ func (td *testDoer) Do(_ context.Context, req etcdserverpb.Request) (etcdserver.
 		return res, nil
 	}
 	if req.Method == "PUT" && td.put != nil {
+		if td.putindex >= len(td.put) {
+			return etcdserver.Response{}, &v2error.Error{
+				ErrorCode: v2error.EcodeKeyNotFound,
+			}
+		}
 		res := td.put[td.putindex]
 		if res.Event == nil {
 			td.putindex++