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_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 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_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 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