Fix some whitespace complaints.
[arvados.git] / services / api / test / unit / user_test.rb
index 9922ef4b3d66245b09c7b9d5b4df5a0b3a518c6e..9f78b1a3650320c29c37cc9d002d0e2cac484c41 100644 (file)
@@ -2,36 +2,36 @@ require 'test_helper'
 
 class UserTest < ActiveSupport::TestCase
 
-       # The fixture services/api/test/fixtures/users.yml serves as the input for this test case
-       setup do
-               @all_users = User.find(:all)
-
-               @all_users.each do |user|
-                       if user.is_admin && user.is_active 
-                               @admin_user = user
-                       elsif user.is_active && !user.is_admin
-                               @active_user = user
-                       elsif !user.is_active && !user.is_invited 
-                               @uninvited_user = user
-                       end
-               end
-       end
+  # The fixture services/api/test/fixtures/users.yml serves as the input for this test case
+  setup do
+    @all_users = User.find(:all)
+
+    @all_users.each do |user|
+      if user.is_admin && user.is_active
+        @admin_user = user
+      elsif user.is_active && !user.is_admin
+        @active_user = user
+      elsif !user.is_active && !user.is_invited
+        @uninvited_user = user
+      end
+    end
+  end
 
   test "check non-admin active user properties" do
     assert !@active_user.is_admin, 'is_admin should not be set for a non-admin user'
     assert @active_user.is_active, 'user should be active'
     assert @active_user.is_invited, 'is_invited should be set'
     assert_not_nil @active_user.prefs, "user's preferences should be non-null, but may be size zero"
-               assert (@active_user.can? :read=>"#{@active_user.uuid}"), "user should be able to read own object"
-               assert (@active_user.can? :write=>"#{@active_user.uuid}"), "user should be able to write own object"
-               assert (@active_user.can? :manage=>"#{@active_user.uuid}"), "user should be able to manage own object"
+    assert (@active_user.can? :read=>"#{@active_user.uuid}"), "user should be able to read own object"
+    assert (@active_user.can? :write=>"#{@active_user.uuid}"), "user should be able to write own object"
+    assert (@active_user.can? :manage=>"#{@active_user.uuid}"), "user should be able to manage own object"
 
-               assert @active_user.groups_i_can(:read).size > 0, "active user should be able read at least one group"
+    assert @active_user.groups_i_can(:read).size > 0, "active user should be able read at least one group"
 
     # non-admin user cannot manage or write other user objects
-               assert !(@active_user.can? :read=>"#{@uninvited_user.uuid}")
-               assert !(@active_user.can? :write=>"#{@uninvited_user.uuid}")
-               assert !(@active_user.can? :manage=>"#{@uninvited_user.uuid}")
+    assert !(@active_user.can? :read=>"#{@uninvited_user.uuid}")
+    assert !(@active_user.can? :write=>"#{@uninvited_user.uuid}")
+    assert !(@active_user.can? :manage=>"#{@uninvited_user.uuid}")
   end
 
   test "check admin user properties" do
@@ -41,101 +41,241 @@ class UserTest < ActiveSupport::TestCase
     assert_not_nil @admin_user.uuid.size, "user's uuid should be non-null"
     assert_not_nil @admin_user.prefs, "user's preferences should be non-null, but may be size zero"
     assert @admin_user.identity_url.size > 0, "user's identity url is expected"
-               assert @admin_user.can? :read=>"#{@admin_user.uuid}"
-               assert @admin_user.can? :write=>"#{@admin_user.uuid}"
-               assert @admin_user.can? :manage=>"#{@admin_user.uuid}"
+    assert @admin_user.can? :read=>"#{@admin_user.uuid}"
+    assert @admin_user.can? :write=>"#{@admin_user.uuid}"
+    assert @admin_user.can? :manage=>"#{@admin_user.uuid}"
 
-               assert @admin_user.groups_i_can(:read).size > 0, "admin active user should be able read at least one group"
-               assert @admin_user.groups_i_can(:write).size > 0, "admin active user should be able write to at least one group"
-               assert @admin_user.groups_i_can(:manage).size > 0, "admin active user should be able manage at least one group"
+    assert @admin_user.groups_i_can(:read).size > 0, "admin active user should be able read at least one group"
+    assert @admin_user.groups_i_can(:write).size > 0, "admin active user should be able write to at least one group"
+    assert @admin_user.groups_i_can(:manage).size > 0, "admin active user should be able manage at least one group"
 
     # admin user can also write or manage other users
