alexvanin/rclone
1
0
Fork 0
forked from TrueCloudLab/rclone
Code Pull requests Activity

gen_tests: allow specification of a build tag

Browse source
This commit is contained in:
Nick Craig-Wood 2017-08-01 13:38:36 +01:00
parent e01741b557
commit d4b9bb9894
1 changed files with 61 additions and 43 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

104
fstest/fstests/gen_tests.go
View file

@ -41,19 +41,23 @@ func findTestFunctions() []string {
// Data to substitute // Data to substitute
type Data struct { type Data struct {
Regenerate string Regenerate string
FsName string FsName string
UpperFsName string UpperFsName string
TestName string TestName string
Fns []string Fns []string
Suffix string Suffix string
BuildConstraint string
} }
var testProgram = ` var testProgram = `
// Test {{ .UpperFsName }} filesystem interface // Test {{ .UpperFsName }} filesystem interface
// //
// Automatically generated - DO NOT EDIT // Automatically generated - DO NOT EDIT
// Regenerate with: {{ .Regenerate }} // Regenerate with: {{ .Regenerate }}{{ if ne .BuildConstraint "" }}
// +build {{ .BuildConstraint }}
{{end}}
package {{ .FsName }}_test package {{ .FsName }}_test
import ( import (
@ -75,23 +79,37 @@ func TestSetup{{ .Suffix }}(t *testing.T)() {
{{ end }} {{ end }}
` `
// options for generateTestProgram
type (
suffix string
buildConstraint string
)
// Generate test file piping it through gofmt // Generate test file piping it through gofmt
func generateTestProgram(t *template.Template, fns []string, Fsname string, suffix string) { func generateTestProgram(t *template.Template, fns []string, Fsname string, options ...interface{}) {
fsname := strings.ToLower(Fsname)
TestName := "Test" + Fsname + suffix + ":"
outfile := "../../" + fsname + "/" + fsname + suffix + "_test.go"
if fsname == "local" {
TestName = ""
}
data := Data{ data := Data{
Regenerate: "make gen_tests", Regenerate: "make gen_tests",
FsName: fsname, FsName: strings.ToLower(Fsname),
UpperFsName: Fsname, UpperFsName: Fsname,
TestName: TestName,
Fns: fns, Fns: fns,
Suffix: suffix, }
for _, option := range options {
switch x := option.(type) {
case suffix:
data.Suffix = string(x)
case buildConstraint:
data.BuildConstraint = string(x)
default:
log.Fatalf("Unknown option type %T", option)
}
}
data.TestName = "Test" + data.UpperFsName + data.Suffix + ":"
outfile := "../../" + data.FsName + "/" + data.FsName + data.Suffix + "_test.go"
if data.FsName == "local" {
data.TestName = ""
} }
cmd := exec.Command("gofmt") cmd := exec.Command("gofmt")
@ -99,50 +117,50 @@ func generateTestProgram(t *template.Template, fns []string, Fsname string, suff
log.Printf("Writing %q", outfile) log.Printf("Writing %q", outfile)
out, err := os.Create(outfile) out, err := os.Create(outfile)
if err != nil { if err != nil {
log.Fatal(err) log.Fatalf("Failed to write %q: %v", outfile, err)
} }
cmd.Stdout = out cmd.Stdout = out
gofmt, err := cmd.StdinPipe() gofmt, err := cmd.StdinPipe()
if err != nil { if err != nil {
log.Fatal(err) log.Fatalf("Failed to StdinPipe %q: %v", outfile, err)
} }
if err = cmd.Start(); err != nil { if err = cmd.Start(); err != nil {
log.Fatal(err) log.Fatalf("Failed to Start %q: %v", outfile, err)
} }
if err = t.Execute(gofmt, data); err != nil { if err = t.Execute(gofmt, data); err != nil {
log.Fatal(err) log.Fatalf("Failed to Execute %q: %v", outfile, err)
} }
if err = gofmt.Close(); err != nil { if err = gofmt.Close(); err != nil {
log.Fatal(err) log.Fatalf("Failed to Close gofmt on %q: %v", outfile, err)
} }
if err = cmd.Wait(); err != nil { if err = cmd.Wait(); err != nil {
log.Fatal(err) log.Fatalf("Failed to Wait %q: %v", outfile, err)
} }
if err = out.Close(); err != nil { if err = out.Close(); err != nil {
log.Fatal(err) log.Fatalf("Failed to Close out %q: %v", outfile, err)
} }
} }
func main() { func main() {
fns := findTestFunctions() fns := findTestFunctions()
t := template.Must(template.New("main").Parse(testProgram)) t := template.Must(template.New("main").Parse(testProgram))
generateTestProgram(t, fns, "Local", "") generateTestProgram(t, fns, "Local")
generateTestProgram(t, fns, "Swift", "") generateTestProgram(t, fns, "Swift")
generateTestProgram(t, fns, "S3", "") generateTestProgram(t, fns, "S3")
generateTestProgram(t, fns, "Drive", "") generateTestProgram(t, fns, "Drive")
generateTestProgram(t, fns, "GoogleCloudStorage", "") generateTestProgram(t, fns, "GoogleCloudStorage")
generateTestProgram(t, fns, "Dropbox", "") generateTestProgram(t, fns, "Dropbox")
generateTestProgram(t, fns, "AmazonCloudDrive", "") generateTestProgram(t, fns, "AmazonCloudDrive")
generateTestProgram(t, fns, "OneDrive", "") generateTestProgram(t, fns, "OneDrive")
generateTestProgram(t, fns, "Hubic", "") generateTestProgram(t, fns, "Hubic")
generateTestProgram(t, fns, "B2", "") generateTestProgram(t, fns, "B2")
generateTestProgram(t, fns, "Yandex", "") generateTestProgram(t, fns, "Yandex")
generateTestProgram(t, fns, "Crypt", "") generateTestProgram(t, fns, "Crypt")
generateTestProgram(t, fns, "Crypt", "2") generateTestProgram(t, fns, "Crypt", suffix("2"))
generateTestProgram(t, fns, "Crypt", "3") generateTestProgram(t, fns, "Crypt", suffix("3"))
generateTestProgram(t, fns, "Sftp", "") generateTestProgram(t, fns, "Sftp")
generateTestProgram(t, fns, "FTP", "") generateTestProgram(t, fns, "FTP")
generateTestProgram(t, fns, "Box", "") generateTestProgram(t, fns, "Box")
log.Printf("Done") log.Printf("Done")
} }