зеркало из https://github.com/github/vitess-gh.git
50 строки
1014 B
Go
50 строки
1014 B
Go
// Copyright 2013, Google Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package testfiles locates test files within the Vitess directory tree.
|
|
package testfiles
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
)
|
|
|
|
func Locate(filename string) string {
|
|
vtroot := os.Getenv("VTROOT")
|
|
if vtroot == "" {
|
|
panic(fmt.Errorf("VTROOT is not set"))
|
|
}
|
|
return path.Join(vtroot, "data", "test", filename)
|
|
}
|
|
|
|
func Glob(pattern string) []string {
|
|
vtroot := os.Getenv("VTROOT")
|
|
if vtroot == "" {
|
|
panic(fmt.Errorf("VTROOT is not set"))
|
|
}
|
|
dir := path.Join(vtroot, "data", "test")
|
|
if exists, err := exists(dir); !exists {
|
|
panic(err)
|
|
}
|
|
resolved := path.Join(dir, pattern)
|
|
out, err := filepath.Glob(resolved)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func exists(path string) (bool, error) {
|
|
_, err := os.Stat(path)
|
|
if err == nil {
|
|
return true, nil
|
|
}
|
|
if os.IsNotExist(err) {
|
|
return false, err
|
|
}
|
|
return false, err
|
|
}
|