component_test.go

// Copyright 2017 The ksonnet authors
//
//
//    Licensed 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 metadata

import (
	"fmt"
	"os"
	"sort"
	"strings"
	"testing"
)

const (
	componentsPath  = "/componentsPath"
	componentSubdir = "subdir"
	componentFile1  = "component1.jsonnet"
	componentFile2  = "component2.jsonnet"
)

func populateComponentPaths(t *testing.T) *manager {
	spec, err := parseClusterSpec(fmt.Sprintf("file:%s", blankSwagger), testFS)
	if err != nil {
		t.Fatalf("Failed to parse cluster spec: %v", err)
	}

	appPath := AbsPath(componentsPath)
	reg := newMockRegistryManager("incubator")
	m, err := initManager("componentPaths", appPath, spec, &mockAPIServer, &mockNamespace, reg, testFS)
	if err != nil {
		t.Fatalf("Failed to init cluster spec: %v", err)
	}

	// Create empty app file.
	components := appendToAbsPath(appPath, componentsDir)
	appFile1 := appendToAbsPath(components, componentFile1)
	f1, err := testFS.OpenFile(string(appFile1), os.O_RDONLY|os.O_CREATE, 0777)
	if err != nil {
		t.Fatalf("Failed to touch app file '%s'\n%v", appFile1, err)
	}
	f1.Close()

	// Create empty file in a nested directory.
	appSubdir := appendToAbsPath(components, componentSubdir)
	err = testFS.MkdirAll(string(appSubdir), os.ModePerm)
	if err != nil {
		t.Fatalf("Failed to create directory '%s'\n%v", appSubdir, err)
	}
	appFile2 := appendToAbsPath(appSubdir, componentFile2)
	f2, err := testFS.OpenFile(string(appFile2), os.O_RDONLY|os.O_CREATE, 0777)
	if err != nil {
		t.Fatalf("Failed to touch app file '%s'\n%v", appFile1, err)
	}
	f2.Close()

	// Create a directory that won't be listed in the call to `ComponentPaths`.
	unlistedDir := string(appendToAbsPath(components, "doNotListMe"))
	err = testFS.MkdirAll(unlistedDir, os.ModePerm)
	if err != nil {
		t.Fatalf("Failed to create directory '%s'\n%v", unlistedDir, err)
	}

	return m
}

func cleanComponentPaths(t *testing.T) {
	testFS.RemoveAll(componentsPath)
}

func TestComponentPaths(t *testing.T) {
	m := populateComponentPaths(t)
	defer cleanComponentPaths(t)

	paths, err := m.ComponentPaths()
	if err != nil {
		t.Fatalf("Failed to find component paths: %v", err)
	}

	sort.Slice(paths, func(i, j int) bool { return paths[i] < paths[j] })

	expectedPath1 := fmt.Sprintf("%s/components/%s", componentsPath, componentFile1)
	expectedPath2 := fmt.Sprintf("%s/components/%s/%s", componentsPath, componentSubdir, componentFile2)

	if len(paths) != 2 || paths[0] != expectedPath1 || paths[1] != expectedPath2 {
		t.Fatalf("m.ComponentPaths failed; expected '%s', got '%s'", []string{expectedPath1, expectedPath2}, paths)
	}
}

func TestGetAllComponents(t *testing.T) {
	m := populateComponentPaths(t)
	defer cleanComponentPaths(t)

	components, err := m.GetAllComponents()
	if err != nil {
		t.Fatalf("Failed to get all components, %v", err)
	}

	expected1 := strings.TrimSuffix(componentFile1, ".jsonnet")
	expected2 := strings.TrimSuffix(componentFile2, ".jsonnet")

	if len(components) != 2 {
		t.Fatalf("Expected exactly 2 components, got %d", len(components))
	}

	if components[0] != expected1 {
		t.Fatalf("Expected component %s, got %s", expected1, components)
	}

	if components[1] != expected2 {
		t.Fatalf("Expected component %s, got %s", expected2, components)
	}
}