grant_helpers_internal_test.go

package resources

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestGrantIDFromString(t *testing.T) {
	r := require.New(t)
	// Vanilla
	id := "database_name|schema|view_name|privilege"
	grant, err := grantIDFromString(id)
	r.NoError(err)

	r.Equal("database_name", grant.ResourceName)
	r.Equal("schema", grant.SchemaName)
	r.Equal("view_name", grant.ObjectName)
	r.Equal("privilege", grant.Privilege)

	// No view
	id = "database_name|||privilege"
	grant, err = grantIDFromString(id)
	r.NoError(err)
	r.Equal("database_name", grant.ResourceName)
	r.Equal("", grant.SchemaName)
	r.Equal("", grant.ObjectName)
	r.Equal("privilege", grant.Privilege)

	// Bad ID -- not enough fields
	id = "database|name-privilege"
	_, err = grantIDFromString(id)
	r.Equal(fmt.Errorf("4 fields allowed"), err)

	// Bad ID -- privilege in wrong area
	id = "database||||name-privilege"
	_, err = grantIDFromString(id)
	r.Equal(fmt.Errorf("4 fields allowed"), err)

	// too many fields
	id = "database_name|schema|view_name|privilege|extra"
	_, err = grantIDFromString(id)
	r.Equal(fmt.Errorf("4 fields allowed"), err)

	// 0 lines
	id = ""
	_, err = grantIDFromString(id)
	r.Equal(fmt.Errorf("1 line per grant"), err)

	// 2 lines
	id = `database_name|schema|view_name|privilege
	database_name|schema|view_name|privilege`
	_, err = grantIDFromString(id)
	r.Equal(fmt.Errorf("1 line per grant"), err)
}

func TestGrantStruct(t *testing.T) {
	r := require.New(t)

	// Vanilla
	grant := &grantID{
		ResourceName: "database_name",
		SchemaName:   "schema",
		ObjectName:   "view_name",
		Privilege:    "priv",
	}
	gID, err := grant.String()
	r.NoError(err)
	r.Equal("database_name|schema|view_name|priv", gID)

	// Empty grant
	grant = &grantID{}
	gID, err = grant.String()
	r.NoError(err)
	r.Equal("|||", gID)

	// Grant with extra delimiters
	grant = &grantID{
		ResourceName: "database|name",
		SchemaName:   "schema|name",
		ObjectName:   "view|name",
		Privilege:    "priv",
	}
	gID, err = grant.String()
	r.NoError(err)
	newGrant, err := grantIDFromString(gID)
	r.NoError(err)
	r.Equal("database|name", newGrant.ResourceName)
	r.Equal("schema|name", newGrant.SchemaName)
	r.Equal("view|name", newGrant.ObjectName)
	r.Equal("priv", newGrant.Privilege)
}