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 test_nodeclassify_command_prints_yaml_for_valid_name(capsys):
command = ncommand.Command()
node = Node(name="TestNode")
node.save()
command.handle(None, node_name=["TestNode"])
captured = capsys.readouterr()
assert captured.out == VALID_YAML
def test_node_view_update_node(client):
classification = Classification(name="ClassTest1")
classification.save()
node = Node(name="TestNode")
node.save()
response = client.post(
reverse("node_update", kwargs={"pk": node.id}),
{"classifications": [classification.id]},
follow=True,
)
assert response.status_code == 200
assert "ClassTest1" in str(response.content)
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
def test_node_list_view_shows_nodes(client):
node_test = Node(name="NodeTest1")
node_test.save()
node_test = Node(name="NodeTest2")
node_test.save()
response = client.get(reverse("node_list"))
assert response.status_code == 200
assert "NodeTest1" in str(response.content)
assert "NodeTest2" in str(response.content)
def test_node_update_increments_checkin_count():
node = Node(name="test node")
node.save()
assert node.total_checkins == 0
node.update()
node.update()
assert node.total_checkins == 2
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_turns_into_string():
node = Node(name="test node")
assert str(node) == "test node"