def test_load_template_vars_from_string(self): pairs = [ "name=John", "type=Human", "age=12" ] expected = { "name": "John", "type": "Human", "age": "12" } self.assertEqual(expected, load_template_vars(template_vars=pairs))
def test_load_template_vars_from_file(self): with tempfile.NamedTemporaryFile() as tf: tf.write(""" name=John type=Human # a comment # type=Test age=12 """.encode()) tf.flush() expected = {"name": "John", "type": "Human", "age": "12"} self.assertEqual( expected, load_template_vars(template_vars=None, template_vars_file=tf.name))
def test_load_template_vars_from_string_and_file(self): """Text pair variables take precedence over file.""" pairs = ["name=John", "age=12"] with tempfile.NamedTemporaryFile() as tf: tf.write(""" name=Mariah type=Human # a comment # type=Test age=70 """.encode()) tf.flush() expected = {"name": "John", "type": "Human", "age": "12"} self.assertEqual( expected, load_template_vars(template_vars=pairs, template_vars_file=tf.name))
def test_load_template_vars_no_params(self): self.assertFalse(load_template_vars())