24 lines
620 B
Go
24 lines
620 B
Go
package jupyter_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jfraeys/fetch_ml/internal/jupyter"
|
|
)
|
|
|
|
func TestGetDefaultServiceConfig_EnvOverridesDefaultImage(t *testing.T) {
|
|
old := os.Getenv("FETCHML_JUPYTER_DEFAULT_IMAGE")
|
|
_ = os.Setenv("FETCHML_JUPYTER_DEFAULT_IMAGE", "quay.io/jupyter/base-notebook:latest")
|
|
t.Cleanup(func() {
|
|
_ = os.Setenv("FETCHML_JUPYTER_DEFAULT_IMAGE", old)
|
|
})
|
|
|
|
cfg := jupyter.GetDefaultServiceConfig()
|
|
if cfg == nil {
|
|
t.Fatalf("expected config")
|
|
}
|
|
if cfg.DefaultImage != "quay.io/jupyter/base-notebook:latest" {
|
|
t.Fatalf("expected overridden image, got %q", cfg.DefaultImage)
|
|
}
|
|
}
|