-               assert @admin_user.can? :read=>"#{@uninvited_user.uuid}"
-               assert @admin_user.can? :write=>"#{@uninvited_user.uuid}"
-               assert @admin_user.can? :manage=>"#{@uninvited_user.uuid}"
+    assert @admin_user.can? :read=>"#{@uninvited_user.uuid}"
+    assert @admin_user.can? :write=>"#{@uninvited_user.uuid}"
+    assert @admin_user.can? :manage=>"#{@uninvited_user.uuid}"
   end
 
   test "check inactive and uninvited user properties" do
     assert !@uninvited_user.is_admin, 'is_admin should not be set for a non-admin user'
     assert !@uninvited_user.is_active, 'user should be inactive'
     assert !@uninvited_user.is_invited, 'is_invited should not be set'
-               assert @uninvited_user.can? :read=>"#{@uninvited_user.uuid}"
-               assert @uninvited_user.can? :write=>"#{@uninvited_user.uuid}"
-               assert @uninvited_user.can? :manage=>"#{@uninvited_user.uuid}"
+    assert @uninvited_user.can? :read=>"#{@uninvited_user.uuid}"
+    assert @uninvited_user.can? :write=>"#{@uninvited_user.uuid}"
+    assert @uninvited_user.can? :manage=>"#{@uninvited_user.uuid}"
 
-               assert @uninvited_user.groups_i_can(:read).size == 0, "inactive and uninvited user should not be able read any groups"
-               assert @uninvited_user.groups_i_can(:write).size == 0, "inactive and uninvited user should not be able write to any groups"
-               assert @uninvited_user.groups_i_can(:manage).size == 0, "inactive and uninvited user should not be able manage any groups"
+    assert @uninvited_user.groups_i_can(:read).size == 0, "inactive and uninvited user should not be able read any groups"
+    assert @uninvited_user.groups_i_can(:write).size == 0, "inactive and uninvited user should not be able write to any groups"
+    assert @uninvited_user.groups_i_can(:manage).size == 0, "inactive and uninvited user should not be able manage any groups"
   end
 
   test "find user method checks" do
-               User.find(:all).each do |user|
-                       assert_not_nil user.uuid, "non-null uuid expected for " + user.full_name
-               end
+    User.find(:all).each do |user|
+      assert_not_nil user.uuid, "non-null uuid expected for " + user.full_name
+    end
 
-    user = users(:active)              # get the active user
+    user = users(:active)     # get the active user
 
-               found_user = User.find(user.id)         # find a user by the row id
+    found_user = User.find(user.id)   # find a user by the row id
 
     assert_equal found_user.full_name, user.first_name + ' ' + user.last_name
     assert_equal found_user.identity_url, user.identity_url
   end
 
-       test "create new user" do 
-               Thread.current[:user] = @admin_user     # set admin user as the current user
-
-               user = User.new
-               user.first_name = "first_name_for_newly_created_user"
-               user.save
-
-               # verify there is one extra user in the db now
-               assert (User.find(:all).size == @all_users.size+1)
-
-               user = User.find(user.id)               # get the user back
-               assert_equal(user.first_name, 'first_name_for_newly_created_user')
-               assert_not_nil user.uuid, 'uuid should be set for newly created user'
-               assert_nil user.email, 'email should be null for newly created user, because it was not passed in'
-               assert_nil user.identity_url, 'identity_url should be null for newly created user, because it was not passed in'
-
-               user.first_name = 'first_name_for_newly_created_user_updated'
-               user.save
-               user = User.find(user.id)               # get the user back
-               assert_equal(user.first_name, 'first_name_for_newly_created_user_updated')
-       end
-
-       test "update existing user" do 
-               Thread.current[:user] = @active_user            # set active user as current user
-               @active_user.first_name = "first_name_changed"
-               @active_user.save
-
-               @active_user = User.find(@active_user.id)               # get the user back
-               assert_equal(@active_user.first_name, 'first_name_changed')
-
-               # admin user also should be able to update the "active" user info
-               Thread.current[:user] = @admin_user # set admin user as current user
-               @active_user.first_name = "first_name_changed_by_admin_for_active_user"
-               @active_user.save
-
-               @active_user = User.find(@active_user.id)               # get the user back
-               assert_equal(@active_user.first_name, 'first_name_changed_by_admin_for_active_user')
-       end
-       
-       test "delete a user and verify" do 
-               active_user_uuid = @active_user.uuid
-
-               Thread.current[:user] = @admin_user             
-               @active_user.delete
-
-               found_deleted_user = false
-               User.find(:all).each do |user| 
-                       if user.uuid == active_user_uuid 
-                               found_deleted_user = true
-                               break
-                       end                     
-               end
-               assert !found_deleted_user, "found deleted user: "+active_user_uuid
-       
-       end
+  test "create new user" do
+    Thread.current[:user] = @admin_user   # set admin user as the current user
 
