mirror of https://github.com/databricks/cli.git
add test for file write
This commit is contained in:
parent
bb35ca090f
commit
b7ff019b60
|
@ -1,6 +1,7 @@
|
||||||
package cache
|
package cache
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
@ -140,3 +141,27 @@ func TestDeleteKeyNotExist(t *testing.T) {
|
||||||
_, err = c.Lookup("x")
|
_, err = c.Lookup("x")
|
||||||
assert.Equal(t, ErrNotConfigured, err)
|
assert.Equal(t, ErrNotConfigured, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestWrite(t *testing.T) {
|
||||||
|
tempFile := filepath.Join(t.TempDir(), "token-cache.json")
|
||||||
|
|
||||||
|
tokenMap := map[string]*oauth2.Token{}
|
||||||
|
token := &oauth2.Token{
|
||||||
|
AccessToken: "some-access-token",
|
||||||
|
}
|
||||||
|
tokenMap["test"] = token
|
||||||
|
|
||||||
|
cache := &FileTokenCache{
|
||||||
|
fileLocation: tempFile,
|
||||||
|
Tokens: tokenMap,
|
||||||
|
}
|
||||||
|
|
||||||
|
err := cache.write()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
content, err := os.ReadFile(tempFile)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expected, _ := json.MarshalIndent(&cache, "", " ")
|
||||||
|
assert.Equal(t, content, expected)
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue