registry_add_test.go 2.33 KB
Newer Older
bryanl's avatar
bryanl committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
// Copyright 2018 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 actions

import (
	"testing"

21
	"github.com/ksonnet/ksonnet/metadata/app"
bryanl's avatar
bryanl committed
22 23
	amocks "github.com/ksonnet/ksonnet/metadata/app/mocks"
	"github.com/ksonnet/ksonnet/pkg/registry"
24
	"github.com/stretchr/testify/assert"
bryanl's avatar
bryanl committed
25 26 27 28 29 30 31 32 33 34 35 36 37
	"github.com/stretchr/testify/require"
)

func TestRegistryAdd(t *testing.T) {
	withApp(t, func(appMock *amocks.App) {
		name := "new"

		cases := []struct {
			name        string
			uri         string
			version     string
			expectedURI string
			protocol    string
38
			isOverride  bool
bryanl's avatar
bryanl committed
39 40 41 42 43 44 45
		}{
			{
				name:        "github",
				uri:         "github.com/foo/bar",
				expectedURI: "github.com/foo/bar",
				protocol:    registry.ProtocolGitHub,
			},
46 47 48 49 50 51 52
			{
				name:        "github override",
				uri:         "github.com/foo/bar",
				expectedURI: "github.com/foo/bar",
				protocol:    registry.ProtocolGitHub,
				isOverride:  true,
			},
bryanl's avatar
bryanl committed
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
			{
				name:        "fs",
				uri:         "/path",
				expectedURI: "file:///path",
				protocol:    registry.ProtocolFilesystem,
			},
			{
				name:        "fs with URL",
				uri:         "file:///path",
				expectedURI: "file:///path",
				protocol:    registry.ProtocolFilesystem,
			},
		}

		for _, tc := range cases {
			t.Run(tc.name, func(t *testing.T) {
69
				a, err := NewRegistryAdd(appMock, name, tc.uri, tc.version, tc.isOverride)
bryanl's avatar
bryanl committed
70 71
				require.NoError(t, err)

72 73 74 75 76 77 78 79 80
				a.registryAdd = func(a app.App, name, protocol, uri, version string, isOverride bool) (*registry.Spec, error) {
					assert.Equal(t, "new", name)
					assert.Equal(t, tc.protocol, protocol)
					assert.Equal(t, tc.expectedURI, uri)
					assert.Equal(t, tc.version, version)
					assert.Equal(t, tc.isOverride, isOverride)

					return &registry.Spec{}, nil
				}
bryanl's avatar
bryanl committed
81 82 83 84 85 86 87 88

				err = a.Run()
				require.NoError(t, err)
			})
		}

	})
}