-end
+    user = User.new
+    user.first_name = "first_name_for_newly_created_user"
+    user.save
+
+    # verify there is one extra user in the db now
+    assert (User.find(:all).size == @all_users.size+1)
+
+    user = User.find(user.id)   # get the user back
+    assert_equal(user.first_name, 'first_name_for_newly_created_user')
+    assert_not_nil user.uuid, 'uuid should be set for newly created user'
+    assert_nil user.email, 'email should be null for newly created user, because it was not passed in'
+    assert_nil user.identity_url, 'identity_url should be null for newly created user, because it was not passed in'
+
+    user.first_name = 'first_name_for_newly_created_user_updated'
+    user.save
+    user = User.find(user.id)   # get the user back
+    assert_equal(user.first_name, 'first_name_for_newly_created_user_updated')
+  end
+
+  test "update existing user" do
+    Thread.current[:user] = @active_user    # set active user as current user
+    @active_user.first_name = "first_name_changed"
+    @active_user.save
+
+    @active_user = User.find(@active_user.id)   # get the user back
+    assert_equal(@active_user.first_name, 'first_name_changed')
+
+    # admin user also should be able to update the "active" user info
+    Thread.current[:user] = @admin_user # set admin user as current user
+    @active_user.first_name = "first_name_changed_by_admin_for_active_user"
+    @active_user.save
+
+    @active_user = User.find(@active_user.id)   # get the user back
+    assert_equal(@active_user.first_name, 'first_name_changed_by_admin_for_active_user')
+  end
+
+  test "delete a user and verify" do
+    active_user_uuid = @active_user.uuid
+
+    Thread.current[:user] = @admin_user
+    @active_user.delete
+
+    found_deleted_user = false
+    User.find(:all).each do |user|
+      if user.uuid == active_user_uuid
+        found_deleted_user = true
+        break
+      end
+    end
+    assert !found_deleted_user, "found deleted user: "+active_user_uuid
+
+  end
+
+  test "create new user as non-admin user" do
+    Thread.current[:user] = @active_user
+
+    begin
+      user = User.new
+      user.save
+    rescue ArvadosModel::PermissionDeniedError => e
+    end
+    assert (e.message.include? 'PermissionDeniedError'),
+        'Expected PermissionDeniedError'
+  end
+
+  test "setup new user" do
+    Thread.current[:user] = @admin_user
+
+    email = 'foo@example.com'
+    openid_prefix = 'http://openid/prefix'
+
+    user = User.new
+    user.email = email
+    user.uuid = 'abcdefghijklmnop'
+
+    vm = VirtualMachine.create
+
+    response = User.setup user, openid_prefix, 'test_repo', vm.uuid
+
+    resp_user = find_obj_in_resp response, 'User'
+    verify_user resp_user, email
+
+    oid_login_perm = find_obj_in_resp response, 'Link', 'arvados#user'
+    verify_link oid_login_perm, 'permission', 'can_login', resp_user[:email],
+        resp_user[:uuid]
+    assert_equal openid_prefix, oid_login_perm[:properties][:identity_url_prefix],
+        'expected identity_url_prefix not found for oid_login_perm'
+
+    group_perm = find_obj_in_resp response, 'Link', 'arvados#group'
+    verify_link group_perm, 'permission', 'can_read', resp_user[:uuid], nil
 
