mirror of https://github.com/databricks/cli.git
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
var homeEnvVar = "HOME"
|
|
|
|
func init() {
|
|
if runtime.GOOS == "windows" {
|
|
homeEnvVar = "USERPROFILE"
|
|
}
|
|
}
|
|
|
|
func setup(t *testing.T) string {
|
|
tempHomeDir := t.TempDir()
|
|
t.Setenv(homeEnvVar, tempHomeDir)
|
|
return tempHomeDir
|
|
}
|
|
|
|
func TestStoreAndLookup(t *testing.T) {
|
|
setup(t)
|
|
c := &TokenCache{}
|
|
err := c.Store("x", &oauth2.Token{
|
|
AccessToken: "abc",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
err = c.Store("y", &oauth2.Token{
|
|
AccessToken: "bcd",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
l := &TokenCache{}
|
|
tok, err := l.Lookup("x")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "abc", tok.AccessToken)
|
|
assert.Equal(t, 2, len(l.Tokens))
|
|
|
|
_, err = l.Lookup("z")
|
|
assert.Equal(t, ErrNotConfigured, err)
|
|
}
|
|
|
|
func TestNoCacheFileReturnsErrNotConfigured(t *testing.T) {
|
|
setup(t)
|
|
l := &TokenCache{}
|
|
_, err := l.Lookup("x")
|
|
assert.Equal(t, ErrNotConfigured, err)
|
|
}
|
|
|
|
func TestLoadCorruptFile(t *testing.T) {
|
|
home := setup(t)
|
|
f := filepath.Join(home, tokenCacheFile)
|
|
err := os.MkdirAll(filepath.Dir(f), ownerExecReadWrite)
|
|
require.NoError(t, err)
|
|
err = os.WriteFile(f, []byte("abc"), ownerExecReadWrite)
|
|
require.NoError(t, err)
|
|
|
|
l := &TokenCache{}
|
|
_, err = l.Lookup("x")
|
|
assert.EqualError(t, err, "load: parse: invalid character 'a' looking for beginning of value")
|
|
}
|
|
|
|
func TestLoadWrongVersion(t *testing.T) {
|
|
home := setup(t)
|
|
f := filepath.Join(home, tokenCacheFile)
|
|
err := os.MkdirAll(filepath.Dir(f), ownerExecReadWrite)
|
|
require.NoError(t, err)
|
|
err = os.WriteFile(f, []byte(`{"version": 823, "things": []}`), ownerExecReadWrite)
|
|
require.NoError(t, err)
|
|
|
|
l := &TokenCache{}
|
|
_, err = l.Lookup("x")
|
|
assert.EqualError(t, err, "load: needs version 1, got version 823")
|
|
}
|
|
|
|
func TestDevNull(t *testing.T) {
|
|
t.Setenv(homeEnvVar, "/dev/null")
|
|
l := &TokenCache{}
|
|
_, err := l.Lookup("x")
|
|
// macOS/Linux: load: read: open /dev/null/.databricks/token-cache.json:
|
|
// windows: databricks OAuth is not configured for this host
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestStoreOnDev(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.SkipNow()
|
|
}
|
|
t.Setenv(homeEnvVar, "/dev")
|
|
c := &TokenCache{}
|
|
err := c.Store("x", &oauth2.Token{
|
|
AccessToken: "abc",
|
|
})
|
|
// Linux: permission denied
|
|
// macOS: read-only file system
|
|
assert.Error(t, err)
|
|
}
|