396 lines
12 KiB
Go
396 lines
12 KiB
Go
package enry
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIsVendor(t *testing.T) {
|
|
tests := []struct {
|
|
path string
|
|
expected bool
|
|
}{
|
|
{"cache/", true},
|
|
{"random/cache/", true},
|
|
{"cache", false},
|
|
{"dependencies/", true},
|
|
{"Dependencies/", true},
|
|
{"dependency/", false},
|
|
{"dist/", true},
|
|
{"dist", false},
|
|
{"random/dist/", true},
|
|
{"random/dist", false},
|
|
{"deps/", true},
|
|
{"configure", true},
|
|
{"a/configure", true},
|
|
{"config.guess", true},
|
|
{"config.guess/", false},
|
|
{".vscode/", true},
|
|
{"doc/_build/", true},
|
|
{"a/docs/_build/", true},
|
|
{"a/dasdocs/_build-vsdoc.js", true},
|
|
{"a/dasdocs/_build-vsdoc.j", false},
|
|
{"foo/bar", false},
|
|
{".sublime-project", true},
|
|
{"foo/vendor/foo", true},
|
|
{"leaflet.draw-src.js", true},
|
|
{"foo/bar/MochiKit.js", true},
|
|
{"foo/bar/dojo.js", true},
|
|
{"foo/env/whatever", true},
|
|
{"foo/.imageset/bar", true},
|
|
{"Vagrantfile", true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.path, func(t *testing.T) {
|
|
if got := IsVendor(tt.path); got != tt.expected {
|
|
t.Errorf("IsVendor() = %v, expected %v", got, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func BenchmarkIsVendor(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
IsVendor(".vscode/")
|
|
IsVendor("cache/")
|
|
IsVendor("foo/bar")
|
|
IsVendor("foo/bar/MochiKit.js")
|
|
}
|
|
}
|
|
|
|
func TestIsDocumentation(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
expected bool
|
|
}{
|
|
{name: "TestIsDocumentation_1", path: "foo", expected: false},
|
|
{name: "TestIsDocumentation_2", path: "README", expected: true},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsDocumentation(test.path)
|
|
assert.Equal(t, is, test.expected, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestIsImage(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
expected bool
|
|
}{
|
|
{name: "TestIsImage_1", path: "invalid.txt", expected: false},
|
|
{name: "TestIsImage_2", path: "image.png", expected: true},
|
|
{name: "TestIsImage_3", path: "image.jpg", expected: true},
|
|
{name: "TestIsImage_4", path: "image.jpeg", expected: true},
|
|
{name: "TestIsImage_5", path: "image.gif", expected: true},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsImage(test.path)
|
|
assert.Equal(t, is, test.expected, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestGetMimeType(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
lang string
|
|
expected string
|
|
}{
|
|
{name: "TestGetMimeType_1", path: "text.txt", lang: "", expected: "text/plain"},
|
|
{name: "TestGetMimeType_2", path: "file.go", lang: "Go", expected: "text/x-go"},
|
|
{name: "TestGetMimeType_3", path: "image.png", lang: "", expected: "image/png"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := GetMIMEType(test.path, test.lang)
|
|
assert.Equal(t, is, test.expected, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestIsConfiguration(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
expected bool
|
|
}{
|
|
{name: "TestIsConfiguration_1", path: "foo", expected: false},
|
|
{name: "TestIsConfiguration_2", path: "foo.ini", expected: true},
|
|
{name: "TestIsConfiguration_3", path: "/test/path/foo.json", expected: true},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsConfiguration(test.path)
|
|
assert.Equal(t, is, test.expected, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestIsBinary(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
data []byte
|
|
expected bool
|
|
}{
|
|
{name: "TestIsBinary_1", data: []byte("foo"), expected: false},
|
|
{name: "TestIsBinary_2", data: []byte{0}, expected: true},
|
|
{name: "TestIsBinary_3", data: bytes.Repeat([]byte{'o'}, 8000), expected: false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsBinary(test.data)
|
|
assert.Equal(t, is, test.expected, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestIsDotFile(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
expected bool
|
|
}{
|
|
{name: "TestIsDotFile_1", path: "foo/bar/./", expected: false},
|
|
{name: "TestIsDotFile_2", path: "./", expected: false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsDotFile(test.path)
|
|
assert.Equal(t, test.expected, is, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
func TestIsTestFile(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
expected bool
|
|
}{
|
|
{name: "TestPHP_Is", path: "tests/FooTest.php", expected: true},
|
|
{name: "TestPHP_Not", path: "foo/FooTest.php", expected: false},
|
|
{name: "TestJava_Is_1", path: "test/FooTest.java", expected: true},
|
|
{name: "TestJava_Is_2", path: "test/FooTests.java", expected: true},
|
|
{name: "TestJava_Is_3", path: "test/TestFoo.java", expected: true},
|
|
{name: "TestJava_Is_4", path: "test/qux/TestFoo.java", expected: true},
|
|
{name: "TestJava_Not", path: "foo/FooTest.java", expected: false},
|
|
{name: "TestScala_Is_1", path: "test/FooTest.scala", expected: true},
|
|
{name: "TestScala_Is_2", path: "test/FooTests.scala", expected: true},
|
|
{name: "TestScala_Is_3", path: "test/FooSpec.scala", expected: true},
|
|
{name: "TestScala_Is_4", path: "test/qux/FooSpecs.scala", expected: true},
|
|
{name: "TestScala_Not", path: "foo/FooTest.scala", expected: false},
|
|
{name: "TestPython_Is", path: "test_foo.py", expected: true},
|
|
{name: "TestPython_Not", path: "foo_test.py", expected: false},
|
|
{name: "TestGo_Is", path: "foo_test.go", expected: true},
|
|
{name: "TestGo_Not", path: "test_foo.go", expected: false},
|
|
{name: "TestRuby_Is_1", path: "foo_test.rb", expected: true},
|
|
{name: "TestRuby_Is_1", path: "foo_spec.rb", expected: true},
|
|
{name: "TestRuby_Not", path: "foo_specs.rb", expected: false},
|
|
{name: "TestCSharp_Is_1", path: "FooTest.cs", expected: true},
|
|
{name: "TestCSharp_Is_2", path: "foo/FooTests.cs", expected: true},
|
|
{name: "TestCSharp_Not", path: "foo/TestFoo.cs", expected: false},
|
|
{name: "TestJavaScript_Is_1", path: "foo.test.js", expected: true},
|
|
{name: "TestJavaScript_Is_2", path: "foo.spec.js", expected: true},
|
|
{name: "TestJavaScript_Not", path: "footest.js", expected: false},
|
|
{name: "TestTypeScript_Is_1", path: "foo.test.ts", expected: true},
|
|
{name: "TestTypeScript_Is_2", path: "foo.spec.ts", expected: true},
|
|
{name: "TestTypeScript_Not", path: "footest.ts", expected: false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
is := IsTest(test.path)
|
|
assert.Equal(t, test.expected, is, fmt.Sprintf("%v: is = %v, expected: %v", test.name, is, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestGetColor(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
language string
|
|
expected string
|
|
}{
|
|
{name: "TestGetColor_1", language: "Go", expected: "#00ADD8"},
|
|
{name: "TestGetColor_2", language: "SomeRandom", expected: "#cccccc"},
|
|
{name: "TestGetColor_3", language: "HTML", expected: "#e34c26"},
|
|
{name: "TestGetColor_4", language: "HTML+PHP", expected: "#4f5d95"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
color := GetColor(test.language)
|
|
assert.Equal(t, test.expected, color, fmt.Sprintf("%v: is = %v, expected: %v", test.name, color, test.expected))
|
|
}
|
|
}
|
|
|
|
func TestIsGenerated(t *testing.T) {
|
|
testCases := []struct {
|
|
file string
|
|
load bool
|
|
generated bool
|
|
}{
|
|
// Xcode project files
|
|
{"Binary/MainMenu.nib", false, true},
|
|
{"Dummy/foo.xcworkspacedata", false, true},
|
|
{"Dummy/foo.xcuserstate", false, true},
|
|
|
|
//Cocoapods
|
|
{"Pods/Pods.xcodeproj", false, true},
|
|
{"Pods/SwiftDependency/foo.swift", false, true},
|
|
{"Pods/ObjCDependency/foo.h", false, true},
|
|
{"Pods/ObjCDependency/foo.m", false, true},
|
|
{"Dummy/Pods/Pods.xcodeproj", false, true},
|
|
{"Dummy/Pods/SwiftDependency/foo.swift", false, true},
|
|
{"Dummy/Pods/ObjCDependency/foo.h", false, true},
|
|
{"Dummy/Pods/ObjCDependency/foo.m", false, true},
|
|
|
|
//Carthage
|
|
{"Carthage/Build/.Dependency.version", false, true},
|
|
{"Carthage/Build/iOS/Dependency.framework", false, true},
|
|
{"Carthage/Build/Mac/Dependency.framework", false, true},
|
|
{"src/Carthage/Build/.Dependency.version", false, true},
|
|
{"src/Carthage/Build/iOS/Dependency.framework", false, true},
|
|
{"src/Carthage/Build/Mac/Dependency.framework", false, true},
|
|
|
|
//Go-specific vendored paths
|
|
{"go/vendor/github.com/foo.go", false, true},
|
|
{"go/vendor/golang.org/src/foo.c", false, true},
|
|
{"go/vendor/gopkg.in/some/nested/path/foo.go", false, true},
|
|
|
|
//.NET designer file
|
|
{"Dummy/foo.designer.cs", false, true},
|
|
{"Dummy/foo.Designer.cs", false, true},
|
|
{"Dummy/foo.designer.vb", false, true},
|
|
{"Dummy/foo.Designer.vb", false, true},
|
|
|
|
//Composer generated composer.lock file
|
|
{"JSON/composer.lock", false, true},
|
|
|
|
//Node modules
|
|
{"Dummy/node_modules/foo.js", false, true},
|
|
|
|
//npm shrinkwrap file
|
|
{"Dummy/npm-shrinkwrap.json", false, true},
|
|
{"Dummy/package-lock.json", false, true},
|
|
{"JavaScript/jquery-1.6.1.min.js", true, true},
|
|
|
|
//Yarn Plug'n'Play file
|
|
{".pnp.js", false, true},
|
|
{".pnp.cjs", false, true},
|
|
{".pnp.mjs", false, true},
|
|
|
|
//Godep saved dependencies
|
|
{"Godeps/Godeps.json", false, true},
|
|
{"Godeps/_workspace/src/github.com/kr/s3/sign.go", false, true},
|
|
|
|
//Generated by Zephir
|
|
{"C/exception.zep.c", false, true},
|
|
{"C/exception.zep.h", false, true},
|
|
{"PHP/exception.zep.php", false, true},
|
|
|
|
//Minified files
|
|
{"JavaScript/jquery-1.6.1.min.js", true, true},
|
|
|
|
//JavaScript with source-maps
|
|
{"JavaScript/namespace.js", true, true},
|
|
{"Generated/inline.js", true, true},
|
|
|
|
//CSS with source-maps
|
|
{"Generated/linked.css", true, true},
|
|
{"Generated/inline.css", true, true},
|
|
|
|
//Source-map
|
|
{"Data/bootstrap.css.map", true, true},
|
|
{"Generated/linked.css.map", true, true},
|
|
{"Data/sourcemap.v3.map", true, true},
|
|
{"Data/sourcemap.v1.map", true, true},
|
|
|
|
//Specflow
|
|
{"Features/BindingCulture.feature.cs", false, true},
|
|
|
|
//JFlex
|
|
{"Java/JFlexLexer.java", true, true},
|
|
|
|
//GrammarKit
|
|
{"Java/GrammarKit.java", true, true},
|
|
|
|
//roxygen2
|
|
{"R/import.Rd", true, true},
|
|
|
|
//PostScript
|
|
{"PostScript/lambda.pfa", true, true},
|
|
|
|
//Perl ppport.h
|
|
{"Generated/ppport.h", true, true},
|
|
|
|
//Graphql Relay
|
|
{"Javascript/__generated__/App_user.graphql.js", false, true},
|
|
|
|
//Game Maker Studio 2
|
|
{"JSON/GMS2_Project.yyp", true, true},
|
|
{"JSON/2ea73365-b6f1-4bd1-a454-d57a67e50684.yy", true, true},
|
|
{"Generated/options_main.inherited.yy", true, true},
|
|
|
|
//Pipenv
|
|
{"Dummy/Pipfile.lock", false, true},
|
|
|
|
//HTML
|
|
{"HTML/attr-swapped.html", true, true},
|
|
{"HTML/extra-attr.html", true, true},
|
|
{"HTML/extra-spaces.html", true, true},
|
|
{"HTML/extra-tags.html", true, true},
|
|
{"HTML/grohtml.html", true, true},
|
|
{"HTML/grohtml.xhtml", true, true},
|
|
{"HTML/makeinfo.html", true, true},
|
|
{"HTML/mandoc.html", true, true},
|
|
{"HTML/node78.html", true, true},
|
|
{"HTML/org-mode.html", true, true},
|
|
{"HTML/quotes-double.html", true, true},
|
|
{"HTML/quotes-none.html", true, true},
|
|
{"HTML/quotes-single.html", true, true},
|
|
{"HTML/uppercase.html", true, true},
|
|
{"HTML/ronn.html", true, true},
|
|
{"HTML/unknown.html", true, false},
|
|
{"HTML/no-content.html", true, false},
|
|
{"HTML/pages.html", true, true},
|
|
|
|
//GIMP
|
|
{"C/image.c", true, true},
|
|
{"C/image.h", true, true},
|
|
|
|
//Haxe
|
|
{"Generated/Haxe/main.js", true, true},
|
|
{"Generated/Haxe/main.py", true, true},
|
|
{"Generated/Haxe/main.lua", true, true},
|
|
{"Generated/Haxe/Main.cpp", true, true},
|
|
{"Generated/Haxe/Main.h", true, true},
|
|
{"Generated/Haxe/Main.java", true, true},
|
|
{"Generated/Haxe/Main.cs", true, true},
|
|
{"Generated/Haxe/Main.php", true, true},
|
|
}
|
|
|
|
for _, tt := range testCases {
|
|
t.Run(tt.file, func(t *testing.T) {
|
|
var content []byte
|
|
if tt.load {
|
|
var err error
|
|
content, err = ioutil.ReadFile(filepath.Join("_testdata", tt.file))
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
result := IsGenerated(tt.file, content)
|
|
require.Equal(t, tt.generated, result)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFoo(t *testing.T) {
|
|
file := "HTML/uppercase.html"
|
|
content, err := ioutil.ReadFile("_testdata/" + file)
|
|
require.NoError(t, err)
|
|
require.True(t, IsGenerated(file, content))
|
|
}
|