mirror of
https://github.com/argoproj/argo-cd
synced 2026-04-21 17:07:16 +00:00
78 lines
3.3 KiB
Go
78 lines
3.3 KiB
Go
package commands
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/argoproj/argo-cd/v2/util/localconfig"
|
|
)
|
|
|
|
const testConfig = `contexts:
|
|
- name: argocd1.example.com:443
|
|
server: argocd1.example.com:443
|
|
user: argocd1.example.com:443
|
|
- name: argocd2.example.com:443
|
|
server: argocd2.example.com:443
|
|
user: argocd2.example.com:443
|
|
- name: localhost:8080
|
|
server: localhost:8080
|
|
user: localhost:8080
|
|
current-context: localhost:8080
|
|
servers:
|
|
- server: argocd1.example.com:443
|
|
- server: argocd2.example.com:443
|
|
- plain-text: true
|
|
server: localhost:8080
|
|
users:
|
|
- auth-token: vErrYS3c3tReFRe$hToken
|
|
name: argocd1.example.com:443
|
|
refresh-token: vErrYS3c3tReFRe$hToken
|
|
- auth-token: vErrYS3c3tReFRe$hToken
|
|
name: argocd2.example.com:443
|
|
refresh-token: vErrYS3c3tReFRe$hToken
|
|
- auth-token: vErrYS3c3tReFRe$hToken
|
|
name: localhost:8080`
|
|
|
|
const testConfigFilePath = "./testdata/local.config"
|
|
|
|
func TestContextDelete(t *testing.T) {
|
|
// Write the test config file
|
|
err := os.WriteFile(testConfigFilePath, []byte(testConfig), os.ModePerm)
|
|
require.NoError(t, err)
|
|
defer os.Remove(testConfigFilePath)
|
|
|
|
err = os.Chmod(testConfigFilePath, 0o600)
|
|
require.NoError(t, err, "Could not change the file permission to 0600 %v", err)
|
|
localConfig, err := localconfig.ReadLocalConfig(testConfigFilePath)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "localhost:8080", localConfig.CurrentContext)
|
|
assert.Contains(t, localConfig.Contexts, localconfig.ContextRef{Name: "localhost:8080", Server: "localhost:8080", User: "localhost:8080"})
|
|
|
|
// Delete a non-current context
|
|
err = deleteContext("argocd1.example.com:443", testConfigFilePath)
|
|
require.NoError(t, err)
|
|
|
|
localConfig, err = localconfig.ReadLocalConfig(testConfigFilePath)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "localhost:8080", localConfig.CurrentContext)
|
|
assert.NotContains(t, localConfig.Contexts, localconfig.ContextRef{Name: "argocd1.example.com:443", Server: "argocd1.example.com:443", User: "argocd1.example.com:443"})
|
|
assert.NotContains(t, localConfig.Servers, localconfig.Server{Server: "argocd1.example.com:443"})
|
|
assert.NotContains(t, localConfig.Users, localconfig.User{AuthToken: "vErrYS3c3tReFRe$hToken", Name: "argocd1.example.com:443"})
|
|
assert.Contains(t, localConfig.Contexts, localconfig.ContextRef{Name: "argocd2.example.com:443", Server: "argocd2.example.com:443", User: "argocd2.example.com:443"})
|
|
assert.Contains(t, localConfig.Contexts, localconfig.ContextRef{Name: "localhost:8080", Server: "localhost:8080", User: "localhost:8080"})
|
|
|
|
// Delete the current context
|
|
err = deleteContext("localhost:8080", testConfigFilePath)
|
|
require.NoError(t, err)
|
|
|
|
localConfig, err = localconfig.ReadLocalConfig(testConfigFilePath)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "", localConfig.CurrentContext)
|
|
assert.NotContains(t, localConfig.Contexts, localconfig.ContextRef{Name: "localhost:8080", Server: "localhost:8080", User: "localhost:8080"})
|
|
assert.NotContains(t, localConfig.Servers, localconfig.Server{PlainText: true, Server: "localhost:8080"})
|
|
assert.NotContains(t, localConfig.Users, localconfig.User{AuthToken: "vErrYS3c3tReFRe$hToken", Name: "localhost:8080"})
|
|
assert.Contains(t, localConfig.Contexts, localconfig.ContextRef{Name: "argocd2.example.com:443", Server: "argocd2.example.com:443", User: "argocd2.example.com:443"})
|
|
}
|