services/api/test/functional/arvados/v1/nodes_controller_test.rb

   1 require 'test_helper'
   2
   3 class Arvados::V1::NodesControllerTest < ActionController::TestCase
   4
   5   test "should get index with ping_secret" do
   6     authorize_with :admin
   7     get :index
   8     assert_response :success
   9     assert_not_nil assigns(:objects)
  10     node_items = JSON.parse(@response.body)['items']
  11     assert_not_equal 0, node_items.size
  12     assert_not_nil node_items[0]['info'].andand['ping_secret']
  13   end
  14
  15   # inactive user does not see any nodes
  16   test "inactive user should get empty index" do
  17     authorize_with :inactive
  18     get :index
  19     assert_response :success
  20     node_items = JSON.parse(@response.body)['items']
  21     assert_equal 0, node_items.size
  22   end
  23
  24   # active user sees non-secret attributes of up and recently-up nodes
  25   test "active user should get non-empty index with no ping_secret" do
  26     authorize_with :active
  27     get :index
  28     assert_response :success
  29     node_items = JSON.parse(@response.body)['items']
  30     assert_not_equal 0, node_items.size
  31     found_busy_node = false
  32     node_items.each do |node|
  33       assert_nil node['info'].andand['ping_secret']
  34       assert_not_nil node['crunch_worker_state']
  35       if node['uuid'] == nodes(:busy).uuid
  36         found_busy_node = true
  37         assert_equal 'busy', node['crunch_worker_state']
  38       end
  39     end
  40     assert_equal true, found_busy_node
  41   end
  42
  43   test "node should ping with ping_secret and no token" do
  44     post :ping, {
  45       id: 'zzzzz-7ekkf-2z3mc76g2q73aio',
  46       instance_id: 'i-0000000',
  47       local_ipv4: '172.17.2.174',
  48       ping_secret: '69udawxvn3zzj45hs8bumvndricrha4lcpi23pd69e44soanc0'
  49     }
  50     assert_response :success
  51     response = JSON.parse(@response.body)
  52     assert_equal 'zzzzz-7ekkf-2z3mc76g2q73aio', response['uuid']
  53     # Ensure we are getting the "superuser" attributes, too
  54     assert_not_nil response['first_ping_at'], '"first_ping_at" attr missing'
  55     assert_not_nil response['info'], '"info" attr missing'
  56     assert_not_nil response['nameservers'], '"nameservers" attr missing'
  57   end
  58
  59   test "node should fail ping with invalid ping_secret" do
  60     post :ping, {
  61       id: 'zzzzz-7ekkf-2z3mc76g2q73aio',
  62       instance_id: 'i-0000000',
  63       local_ipv4: '172.17.2.174',
  64       ping_secret: 'dricrha4lcpi23pd69e44soanc069udawxvn3zzj45hs8bumvn'
  65     }
  66     assert_response 401
  67   end
  68
  69   test "create node" do
  70     authorize_with :admin
  71     post :create
  72     assert_response :success
  73     assert_not_nil json_response['uuid']
  74     assert_not_nil json_response['info'].is_a? Hash
  75     assert_not_nil json_response['info']['ping_secret']
  76   end
  77
  78   test "ping adds node stats to info" do
  79     node = nodes(:idle)
  80     post :ping, {
  81       id: node.uuid,
  82       ping_secret: node.info['ping_secret'],
  83       total_cpu_cores: 32,
  84       total_ram_mb: 1024,
  85       total_scratch_mb: 2048
  86     }
  87     assert_response :success
  88     info = JSON.parse(@response.body)['info']
  89     assert_equal(node.info['ping_secret'], info['ping_secret'])
  90     assert_equal(32, info['total_cpu_cores'].to_i)
  91     assert_equal(1024, info['total_ram_mb'].to_i)
  92     assert_equal(2048, info['total_scratch_mb'].to_i)
  93   end
  94 end