X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2ff9ca2fb06d644328896be242186d5fbc1ffd17..6d73acc845a2dd7413fdde0742473d83bf3d0719:/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..385710e060 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,7 +2,64 @@ require 'test_helper'
 
 class Arvados::V1::KeepDisksControllerTest < ActionController::TestCase
 
-  test "should get index with ping_secret" do
+  test "add keep disk 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_disk = JSON.parse(@response.body)
+    assert_not_nil new_keep_disk['uuid']
+    assert_not_nil new_keep_disk['ping_secret']
+    assert_not_equal '', new_keep_disk['ping_secret']
+  end
+
+  test "add keep disk 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 disk without admin token" do
+    post :ping, {
+      ping_secret: '',
+      service_host: '::1',
+      service_port: 55555,
+      service_ssl_flag: false
+    }
+    assert_response 404
+  end
+
+  test "ping keep disk" 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_disk = JSON.parse(@response.body)
+    assert_not_nil keep_disk['uuid']
+    assert_not_nil keep_disk['ping_secret']
+  end
+
+  test "admin should get index with ping_secret" do
     authorize_with :admin
     get :index
     assert_response :success
@@ -12,13 +69,13 @@ class Arvados::V1::KeepDisksControllerTest < ActionController::TestCase
     assert_not_nil items[0]['ping_secret']
   end
 
-  # inactive user does not see any keep disks
-  test "inactive user should get empty index" do
+  # inactive user sees keep disks
+  test "inactive user should get index" do
     authorize_with :inactive
     get :index
     assert_response :success
     items = JSON.parse(@response.body)['items']
-    assert_equal 0, items.size
+    assert_not_equal 0, items.size
   end
 
   # active user sees non-secret attributes of keep disks
@@ -37,4 +94,32 @@ class Arvados::V1::KeepDisksControllerTest < ActionController::TestCase
     end
   end
 
+  test "search keep_disks by service_port with >= query" do
+    authorize_with :active
+    get :index, {
+      filters: [['service_port', '>=', 25107]]
+    }
+    assert_response :success
+    assert_equal true, assigns(:objects).any?
+  end
+
+  test "search keep_disks by service_port with < query" do
+    authorize_with :active
+    get :index, {
+      filters: [['service_port', '<', 25107]]
+    }
+    assert_response :success
+    assert_equal false, assigns(:objects).any?
+  end
+
+  test "search keep_disks with 'any' operator" do
+    authorize_with :active
+    get :index, {
+      where: { any: ['contains', 'o2t1q5w'] }
+    }
+    assert_response :success
+    found = assigns(:objects).collect(&:uuid)
+    assert_equal true, !!found.index('zzzzz-penuu-5w2o2t1q5wy7fhn')
+  end
+
 end