github.com/cockroachdb/cockroach@v20.2.0-alpha.1+incompatible/pkg/sql/sqlbase/database_test.go (about)

     1  // Copyright 2020 The Cockroach Authors.
     2  //
     3  // Use of this software is governed by the Business Source License
     4  // included in the file licenses/BSL.txt.
     5  //
     6  // As of the Change Date specified in that file, in accordance with
     7  // the Business Source License, use of this software will be governed
     8  // by the Apache License, Version 2.0, included in the file
     9  // licenses/APL.txt.
    10  
    11  package sqlbase
    12  
    13  import (
    14  	"testing"
    15  
    16  	"github.com/cockroachdb/cockroach/pkg/sql/parser"
    17  	"github.com/cockroachdb/cockroach/pkg/sql/sem/tree"
    18  	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
    19  )
    20  
    21  func TestMakeDatabaseDesc(t *testing.T) {
    22  	defer leaktest.AfterTest(t)()
    23  
    24  	stmt, err := parser.ParseOne("CREATE DATABASE test")
    25  	if err != nil {
    26  		t.Fatal(err)
    27  	}
    28  	desc := MakeDatabaseDesc(stmt.AST.(*tree.CreateDatabase))
    29  	if desc.Name != "test" {
    30  		t.Fatalf("expected Name == test, got %s", desc.Name)
    31  	}
    32  	// ID is not set yet.
    33  	if desc.ID != 0 {
    34  		t.Fatalf("expected ID == 0, got %d", desc.ID)
    35  	}
    36  	if len(desc.GetPrivileges().Users) != 2 {
    37  		t.Fatalf("wrong number of privilege users, expected 2, got: %d", len(desc.GetPrivileges().Users))
    38  	}
    39  }