68 lines
2.7 KiB
Go
68 lines
2.7 KiB
Go
package dbidentifier
|
|
|
|
import (
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
)
|
|
|
|
func TestToGoIdentifier(t *testing.T) {
|
|
require.Equal(t, ToGoIdentifier(""), "")
|
|
require.Equal(t, ToGoIdentifier("uuid"), "UUID")
|
|
require.Equal(t, ToGoIdentifier("uuid_ptr"), "UUIDPtr")
|
|
require.Equal(t, ToGoIdentifier("col1"), "Col1")
|
|
require.Equal(t, ToGoIdentifier("PG-13"), "Pg13")
|
|
require.Equal(t, ToGoIdentifier("13_pg"), "13Pg")
|
|
|
|
require.Equal(t, ToGoIdentifier("mytable"), "Mytable")
|
|
require.Equal(t, ToGoIdentifier("MYTABLE"), "Mytable")
|
|
require.Equal(t, ToGoIdentifier("MyTaBlE"), "MyTaBlE")
|
|
require.Equal(t, ToGoIdentifier("myTaBlE"), "MyTaBlE")
|
|
|
|
require.Equal(t, ToGoIdentifier("my_table"), "MyTable")
|
|
require.Equal(t, ToGoIdentifier("my_____table"), "MyTable")
|
|
require.Equal(t, ToGoIdentifier("MY_TABLE"), "MyTable")
|
|
require.Equal(t, ToGoIdentifier("My_Table"), "MyTable")
|
|
require.Equal(t, ToGoIdentifier("My Table"), "MyTable")
|
|
require.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
|
|
|
|
require.Equal(t, ToGoIdentifier("EN\bUM"), "Enum") // control character
|
|
require.Equal(t, ToGoIdentifier("EN\tUM"), "EnUm") // space character
|
|
require.Equal(t, ToGoIdentifier("S3:INIT"), "S3ColonInit") // replacement chars
|
|
require.Equal(t, ToGoIdentifier("Entity-"), "Entity")
|
|
require.Equal(t, ToGoIdentifier("Entity+"), "EntityPlus")
|
|
require.Equal(t, ToGoIdentifier("="), "Equal")
|
|
require.Equal(t, ToGoIdentifier("<="), "LessEqual")
|
|
require.Equal(t, ToGoIdentifier(">="), "GreaterEqual")
|
|
require.Equal(t, ToGoIdentifier("some#$%name"), "SomeNumberDollarPercentName")
|
|
require.Equal(t, ToGoIdentifier(`An!"them`), "AnExclamationQuotationThem")
|
|
require.Equal(t, ToGoIdentifier(`An(Um)`),
|
|
"AnOpeningParenthesesUmClosingParentheses")
|
|
}
|
|
|
|
func TestNeedsCharReplacement(t *testing.T) {
|
|
increase, needs := needsCharReplacement("some_name")
|
|
require.False(t, needs)
|
|
require.Zero(t, increase)
|
|
|
|
increase, needs = needsCharReplacement("some name")
|
|
require.True(t, needs)
|
|
require.Zero(t, increase)
|
|
|
|
increase, needs = needsCharReplacement("some\bname")
|
|
require.True(t, needs)
|
|
require.Equal(t, increase, -1)
|
|
|
|
increase, needs = needsCharReplacement("some#$%name")
|
|
require.True(t, needs)
|
|
require.Equal(t, increase, 22)
|
|
}
|
|
|
|
func TestToGoFileName(t *testing.T) {
|
|
require.Equal(t, ToGoFileName("FileName"), "filename")
|
|
require.Equal(t, ToGoFileName("File_Name"), "file_name")
|
|
require.Equal(t, ToGoFileName("File___Name__"), "file___name__")
|
|
require.Equal(t, ToGoFileName("File___Name__"), "file___name__")
|
|
require.Equal(t, ToGoFileName("File\bName"), "filename")
|
|
require.Equal(t, ToGoFileName("File\tName"), "file_name")
|
|
require.Equal(t, ToGoFileName("File^^Name"), "file_caret__caret_name")
|
|
}
|