rclone/fstest/fstests/gen_tests.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", "")
	generateTestProgram(t, fns, "FTP", "")
	log.Printf("Done")
}