146 lines
3.3 KiB
Go
146 lines
3.3 KiB
Go
// +build ignore
|
|
|
|
// Make the test files from fstests.go
|
|
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"html/template"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// Search fstests.go and return all the test function names
|
|
func findTestFunctions() []string {
|
|
fns := []string{}
|
|
matcher := regexp.MustCompile(`^func\s+(Test.*?)\(`)
|
|
|
|
in, err := os.Open("fstests.go")
|
|
if err != nil {
|
|
log.Fatalf("Couldn't open fstests.go: %v", err)
|
|
}
|
|
defer in.Close()
|
|
|
|
scanner := bufio.NewScanner(in)
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
matches := matcher.FindStringSubmatch(line)
|
|
if len(matches) > 0 {
|
|
fns = append(fns, matches[1])
|
|
}
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
log.Fatalf("Error scanning file: %v", err)
|
|
}
|
|
return fns
|
|
}
|
|
|
|
// Data to substitute
|
|
type Data struct {
|
|
Regenerate string
|
|
FsName string
|
|
UpperFsName string
|
|
TestName string
|
|
Fns []string
|
|
Suffix string
|
|
}
|
|
|
|
var testProgram = `
|
|
// Test {{ .UpperFsName }} filesystem interface
|
|
//
|
|
// Automatically generated - DO NOT EDIT
|
|
// Regenerate with: {{ .Regenerate }}
|
|
package {{ .FsName }}_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ncw/rclone/fs"
|
|
"github.com/ncw/rclone/fstest/fstests"
|
|
"github.com/ncw/rclone/{{ .FsName }}"
|
|
{{ if eq .FsName "crypt" }} _ "github.com/ncw/rclone/local"
|
|
{{end}})
|
|
|
|
func TestSetup{{ .Suffix }}(t *testing.T)() {
|
|
fstests.NilObject = fs.Object((*{{ .FsName }}.Object)(nil))
|
|
fstests.RemoteName = "{{ .TestName }}"
|
|
}
|
|
|
|
// Generic tests for the Fs
|
|
{{ range $fn := .Fns }}func {{ $fn }}{{ $.Suffix }}(t *testing.T){ fstests.{{ $fn }}(t) }
|
|
{{ end }}
|
|
`
|
|
|
|
// Generate test file piping it through gofmt
|
|
func generateTestProgram(t *template.Template, fns []string, Fsname string, suffix string) {
|
|
fsname := strings.ToLower(Fsname)
|
|
TestName := "Test" + Fsname + suffix + ":"
|
|
outfile := "../../" + fsname + "/" + fsname + suffix + "_test.go"
|
|
|
|
if fsname == "local" {
|
|
TestName = ""
|
|
}
|
|
|
|
data := Data{
|
|
Regenerate: "make gen_tests",
|
|
FsName: fsname,
|
|
UpperFsName: Fsname,
|
|
TestName: TestName,
|
|
Fns: fns,
|
|
Suffix: suffix,
|
|
}
|
|
|
|
cmd := exec.Command("gofmt")
|
|
|
|
log.Printf("Writing %q", outfile)
|
|
out, err := os.Create(outfile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
cmd.Stdout = out
|
|
|
|
gofmt, err := cmd.StdinPipe()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = cmd.Start(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = t.Execute(gofmt, data); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = gofmt.Close(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = cmd.Wait(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = out.Close(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
fns := findTestFunctions()
|
|
t := template.Must(template.New("main").Parse(testProgram))
|
|
generateTestProgram(t, fns, "Local", "")
|
|
generateTestProgram(t, fns, "Swift", "")
|
|
generateTestProgram(t, fns, "S3", "")
|
|
generateTestProgram(t, fns, "Drive", "")
|
|
generateTestProgram(t, fns, "GoogleCloudStorage", "")
|
|
generateTestProgram(t, fns, "Dropbox", "")
|
|
generateTestProgram(t, fns, "AmazonCloudDrive", "")
|
|
generateTestProgram(t, fns, "OneDrive", "")
|
|
generateTestProgram(t, fns, "Hubic", "")
|
|
generateTestProgram(t, fns, "B2", "")
|
|
generateTestProgram(t, fns, "Yandex", "")
|
|
generateTestProgram(t, fns, "Crypt", "")
|
|
generateTestProgram(t, fns, "Crypt", "2")
|
|
generateTestProgram(t, fns, "Crypt", "3")
|
|
generateTestProgram(t, fns, "Sftp", "")
|
|
log.Printf("Done")
|
|
}
|