conf_test.go

package hdfs

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestConfFallback(t *testing.T) {
	os.Setenv("HADOOP_HOME", "test") // This will resolve to test/conf.
	os.Setenv("HADOOP_CONF_DIR", "test/conf2")

	confNamenodes := []string{"namenode1:8020", "namenode2:8020"}
	conf2Namenodes := []string{"namenode3:8020"}
	conf3Namenodes := []string{"namenode4:8020"}

	conf := LoadHadoopConf("test/conf3")
	nns, err := conf.Namenodes()
	assert.Nil(t, err)
	assert.EqualValues(t, conf3Namenodes, nns, "loading via specified path (test/conf3)")

	conf = LoadHadoopConf("")
	nns, err = conf.Namenodes()
	assert.Nil(t, err)
	assert.EqualValues(t, conf2Namenodes, nns, "loading via HADOOP_CONF_DIR (test/conf2)")

	os.Unsetenv("HADOOP_CONF_DIR")

	conf = LoadHadoopConf("")
	nns, err = conf.Namenodes()
	assert.Nil(t, err)
	assert.EqualValues(t, confNamenodes, nns, "loading via HADOOP_HOME (test/conf)")

	os.Unsetenv("HADOOP_HOME")
}