package auditsetting

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestStorage_getTEPAccessType(t *testing.T) {
	tepTypes := getTEPAccessType("")
	assert.Equal(t, 1, len(tepTypes))
	assert.Equal(t, AccessTypeAll, tepTypes[0])
	tepTypes = getTEPAccessType(dbDML)
	assert.Equal(t, 1, len(tepTypes))
	assert.Equal(t, AccessTypeDML, tepTypes[0])
}

func TestStorage_toDBAccessType(t *testing.T) {
	tepTypes, err := toDBAccessType([]string{AccessTypeAll, AccessTypeDML})
	assert.Nil(t, err)
	assert.Equal(t, dbALL, tepTypes)

	dbTypes, err := toDBAccessType([]string{AccessTypeDQL, AccessTypeDML})
	assert.Nil(t, err)
	tep := getTEPAccessType(dbTypes)
	assert.Equal(t, 2, len(tep))
}