X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/8994f77f368ee813738f4406f68239bc4013e20e..45da0e8226577f64ec397f0af03e6a46fc8cb386:/services/api/test/functional/arvados/v1/keep_disks_controller_test.rb

diff --git a/services/api/test/functional/arvados/v1/keep_disks_controller_test.rb b/services/api/test/functional/arvados/v1/keep_disks_controller_test.rb
index ea3b18c707..57cbd0024e 100644
--- a/services/api/test/functional/arvados/v1/keep_disks_controller_test.rb
+++ b/services/api/test/functional/arvados/v1/keep_disks_controller_test.rb
@@ -2,6 +2,63 @@ require 'test_helper'
 
 class Arvados::V1::KeepDisksControllerTest < ActionController::TestCase
 
+  test "add keep node with admin token" do
+    authorize_with :admin
+    post :ping, {
+      ping_secret: '',          # required by discovery doc, but ignored
+      service_host: '::1',
+      service_port: 55555,
+      service_ssl_flag: false,
+      filesystem_uuid: 'eb1e77a1-db84-4193-b6e6-ca2894f67d5f'
+    }
+    assert_response :success
+    assert_not_nil assigns(:object)
+    new_keep_node = JSON.parse(@response.body)
+    assert_not_nil new_keep_node['uuid']
+    assert_not_nil new_keep_node['ping_secret']
+    assert_not_equal '', new_keep_node['ping_secret']
+  end
+
+  test "add keep node with no filesystem_uuid" do
+    authorize_with :admin
+    opts = {
+      ping_secret: '',
+      service_host: '::1',
+      service_port: 55555,
+      service_ssl_flag: false
+    }
+    post :ping, opts
+    assert_response :success
+    assert_not_nil JSON.parse(@response.body)['uuid']
+
+    post :ping, opts.merge(filesystem_uuid: '')
+    assert_response :success
+    assert_not_nil JSON.parse(@response.body)['uuid']
+  end
+
+  test "refuse to add keep node without admin token" do
+    post :ping, {
+      ping_secret: '',
+      service_host: '::1',
+      service_port: 55555,
+      service_ssl_flag: false
+    }
+    assert_response 404
+  end
+
+  test "ping from keep node" do
+    post :ping, {
+      uuid: keep_disks(:nonfull).uuid,
+      ping_secret: keep_disks(:nonfull).ping_secret,
+      filesystem_uuid: keep_disks(:nonfull).filesystem_uuid
+    }
+    assert_response :success
+    assert_not_nil assigns(:object)
+    keep_node = JSON.parse(@response.body)
+    assert_not_nil keep_node['uuid']
+    assert_not_nil keep_node['ping_secret']
+  end
+
   test "should get index with ping_secret" do
     authorize_with :admin
     get :index