+    repo_perm = find_obj_in_resp response, 'Link', 'arvados#repository'
+    verify_link repo_perm, 'permission', 'can_write', resp_user[:uuid], nil
+
+    vm_perm = find_obj_in_resp response, 'Link', 'arvados#virtualMachine'
+    verify_link vm_perm, 'permission', 'can_login', resp_user[:uuid], vm.uuid
+  end
+
+  test "setup new user in multiple steps" do
+    Thread.current[:user] = @admin_user
+
+    email = 'foo@example.com'
+    openid_prefix = 'http://openid/prefix'
+
+    user = User.new
+    user.email = email
+    user.uuid = 'abcdefghijklmnop'
+
+    response = User.setup user, openid_prefix
+
+    resp_user = find_obj_in_resp response, 'User'
+    verify_user resp_user, email
+
+    oid_login_perm = find_obj_in_resp response, 'Link', 'arvados#user'
+    verify_link oid_login_perm, 'permission', 'can_login', resp_user[:email],
+        resp_user[:uuid]
+    assert_equal openid_prefix, oid_login_perm[:properties][:identity_url_prefix],
+        'expected identity_url_prefix not found for oid_login_perm'
+
+    group_perm = find_obj_in_resp response, 'Link', 'arvados#group'
+    verify_link group_perm, 'permission', 'can_read', resp_user[:uuid], nil
+
+    # invoke setup again with repo_name
+    response = User.setup user, openid_prefix, 'test_repo'
+    resp_user = find_obj_in_resp response, 'User', nil
+    verify_user resp_user, email
+    assert_equal user.uuid, resp_user[:uuid], 'expected uuid not found'
+
+    group_perm = find_obj_in_resp response, 'Link', 'arvados#group'
+    verify_link group_perm, 'permission', 'can_read', resp_user[:uuid], nil
+
+    repo_perm = find_obj_in_resp response, 'Link', 'arvados#repository'
+    verify_link repo_perm, 'permission', 'can_write', resp_user[:uuid], nil
+
+    # invoke setup again with a vm_uuid
+    vm = VirtualMachine.create
+
+    response = User.setup user, openid_prefix, 'test_repo', vm.uuid
+
+    resp_user = find_obj_in_resp response, 'User', nil
+    verify_user resp_user, email
+    assert_equal user.uuid, resp_user[:uuid], 'expected uuid not found'
+
+    group_perm = find_obj_in_resp response, 'Link', 'arvados#group'
+    verify_link group_perm, 'permission', 'can_read', resp_user[:uuid], nil
+
+    repo_perm = find_obj_in_resp response, 'Link', 'arvados#repository'
+    verify_link repo_perm, 'permission', 'can_write', resp_user[:uuid], nil
+
+    vm_perm = find_obj_in_resp response, 'Link', 'arvados#virtualMachine'
+    verify_link vm_perm, 'permission', 'can_login', resp_user[:uuid], vm.uuid
+  end
+
+  def find_obj_in_resp (response, object_type, head_kind=nil)
+    return_obj = nil
+    response.each { |x|
+      if x.class.name == object_type
+        if head_kind
+          if x.head_kind == head_kind
+            return_obj = x
+            break
+          end
+        else
+          return_obj = x
+          break
+        end
+      end
+    }
+    return return_obj
+  end
+
+  def verify_user (resp_user, email)
+    assert_not_nil resp_user, 'expected user object'
+    assert_not_nil resp_user['uuid'], 'expected user object'
+    assert_equal email, resp_user['email'], 'expected email not found'
+
+  end
+
+  def verify_link (link_object, link_class, link_name, tail_uuid, head_uuid)
+    assert_not_nil link_object, 'expected link for #{link_class} #{link_name}'
+    assert_not_nil link_object[:uuid],
+        'expected non-nil uuid for link for #{link_class} #{link_name}'
+    assert_equal link_class, link_object[:link_class],
+        'expected link_class not found for #{link_class} #{link_name}'
+    assert_equal link_name, link_object[:name],
+        'expected link_name not found for #{link_class} #{link_name}'
+    assert_equal tail_uuid, link_object[:tail_uuid],
+        'expected tail_uuid not found for #{link_class} #{link_name}'
+    if head_uuid
+      assert_equal head_uuid, link_object[:head_uuid],
+          'expected head_uuid not found for #{link_class} #{link_name}'
+    end
+  end
+
+end