youtubebeat/vendor/github.com/elastic/beats/dev-tools/mage/pkg_test.go

120 lines
2.7 KiB
Go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
package mage
import (
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/assert"
"gopkg.in/yaml.v2"
)
func testPackageSpec() PackageSpec {
return PackageSpec{
Name: "brewbeat",
Version: "7.0.0",
Snapshot: true,
OS: "windows",
Arch: "x86_64",
Files: map[string]PackageFile{
"brewbeat.yml": PackageFile{
Source: "./testdata/config.yml",
Mode: 0644,
},
"README.txt": PackageFile{
Content: "Hello! {{.Version}}\n",
Mode: 0644,
},
},
}
}
func TestPackageZip(t *testing.T) {
testPackage(t, PackageZip)
}
func TestPackageTarGz(t *testing.T) {
testPackage(t, PackageTarGz)
}
func TestPackageRPM(t *testing.T) {
if err := HaveDocker(); err != nil {
t.Skip("docker is required")
}
testPackage(t, PackageRPM)
}
func TestPackageDeb(t *testing.T) {
if err := HaveDocker(); err != nil {
t.Skip("docker is required")
}
testPackage(t, PackageDeb)
}
func testPackage(t testing.TB, pack func(PackageSpec) error) {
spec := testPackageSpec().Evaluate()
readme := spec.Files["README.txt"]
readmePath := filepath.ToSlash(filepath.Clean(readme.Source))
assert.True(t, strings.HasPrefix(readmePath, packageStagingDir))
if err := pack(spec); err != nil {
t.Fatal(err)
}
}
func TestRepoRoot(t *testing.T) {
repo, err := GetProjectRepoInfo()
if err != nil {
t.Error(err)
}
assert.Equal(t, "github.com/elastic/beats", repo.RootImportPath)
assert.True(t, filepath.IsAbs(repo.RootDir))
cwd := filepath.Join(repo.RootDir, repo.SubDir)
assert.Equal(t, CWD(), cwd)
}
func TestDumpVariables(t *testing.T) {
out, err := dumpVariables()
if err != nil {
t.Fatal(err)
}
t.Log(out)
}
func TestLoadSpecs(t *testing.T) {
pkgs, err := LoadSpecs("files/packages.yml")
if err != nil {
t.Fatal(err)
}
for flavor, s := range pkgs {
out, err := yaml.Marshal(s)
if err != nil {
t.Fatal(err)
}
if testing.Verbose() {
t.Log("Packaging flavor:", flavor, "\n", string(out))
}
}
}