You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package migrationnotes
|
|
|
|
import (
|
|
"io/fs"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_needShow1_3_4(t *testing.T) {
|
|
root, err := os.MkdirTemp("", "goctl-model")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(root)
|
|
|
|
dir1 := path.Join(root, "dir1")
|
|
require.NoError(t, os.Mkdir(dir1, fs.ModePerm))
|
|
ioutil.WriteFile(filepath.Join(dir1, "foo_gen.go"), nil, fs.ModePerm)
|
|
|
|
dir2 := path.Join(root, "dir2")
|
|
require.NoError(t, os.Mkdir(dir2, fs.ModePerm))
|
|
ioutil.WriteFile(filepath.Join(dir2, "foomodel.go"), nil, fs.ModePerm)
|
|
|
|
dir3 := path.Join(root, "dir3")
|
|
require.NoError(t, os.Mkdir(dir3, fs.ModePerm))
|
|
ioutil.WriteFile(filepath.Join(dir3, "irrelevant.go"), nil, fs.ModePerm)
|
|
|
|
type args struct {
|
|
dir string
|
|
style string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want bool
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "dir that contains *_gen.go should return false",
|
|
args: args{
|
|
dir: dir1,
|
|
},
|
|
want: false,
|
|
},
|
|
{
|
|
name: "dir that contains *model.go without *_gen.go should return true",
|
|
args: args{
|
|
dir: dir2,
|
|
},
|
|
want: true,
|
|
},
|
|
{
|
|
name: "dir that only contains irrelevant files should return false",
|
|
args: args{
|
|
dir: dir3,
|
|
},
|
|
want: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := needShow1_3_4(tt.args.dir, tt.args.style)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("needShow1_3_4() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("needShow1_3_4() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|