995538829e
Management support is only available to the ephemeral host. This change extends support of management features to hosts not labeled as the ephemeral host. Change-Id: I0ac917f2633f659e56e67cafc52d2db9d967754f Signed-off-by: Drew Walters <andrew.walters@att.com>
138 lines
4.4 KiB
Go
138 lines
4.4 KiB
Go
/*
|
|
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
|
|
|
|
https://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 remote
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"opendev.org/airship/airshipctl/pkg/config"
|
|
"opendev.org/airship/airshipctl/pkg/document"
|
|
"opendev.org/airship/airshipctl/pkg/environment"
|
|
"opendev.org/airship/airshipctl/testutil"
|
|
)
|
|
|
|
type Configuration func(*environment.AirshipCTLSettings)
|
|
|
|
// initSettings initializes the global airshipctl settings with test data by accepting functions as arguments that
|
|
// manipulate configuration sections.
|
|
func initSettings(t *testing.T, configs ...Configuration) *environment.AirshipCTLSettings {
|
|
t.Helper()
|
|
|
|
settings := &environment.AirshipCTLSettings{Config: testutil.DummyConfig()}
|
|
|
|
for _, cfg := range configs {
|
|
cfg(settings)
|
|
}
|
|
|
|
return settings
|
|
}
|
|
|
|
// withManagementConfig initializes the management config when used as an argument to initSettings.
|
|
func withManagementConfig(cfg *config.ManagementConfiguration) Configuration {
|
|
return func(settings *environment.AirshipCTLSettings) {
|
|
settings.Config.ManagementConfiguration["dummy_management_config"] = cfg
|
|
}
|
|
}
|
|
|
|
// withTestDataPath sets the test data path when used as an argument to initSettings.
|
|
func withTestDataPath(path string) Configuration {
|
|
return func(settings *environment.AirshipCTLSettings) {
|
|
manifest, err := settings.Config.CurrentContextManifest()
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Unable to initialize management tests. Current Context error %q", err))
|
|
}
|
|
|
|
manifest.TargetPath = fmt.Sprintf("testdata/%s", path)
|
|
}
|
|
}
|
|
|
|
func TestNewManagerEphemeralHost(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
manager, err := NewManager(settings, config.BootstrapPhase, ByLabel(document.EphemeralHostSelector))
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(manager.Hosts))
|
|
|
|
assert.Equal(t, "ephemeral", manager.Hosts[0].NodeID())
|
|
}
|
|
|
|
func TestNewManagerByName(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
manager, err := NewManager(settings, config.BootstrapPhase, ByName("master-1"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(manager.Hosts))
|
|
|
|
assert.Equal(t, "node-master-1", manager.Hosts[0].NodeID())
|
|
}
|
|
|
|
func TestNewManagerMultipleNodes(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
manager, err := NewManager(settings, config.BootstrapPhase, ByLabel("airshipit.org/test-node=true"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, len(manager.Hosts))
|
|
|
|
assert.Equal(t, "node-master-1", manager.Hosts[0].NodeID())
|
|
assert.Equal(t, "node-master-2", manager.Hosts[1].NodeID())
|
|
}
|
|
|
|
func TestNewManagerByNameNoHostFound(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
_, err := NewManager(settings, config.BootstrapPhase, ByName("bad-name"))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewManagerByLabelNoHostsFound(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
_, err := NewManager(settings, config.BootstrapPhase, ByLabel("bad-label=true"))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewManagerUnknownRemoteType(t *testing.T) {
|
|
badCfg := &config.ManagementConfiguration{Type: "bad-remote-type"}
|
|
settings := initSettings(t, withManagementConfig(badCfg), withTestDataPath("base"))
|
|
|
|
_, err := NewManager(settings, config.BootstrapPhase, ByLabel(document.EphemeralHostSelector))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewManagerMissingBMCAddress(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("emptyurl"))
|
|
|
|
_, err := NewManager(settings, config.BootstrapPhase, ByLabel(document.EphemeralHostSelector))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewManagerMissingCredentials(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("emptyurl"))
|
|
|
|
_, err := NewManager(settings, config.BootstrapPhase, ByName("no-creds"))
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestNewManagerBadPhase(t *testing.T) {
|
|
settings := initSettings(t, withTestDataPath("base"))
|
|
|
|
_, err := NewManager(settings, "bad-phase", ByLabel(document.EphemeralHostSelector))
|
|
assert.Error(t, err)
|
|
}
|