def test_probe_spark_conf_value_default_value(self):
parse_conf_before_probe()
conf_name = "Whatever~"
default_value = "RIGHT_ANSWER"
answer = load_config.probe_spark_conf_value(conf_name, default_value)
expected = default_value
self.assertEqual(answer, expected)
def test_probe_spark_conf_value_default_value(self):
parse_conf_before_probe()
conf_name = "Whatever~"
default_value = "RIGHT_ANSWER"
answer = load_config.probe_spark_conf_value(conf_name, default_value)
expected = default_value
self.assertEqual(answer, expected)
def test(self):
load_config.HibenchConf[
"hibench.spark.home"] = "/tmp/test/spark_home"
conf_name = spark_conf_test_case_list[case_num][1]
line = spark_conf_test_case_list[case_num][2]
default = spark_conf_test_case_list[case_num][3]
def read_file_content(filepath):
if filepath == "/tmp/test/spark_home/conf/spark-env.sh":
return [line]
else:
return []
mock_read_file_content = mock.Mock(side_effect=read_file_content)
with mock.patch("load_config.read_file_content",
mock_read_file_content):
answer = ""
try:
from load_config import probe_spark_conf_value
answer = probe_spark_conf_value(conf_name, default)
except:
pass
expected = default
if len(line.split("=")) >= 2:
expected = line.split("=")[1]
expected = expected.strip("\'")
expected = expected.strip("\"")
self.assertEqual(str(answer), expected)
def test(self):
load_config.HibenchConf[
"hibench.spark.home"] = "/tmp/test/spark_home"
conf_name = spark_conf_test_case_list[case_num][1]
line = spark_conf_test_case_list[case_num][2]
default = spark_conf_test_case_list[case_num][3]
def read_file_content(filepath):
if filepath == "/tmp/test/spark_home/conf/spark-env.sh":
return [line]
else:
return []
mock_read_file_content = mock.Mock(side_effect=read_file_content)
with mock.patch("load_config.read_file_content",
mock_read_file_content):
answer = ""
try:
from load_config import probe_spark_conf_value
answer = probe_spark_conf_value(conf_name, default)
except:
pass
expected = default
if len(line.split("=")) >= 2:
expected = line.split("=")[1]
expected = expected.strip("\'")
expected = expected.strip("\"")
self.assertEqual(str(answer), expected)
