class Yangify(translator.TranslatorData):
def init(self) -> None:
self.root_result = ConfigTree()
self.result = self.root_result
def post(self) -> None:
self.root_result = self.root_result.to_string()
def test_simple(self) -> None:
config = ConfigTree()
gi1 = config.new_section("interface Gi1")
gi1.add_command(' description "A description for Gi1"')
gi1.add_command(" shutdown")
gi1.add_command(" exit")
gi1.add_command("!")
gi2 = config.new_section("interface Gi2")
gi2.add_command(' description "A description for Gi2"')
gi2.add_command(" exit")
gi2.add_command("!")
config.add_command("logging something something")
config.add_command("logging something else")
assert config.to_string() == expected_simple
def test_double_nest(self) -> None:
config = ConfigTree()
gi1 = config.new_section("interface Gi1")
gi1.add_command(' description "A description for Gi1"')
gi1.add_command(" shutdown")
nest = gi1.new_section(" another nest")
nest.add_command(" more subsubcommands")
gi1.add_command(" exit")
gi1.add_command("!")
gi2 = config.new_section("interface Gi2")
gi2.add_command(' description "A description for Gi2"')
gi2.add_command(" exit")
gi2.add_command("!")
config.add_command("logging something something")
config.add_command("logging something else")
assert config.to_string() == expected_double_nested