test(pkg/integration): show users

Signed-off-by: Jeronimo Irazabal <[email protected]>
codenotary · Nov 8, 2023 · 932d73f · 932d73f
1 parent 58dfcca
commit 932d73f
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/embedded/sql/engine_test.go b/embedded/sql/engine_test.go
@@ -6043,6 +6043,16 @@ func TestMultiDBCatalogQueries(t *testing.T) {
 			require.NoError(t, err)
 		})
 
+		t.Run("show users", func(t *testing.T) {
+			r, err := engine.Query(context.Background(), nil, "SHOW USERS", nil)
+			require.NoError(t, err)
+
+			defer r.Close()
+
+			_, err = r.Read(context.Background())
+			require.ErrorIs(t, err, ErrNoMoreRows)
+		})
+
 		t.Run("query databases using conditions with table and column aliasing", func(t *testing.T) {
 			r, err := engine.Query(context.Background(), nil, "SELECT dbs.name as dbname FROM DATABASES() as dbs WHERE name LIKE 'db*'", nil)
 			require.NoError(t, err)

diff --git a/pkg/integration/session_test.go b/pkg/integration/session_test.go
@@ -197,3 +197,20 @@ func TestSession_CreateDBFromSQLStmts(t *testing.T) {
 	err = client.CloseSession(context.Background())
 	require.NoError(t, err)
 }
+
+func TestSession_ListUSersFromSQLStmts(t *testing.T) {
+	_, client, ctx := setupTestServerAndClient(t)
+
+	_, err := client.SQLExec(ctx, "CREATE DATABASE db1", nil)
+	require.NoError(t, err)
+
+	err = client.CreateUser(ctx, []byte("user1"), []byte("user1Password!"), 1, "db1")
+	require.NoError(t, err)
+
+	res, err := client.SQLQuery(ctx, "SHOW USERS", nil, false)
+	require.NoError(t, err)
+	require.Len(t, res.Rows, 1)
+
+	err = client.CloseSession(context.Background())
+	require.NoError(t, err)
+}