refute_includes(node.properties, 'total_cpu_cores')
assert_equal(256, node.properties['total_ram_mb'])
end
+
+ test "worker state is down for node with no slot" do
+ node = nodes(:was_idle_now_down)
+ assert_nil node.slot_number, "fixture is not what I expected"
+ assert_equal 'down', node.crunch_worker_state, "wrong worker state"
+ end
+
+ test "dns_server_conf_template" do
+ Rails.configuration.dns_server_conf_dir = Rails.root.join 'tmp'
+ Rails.configuration.dns_server_conf_template = Rails.root.join 'config', 'unbound.template'
+ conffile = Rails.root.join 'tmp', 'compute65535.conf'
+ File.unlink conffile rescue nil
+ assert Node.dns_server_update 'compute65535', '127.0.0.1'
+ assert_match /\"1\.0\.0\.127\.in-addr\.arpa\. IN PTR compute65535\.zzzzz\.arvadosapi\.com\"/, IO.read(conffile)
+ File.unlink conffile
+ end
+
+ test "dns_server_restart_command" do
+ Rails.configuration.dns_server_conf_dir = Rails.root.join 'tmp'
+ Rails.configuration.dns_server_reload_command = 'foobar'
+ restartfile = Rails.root.join 'tmp', 'restart.txt'
+ File.unlink restartfile rescue nil
+ assert Node.dns_server_update 'compute65535', '127.0.0.127'
+ assert_equal "foobar\n", IO.read(restartfile)
+ File.unlink restartfile
+ end
+
+ test "dns_server_restart_command fail" do
+ Rails.configuration.dns_server_conf_dir = Rails.root.join 'tmp', 'bogusdir'
+ Rails.configuration.dns_server_reload_command = 'foobar'
+ refute Node.dns_server_update 'compute65535', '127.0.0.127'
+ end
+
+ test "dns_server_update_command with valid command" do
+ testfile = Rails.root.join('tmp', 'node_test_dns_server_update_command.txt')
+ Rails.configuration.dns_server_update_command =
+ ('echo -n "%{hostname} == %{ip_address}" >' +
+ testfile.to_s.shellescape)
+ assert Node.dns_server_update 'compute65535', '127.0.0.1'
+ assert_equal 'compute65535 == 127.0.0.1', IO.read(testfile)
+ File.unlink testfile
+ end
+
+ test "dns_server_update_command with failing command" do
+ Rails.configuration.dns_server_update_command = 'false %{hostname}'
+ refute Node.dns_server_update 'compute65535', '127.0.0.1'
+ end
+
+ test "dns update with no commands/dirs configured" do
+ Rails.configuration.dns_server_update_command = false
+ Rails.configuration.dns_server_conf_dir = false
+ Rails.configuration.dns_server_conf_template = 'ignored!'
+ Rails.configuration.dns_server_reload_command = 'ignored!'
+ assert Node.dns_server_update 'compute65535', '127.0.0.127'
+ end
+
+ test "ping new node with no hostname and default config" do
+ node = ping_node(:new_with_no_hostname, {})
+ slot_number = node.slot_number
+ refute_nil slot_number
+ assert_equal("compute#{slot_number}", node.hostname)
+ end
+
+ test "ping new node with no hostname and no config" do
+ Rails.configuration.assign_node_hostname = false
+ node = ping_node(:new_with_no_hostname, {})
+ refute_nil node.slot_number
+ assert_nil node.hostname
+ end
+
+ test "ping new node with zero padding config" do
+ Rails.configuration.assign_node_hostname = 'compute%<slot_number>04d'
+ node = ping_node(:new_with_no_hostname, {})
+ slot_number = node.slot_number
+ refute_nil slot_number
+ assert_equal("compute000#{slot_number}", node.hostname)
+ end
+
+ test "ping node with hostname and config and expect hostname unchanged" do
+ node = ping_node(:new_with_custom_hostname, {})
+ assert_equal(23, node.slot_number)
+ assert_equal("custom1", node.hostname)
+ end
+
+ test "ping node with hostname and no config and expect hostname unchanged" do
+ Rails.configuration.assign_node_hostname = false
+ node = ping_node(:new_with_custom_hostname, {})
+ assert_equal(23, node.slot_number)
+ assert_equal("custom1", node.hostname)
+ end
+
+ # Ping two nodes: one without a hostname and the other with a hostname.
+ # Verify that the first one gets a hostname and second one is unchanged.
+ test "ping two nodes one with no hostname and one with hostname and check hostnames" do
+ # ping node with no hostname and expect it set with config format
+ node = ping_node(:new_with_no_hostname, {})
+ slot_number = node.slot_number
+ refute_nil node.slot_number
+ assert_equal "compute#{slot_number}", node.hostname
+
+ # ping node with a hostname and expect it to be unchanged
+ node2 = ping_node(:new_with_custom_hostname, {})
+ refute_nil node2.slot_number
+ assert_equal "custom1", node2.hostname
+ end
+
+ test "update dns when nodemanager clears hostname and ip_address" do
+ act_as_system_user do
+ node = ping_node(:new_with_custom_hostname, {})
+ Node.expects(:dns_server_update).with(node.hostname, Node::UNUSED_NODE_IP)
+ node.update_attributes(hostname: nil, ip_address: nil)
+ end
+ end
+
+ test "update dns when hostname changes" do
+ act_as_system_user do
+ node = ping_node(:new_with_custom_hostname, {})
+
+ Node.expects(:dns_server_update).with(node.hostname, Node::UNUSED_NODE_IP)
+ Node.expects(:dns_server_update).with('foo0', node.ip_address)
+ node.update_attributes!(hostname: 'foo0')
+
+ Node.expects(:dns_server_update).with('foo0', Node::UNUSED_NODE_IP)
+ node.update_attributes!(hostname: nil, ip_address: nil)
+
+ Node.expects(:dns_server_update).with('foo0', '10.11.12.13')
+ node.update_attributes!(hostname: 'foo0', ip_address: '10.11.12.13')
+
+ Node.expects(:dns_server_update).with('foo0', '10.11.12.14')
+ node.update_attributes!(hostname: 'foo0', ip_address: '10.11.12.14')
+ end
+ end
end