def test_node_external_classify_doesnt_list_unrelated_configuration():
expected_data = {
"classes": {
"testconfiguration": {}
},
"parameters": {},
"environment": "production",
}
expected_data = yaml.dump(expected_data,
default_flow_style=False,
explicit_start=True)
configuration = Configuration(name="testconfiguration")
configuration.save()
configuration2 = Configuration(name="testconfiguration2")
configuration2.save()
classification = Classification()
classification.save()
classification.configurations.add(configuration)
classification2 = Classification()
classification2.save()
classification2.configurations.add(configuration2)
node = Node(name="test node")
node.save()
node.classifications.add(classification)
yaml_data = node.external_classify()
assert yaml_data == expected_data
def handle(self, *args, **options):
try:
if len(options["node_name"]) != 1:
raise CommandError("only one node name is accepted")
except KeyError:
raise CommandError("node_name required")
node_name = options["node_name"][0]
try:
node = Node.objects.get(name=node_name)
except Node.DoesNotExist:
node = Node(name=node_name)
node.save()
node.update()
print(node.external_classify())
def test_node_external_classify_format():
expected_data = yaml.dump(
{
"classes": {},
"parameters": {},
"environment": "production"
},
default_flow_style=False,
explicit_start=True,
)
node = Node(name="test node")
node.save()
yaml_data = node.external_classify()
assert yaml_data == expected_data