120 lines
2.7 KiB
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))
|
|
}
|
|
}
|
|
}
|