From 6bdbe23e9f9327c98d78ce2a8b7f2138d9a9d8ce Mon Sep 17 00:00:00 2001
From: dragondriver <jiquan.long@zilliz.com>
Date: Wed, 8 Dec 2021 14:51:18 +0800
Subject: [PATCH] Add ut for DropAlias with a short context (#12910)

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
---
 internal/proxy/proxy_test.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/internal/proxy/proxy_test.go b/internal/proxy/proxy_test.go
index 40e0618207..7b380e54a3 100644
--- a/internal/proxy/proxy_test.go
+++ b/internal/proxy/proxy_test.go
@@ -2461,6 +2461,14 @@ func TestProxy(t *testing.T) {
 		assert.NotEqual(t, commonpb.ErrorCode_Success, resp.ErrorCode)
 	})
 
+	wg.Add(1)
+	t.Run("DropAlias fail, timeout", func(t *testing.T) {
+		defer wg.Done()
+		resp, err := proxy.DropAlias(shortCtx, &milvuspb.DropAliasRequest{})
+		assert.NoError(t, err)
+		assert.NotEqual(t, commonpb.ErrorCode_Success, resp.ErrorCode)
+	})
+
 	wg.Wait()
 	cancel()
 }