mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-01-29 20:10:56 -08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix(az): fetch path from AZURE_CONFIG_DIR

Browse source 
resolves #1775
This commit is contained in:
Jan De Dobbeleer 2022-02-18 19:42:17 +01:00 committed by Jan De Dobbeleer
parent 6a06bc5b4c
commit 3efb979053
2 changed files with 25 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/segments/az.go
View file

@ -101,7 +101,7 @@ func (a *Az) FileContentWithoutBom(file string) string {
func (a *Az) getAzureProfile() bool {
var content string
profile := filepath.Join(a.env.Home(), ".azure", "azureProfile.json")
profile := filepath.Join(a.ConfigHome(), "azureProfile.json")
if content = a.FileContentWithoutBom(profile); len(content) == 0 {
return false
}
@ -121,9 +121,9 @@ func (a *Az) getAzureProfile() bool {
func (a *Az) getAzureRmContext() bool {
var content string
cfgHome := a.ConfigHome()
profiles := []string{
filepath.Join(a.env.Home(), ".azure", "AzureRmContext.json"),
filepath.Join(a.env.Home(), ".Azure", "AzureRmContext.json"),
filepath.Join(cfgHome, "AzureRmContext.json"),
}
for _, profile := range profiles {
if content = a.FileContentWithoutBom(profile); len(content) != 0 {
@ -153,3 +153,11 @@ func (a *Az) getAzureRmContext() bool {
a.Origin = "PWSH"
return true
}
func (a *Az) ConfigHome() string {
cfgHome := a.env.Getenv("AZURE_CONFIG_DIR")
if len(cfgHome) != 0 {
return cfgHome
}
return filepath.Join(a.env.Home(), ".azure")
}

33
src/segments/az_test.go
View file

@ -14,13 +14,12 @@ import (
func TestAzSegment(t *testing.T) {
cases := []struct {
Case string
ExpectedEnabled bool
ExpectedString string
HasCLI bool
HasPowerShell bool
HasPowerShellUnix bool
Template string
Case string
ExpectedEnabled bool
ExpectedString string
HasCLI bool
HasPowerShell bool
Template string
}{
{
Case: "no config files found",
@ -41,11 +40,11 @@ func TestAzSegment(t *testing.T) {
HasPowerShell: true,
},
{
Case: "Az Pwsh Profile",
ExpectedEnabled: true,
ExpectedString: "AzurePoshCloud",
Template: "{{ .EnvironmentName }}",
HasPowerShellUnix: true,
Case: "Az Pwsh Profile",
ExpectedEnabled: true,
ExpectedString: "AzurePoshCloud",
Template: "{{ .EnvironmentName }}",
HasPowerShell: true,
},
{
Case: "Faulty template",
@ -74,7 +73,7 @@ func TestAzSegment(t *testing.T) {
env := new(mock.MockedEnvironment)
home := "/Users/posh"
env.On("Home").Return(home)
var azureProfile, azureRmContext, azureRMContext string
var azureProfile, azureRmContext string
if tc.HasCLI {
content, _ := ioutil.ReadFile("../test/azureProfile.json")
azureProfile = string(content)
@ -83,14 +82,10 @@ func TestAzSegment(t *testing.T) {
content, _ := ioutil.ReadFile("../test/AzureRmContext.json")
azureRmContext = string(content)
}
if tc.HasPowerShellUnix {
content, _ := ioutil.ReadFile("../test/AzureRmContext.json")
azureRMContext = string(content)
}
env.On("GOOS").Return(environment.LinuxPlatform)
env.On("FileContent", filepath.Join(home, ".azure", "azureProfile.json")).Return(azureProfile)
env.On("FileContent", filepath.Join(home, ".Azure", "AzureRmContext.json")).Return(azureRmContext)
env.On("FileContent", filepath.Join(home, ".azure", "AzureRmContext.json")).Return(azureRMContext)
env.On("FileContent", filepath.Join(home, ".azure", "AzureRmContext.json")).Return(azureRmContext)
env.On("Getenv", "AZURE_CONFIG_DIR").Return("")
az := &Az{
env: env,
props: properties.Map{},