foreman_dlm 0.1.0 → 2.0.0

data/lib/tasks/foreman_dlm_tasks.rake

@@ -21,7 +21,7 @@ namespace :foreman_dlm do
                          "#{ForemanDlm::Engine.root}/lib/**/*.rb",
                          "#{ForemanDlm::Engine.root}/test/**/*.rb"]
       end
-    rescue
+    rescue StandardError
       puts 'Rubocop not loaded.'
     end
 
@@ -32,6 +32,4 @@ end
 Rake::Task[:test].enhance ['test:foreman_dlm']
 
 load 'tasks/jenkins.rake'
-if Rake::Task.task_defined?(:'jenkins:unit')
-  Rake::Task['jenkins:unit'].enhance ['test:foreman_dlm', 'foreman_dlm:rubocop']
-end
+Rake::Task['jenkins:unit'].enhance ['test:foreman_dlm', 'foreman_dlm:rubocop'] if Rake::Task.task_defined?(:'jenkins:unit')

data/test/controllers/api/v2/dlmlocks_controller_test.rb

@@ -13,7 +13,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
         body = ActiveSupport::JSON.decode(@response.body)
         results = body['results']
         assert results
-        entry = results.detect { |entry| entry['id'] == dlmlock.id }
+        entry = results.detect { |e| e['id'] == dlmlock.id }
         assert entry
         assert_equal dlmlock.name, entry['name']
         assert_equal dlmlock.type, entry['type']
@@ -24,12 +24,12 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
 
     context '#create' do
       test 'should create dlmlock' do
-        assert_difference('Dlmlock.unscoped.count') do
-          post :create, valid_attrs_with_root
+        assert_difference('ForemanDlm::Dlmlock.unscoped.count') do
+          post :create, params: valid_attrs_with_root
         end
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        dlmlock = Dlmlock.find(body['id'])
+        dlmlock = ForemanDlm::Dlmlock.find(body['id'])
         assert dlmlock
         assert_equal valid_attrs['name'], dlmlock.name
         assert_equal valid_attrs['type'], dlmlock.type
@@ -41,10 +41,10 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#show' do
       test 'should show individual record with host' do
         dlmlock = FactoryBot.create(:dlmlock, :host => host1)
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.name, body['name']
         assert_equal dlmlock.type, body['type']
         assert_equal true, body['enabled']
@@ -53,10 +53,10 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
 
       test 'should show individual record that is disabled' do
         dlmlock = FactoryBot.create(:dlmlock, :enabled => false)
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.name, body['name']
         assert_equal dlmlock.type, body['type']
         assert_equal false, body['enabled']
@@ -65,20 +65,20 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
 
       test 'should show individual record by name' do
         dlmlock = FactoryBot.create(:dlmlock, :host => host1)
-        get :show, { :id => dlmlock.name }
+        get :show, params: { :id => dlmlock.name }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.id, body['id']
         assert_equal dlmlock.name, body['name']
         host = body['host']
         assert host
         assert_equal host1.name, host['name']
-        refute host.has_key?('self')
+        assert_not host.key?('self')
       end
 
       test 'should not find dlmlock with invalid id' do
-        get :show, { :id => 9999999 }
+        get :show, params: { :id => 9_999_999 }
         assert_response :not_found
       end
     end
@@ -86,7 +86,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#update' do
       test 'should update dlmlock' do
         dlmlock = FactoryBot.create(:dlmlock)
-        put :update, { :id => dlmlock.to_param, :dlmlock => valid_attrs.merge(:host_id => host1.id, :enabled => false) }
+        put :update, params: { :id => dlmlock.to_param, :dlmlock => valid_attrs.merge(:host_id => host1.id, :enabled => false) }
         assert_response :success
         dlmlock.reload
         assert_equal valid_attrs['name'], dlmlock.name
@@ -99,18 +99,18 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#destroy' do
       test 'should destroy dlmlock' do
         dlmlock = FactoryBot.create(:dlmlock)
-        assert_difference('Dlmlock.unscoped.count', -1) do
-          delete :destroy, { :id => dlmlock.to_param }
+        assert_difference('ForemanDlm::Dlmlock.unscoped.count', -1) do
+          delete :destroy, params: { :id => dlmlock.to_param }
         end
         assert_response :success
-        assert_equal 0, Dlmlock.where(:id => dlmlock.id).count
+        assert_equal 0, ForemanDlm::Dlmlock.where(:id => dlmlock.id).count
       end
     end
 
     context '#acquire' do
       test 'should deny access' do
         dlmlock = FactoryBot.create(:dlmlock)
-        put :acquire, { :id => dlmlock.to_param }
+        put :acquire, params: { :id => dlmlock.to_param }
         assert_response :forbidden
         assert_nil dlmlock.reload.host
       end
@@ -119,7 +119,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#release' do
       test 'should deny access' do
         dlmlock = FactoryBot.create(:dlmlock, :host => host2)
-        delete :release, { :id => dlmlock.to_param }
+        delete :release, params: { :id => dlmlock.to_param }
         assert_response :forbidden
         assert_equal host2, dlmlock.reload.host
       end
@@ -142,15 +142,15 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#index' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        get :index, { :id => dlmlock.to_param }
+        get :index, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
       end
     end
 
     context '#create' do
       test 'should deny access' do
-        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        post :create, valid_attrs_with_root
+        as_admin { FactoryBot.create(:dlmlock) }
+        post :create, params: valid_attrs_with_root
         assert_response :unauthorized
       end
     end
@@ -158,7 +158,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#update' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        put :update, { :id => dlmlock.to_param, :dlmlock => valid_attrs }
+        put :update, params: { :id => dlmlock.to_param, :dlmlock => valid_attrs }
         assert_response :unauthorized
       end
     end
@@ -166,19 +166,19 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#destroy' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        delete :destroy, { :id => dlmlock.to_param }
+        delete :destroy, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
-        assert_equal 1, as_admin { Dlmlock.where(:id => dlmlock.id).count }
+        assert_equal 1, as_admin { ForemanDlm::Dlmlock.where(:id => dlmlock.id).count }
       end
     end
 
     context '#show' do
       test 'should show individual free lock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.name, body['name']
         assert_equal dlmlock.type, body['type']
         assert_equal true, body['enabled']
@@ -188,10 +188,10 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
 
       test 'should show individual acquired lock by me' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host1) }
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.name, body['name']
         assert_equal dlmlock.type, body['type']
         assert_equal true, body['enabled']
@@ -204,10 +204,10 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
 
       test 'should show individual acquired lock by other' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host2) }
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :success
         body = ActiveSupport::JSON.decode(@response.body)
-        refute_empty body
+        assert_not_empty body
         assert_equal dlmlock.name, body['name']
         assert_equal dlmlock.type, body['type']
         assert_equal true, body['enabled']
@@ -217,72 +217,93 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
         assert_equal host2.name, host['name']
         assert_equal false, host['self']
       end
+
+      test 'should update checkin time' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        assert_not as_admin { host1.dlm_facet }
+        put :show, params: { :id => dlmlock.to_param }
+        assert as_admin { host1.reload.dlm_facet.last_checkin_at }
+      end
     end
 
     context '#acquire' do
       test 'should acquire empty dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        put :acquire, { :id => dlmlock.to_param }
+        put :acquire, params: { :id => dlmlock.to_param }
         assert_response :success
         assert_equal host1, as_admin { dlmlock.reload.host }
       end
 
       test 'should acquire own dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host1) }
-        put :acquire, { :id => dlmlock.to_param }
+        put :acquire, params: { :id => dlmlock.to_param }
         assert_response :success
         assert_equal host1, as_admin { dlmlock.reload.host }
       end
 
       test 'should not acquire foreign dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host2) }
-        put :acquire, { :id => dlmlock.to_param }
+        put :acquire, params: { :id => dlmlock.to_param }
         assert_response :precondition_failed
         assert_equal host2, as_admin { dlmlock.reload.host }
       end
 
       test 'should transparently create non-existing dlmlock' do
         lockname = 'Test Lock'
-        assert_equal 0, as_admin { Dlmlock.where(:name => lockname).count }
-        put :acquire, { :id => lockname }
+        assert_equal 0, as_admin { ForemanDlm::Dlmlock.where(:name => lockname).count }
+        put :acquire, params: { :id => lockname }
         assert_response :success
-        dlmlock = as_admin { Dlmlock.find_by(:name => lockname) }
+        dlmlock = as_admin { ForemanDlm::Dlmlock.find_by(:name => lockname) }
         assert_equal lockname, dlmlock.name
         assert_equal host1, dlmlock.host
       end
+
+      test 'should update checkin time' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        assert_not as_admin { host1.dlm_facet }
+        put :acquire, params: { :id => dlmlock.to_param }
+        assert as_admin { host1.reload.dlm_facet.last_checkin_at }
+      end
     end
 
     context '#release' do
       test 'should release empty dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        delete :release, { :id => dlmlock.to_param }
+        delete :release, params: { :id => dlmlock.to_param }
         assert_response :success
         assert_nil as_admin { dlmlock.reload.host }
       end
 
       test 'should release own dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host1) }
-        delete :release, { :id => dlmlock.to_param }
+        delete :release, params: { :id => dlmlock.to_param }
         assert_response :success
         assert_nil as_admin { dlmlock.reload.host }
       end
 
       test 'should not acquire foreign dlmlock' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host2) }
-        delete :release, { :id => dlmlock.to_param }
+        delete :release, params: { :id => dlmlock.to_param }
         assert_response :precondition_failed
         assert_equal host2, as_admin { dlmlock.reload.host }
       end
 
       test 'should transparently create non-existing dlmlock' do
         lockname = 'Test Lock'
-        assert_equal 0, as_admin { Dlmlock.where(:name => lockname).count }
-        delete :release, { :id => lockname }
+        assert_equal 0, as_admin { ForemanDlm::Dlmlock.where(:name => lockname).count }
+        delete :release, params: { :id => lockname }
         assert_response :success
-        dlmlock = as_admin { Dlmlock.find_by(:name => lockname) }
+        dlmlock = as_admin { ForemanDlm::Dlmlock.find_by(:name => lockname) }
         assert_equal lockname, dlmlock.name
         assert_nil dlmlock.host
       end
+
+      test 'should update checkin time' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        assert_not as_admin { host1.dlm_facet }
+        put :release, params: { :id => dlmlock.to_param }
+        assert as_admin { host1.reload.dlm_facet.last_checkin_at }
+      end
     end
   end
 
@@ -295,7 +316,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#index' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        get :index, { :id => dlmlock.to_param }
+        get :index, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
       end
     end
@@ -303,15 +324,15 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#show' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        get :show, { :id => dlmlock.to_param }
+        get :show, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
       end
     end
 
     context '#create' do
       test 'should deny access' do
-        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        post :create, valid_attrs_with_root
+        as_admin { FactoryBot.create(:dlmlock) }
+        post :create, params: valid_attrs_with_root
         assert_response :unauthorized
       end
     end
@@ -319,7 +340,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#update' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        put :update, { :id => dlmlock.to_param, :dlmlock => valid_attrs }
+        put :update, params: { :id => dlmlock.to_param, :dlmlock => valid_attrs }
         assert_response :unauthorized
       end
     end
@@ -327,16 +348,16 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#destroy' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        delete :destroy, { :id => dlmlock.to_param }
+        delete :destroy, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
-        assert_equal 1, as_admin { Dlmlock.where(:id => dlmlock.id).count }
+        assert_equal 1, as_admin { ForemanDlm::Dlmlock.where(:id => dlmlock.id).count }
       end
     end
 
     context '#acquire' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock) }
-        put :acquire, { :id => dlmlock.to_param }
+        put :acquire, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
         assert_nil as_admin { dlmlock.reload.host }
       end
@@ -345,7 +366,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
     context '#release' do
       test 'should deny access' do
         dlmlock = as_admin { FactoryBot.create(:dlmlock, :host => host2) }
-        delete :release, { :id => dlmlock.to_param }
+        delete :release, params: { :id => dlmlock.to_param }
         assert_response :unauthorized
         assert_equal host2, as_admin { dlmlock.reload.host }
       end
@@ -357,7 +378,7 @@ class Api::V2::DlmlocksControllerTest < ActionController::TestCase
   def valid_attrs
     {
       'name' => 'testlock',
-      'type' => 'Dlmlock::Update'
+      'type' => 'ForemanDlm::Dlmlock::Update'
     }
   end
 

data/test/controllers/api/v2/dlmlocks_dlmlock_events_controller_test.rb

@@ -0,0 +1,81 @@
+require 'test_plugin_helper'
+
+class Api::V2::DlmlockEventsControllerTest < ActionController::TestCase
+  let(:host) { as_admin { FactoryBot.create(:host, :managed) } }
+
+  context 'with user authentication' do
+    context '#index' do
+      test 'should return the dlmlock_events for a given dlm_lock' do
+        # A random dlmlock with events, that should not be in the response.
+        dlmlock1 = FactoryBot.create(:dlmlock, host: host)
+        FactoryBot.create_list(:dlmlock_event, 2, dlmlock: dlmlock1)
+
+        dlmlock = FactoryBot.create(:dlmlock, host: host)
+        FactoryBot.create_list(:dlmlock_event, 3, dlmlock: dlmlock)
+
+        expected_size = dlmlock.dlmlock_events.count
+        expected_ids = dlmlock.dlmlock_events.pluck(:id).sort
+        expected_keys = ['id', 'event_type', 'created_at', 'updated_at']
+
+        get :index, params: { dlmlock_id: dlmlock.id }
+        assert_response :success
+
+        body = ActiveSupport::JSON.decode(@response.body)
+        assert_equal expected_size, body['total']
+        assert_equal expected_keys, body['results'].first.keys
+        assert_equal expected_ids, body['results'].map { |event| event['id'] }.sort
+      end
+    end
+  end
+
+  context 'as user with required permissions' do
+    let(:permissions) { Permission.where(name: ['view_dlmlocks', 'view_dlmlock_events']) }
+    let(:role) { FactoryBot.create(:role, permissions: permissions) }
+    let(:user) { FactoryBot.create(:user, roles: [role]) }
+
+    setup do
+      User.current = user
+      reset_api_credentials
+    end
+
+    context '#index' do
+      test 'should allow access' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        get :index, params: { dlmlock_id: dlmlock.id }
+        assert_response :success
+      end
+    end
+  end
+
+  context 'as user without required permissions' do
+    let(:user) { FactoryBot.create(:user, roles: []) }
+
+    setup do
+      User.current = user
+      reset_api_credentials
+    end
+
+    context '#index' do
+      test 'should deny access' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        get :index, params: { dlmlock_id: dlmlock.id }
+        assert_response :forbidden
+      end
+    end
+  end
+
+  context 'without any credentials' do
+    setup do
+      User.current = nil
+      reset_api_credentials
+    end
+
+    context '#index' do
+      test 'should deny access' do
+        dlmlock = as_admin { FactoryBot.create(:dlmlock) }
+        get :index, params: { dlmlock_id: dlmlock.id }
+        assert_response :unauthorized
+      end
+    end
+  end
+end

data/test/controllers/api/v2/hosts_controller_test.rb

@@ -0,0 +1,28 @@
+require 'test_plugin_helper'
+
+class Api::V2::HostsControllerTest < ActionController::TestCase
+  let(:host) { FactoryBot.create(:host, :managed, :with_dlm_facet) }
+
+  test 'should get index with dlm_facet attributes' do
+    host
+    get :index
+    assert_response :success
+    results = ActiveSupport::JSON.decode(@response.body)
+    hosts = results['results']
+    assert_not_empty hosts
+    host_with_facet = hosts.detect { |h| h['dlm_facet_attributes'].present? }
+    assert_not_nil host_with_facet
+    assert_equal host.dlm_facet.id, host_with_facet['dlm_facet_attributes']['id']
+    assert_equal host.dlm_facet.last_checkin_at.to_s, host_with_facet['dlm_facet_attributes']['last_checkin_at']
+  end
+
+  test 'should show individual record with dlm_facet attributes' do
+    get :show, params: { :id => host.to_param }
+    assert_response :success
+    show_response = ActiveSupport::JSON.decode(@response.body)
+    assert_not_empty show_response
+    assert_includes show_response.keys, 'dlm_facet_attributes'
+    assert_equal host.dlm_facet.id, show_response['dlm_facet_attributes']['id']
+    assert_equal host.dlm_facet.last_checkin_at.to_s, show_response['dlm_facet_attributes']['last_checkin_at']
+  end
+end

data/test/controllers/find_host_by_client_cert_test.rb

@@ -1,7 +1,7 @@
 require 'test_plugin_helper'
 
 class FindHostByClientCertTest < ActionController::TestCase
-  tests "api/v2/dlmlocks"
+  tests 'api/v2/dlmlocks'
 
   def described_class
     Api::V2::DlmlocksController
@@ -57,7 +57,7 @@ class FindHostByClientCertTest < ActionController::TestCase
 
       test 'certificate with unknown dn denies access' do
         @request.env['HTTPS'] = 'on'
-        @request.env['SSL_CLIENT_S_DN'] = "CN=doesnotexist.example.com,DN=example,DN=com"
+        @request.env['SSL_CLIENT_S_DN'] = 'CN=doesnotexist.example.com,DN=example,DN=com'
         @request.env['SSL_CLIENT_VERIFY'] = 'SUCCESS'
 
         get :index

data/test/controllers/foreman_dlm/dlmlocks_test.rb

@@ -0,0 +1,55 @@
+require 'test_plugin_helper'
+
+module ForemanDlm
+  class DlmlocksControllerTest < ActionController::TestCase
+    test '#index' do
+      FactoryBot.create(:dlmlock)
+      get :index, session: set_session_user
+      assert_response :success
+      assert_not_nil assigns('dlmlocks')
+      assert_template 'index'
+    end
+
+    test '#index with no lock shows welcome page' do
+      get :index, session: set_session_user
+      assert_response :success
+      assert_template 'welcome'
+    end
+
+    test '#show' do
+      dlmlock = FactoryBot.create(:dlmlock)
+      get :show, params: { :id => dlmlock.id }, session: set_session_user
+      assert_response :success
+      assert_template 'show'
+    end
+
+    test '#destroy' do
+      dlmlock = FactoryBot.create(:dlmlock)
+      delete :destroy, params: { :id => dlmlock.id }, session: set_session_user
+      assert_redirected_to foreman_dlm_dlmlocks_url
+      assert_not Dlmlock.exists?(dlmlock.id)
+    end
+
+    test '#enable' do
+      dlmlock = FactoryBot.create(:dlmlock, enabled: false)
+      put :enable, params: { :id => dlmlock.id }, session: set_session_user
+      assert_redirected_to foreman_dlm_dlmlocks_url
+      assert dlmlock.reload.enabled?
+    end
+
+    test '#disable' do
+      dlmlock = FactoryBot.create(:dlmlock, enabled: true)
+      put :disable, params: { :id => dlmlock.id }, session: set_session_user
+      assert_redirected_to foreman_dlm_dlmlocks_url
+      assert dlmlock.reload.disabled?
+    end
+
+    test '#release' do
+      host = FactoryBot.create(:host)
+      dlmlock = FactoryBot.create(:dlmlock, host: host)
+      put :release, params: { :id => dlmlock.id }, session: set_session_user
+      assert_redirected_to foreman_dlm_dlmlocks_url
+      assert_not dlmlock.reload.taken?
+    end
+  end
+end

data/test/controllers/hosts_controller_test.rb

@@ -0,0 +1,12 @@
+require 'test_plugin_helper'
+
+class HostsControllerTest < ActionController::TestCase
+  let(:host) { FactoryBot.create(:host, :with_dlm_facet) }
+
+  test '#show shows dlm locks of that host' do
+    FactoryBot.create_list(:dlmlock, 2, host: host)
+    get :show, params: { :id => host.to_param }, session: set_session_user
+    assert_response :success
+    assert @response.body.match(/id='pagelet-id-locks'/)
+  end
+end

data/test/factories/dlm_facets.rb

@@ -0,0 +1,6 @@
+FactoryBot.define do
+  factory :dlm_facet, class: 'ForemanDlm::DlmFacet' do
+    sequence(:last_checkin_at) { |n| n.minutes.ago }
+    host
+  end
+end

data/test/factories/dlmlock.rb

@@ -1,6 +1,10 @@
 FactoryBot.define do
-  factory :dlmlock do
+  factory :dlmlock, class: ::ForemanDlm::Dlmlock::Update do
     sequence(:name) { |n| "Lock #{n}" }
-    type 'Dlmlock::Update'
+    type { 'ForemanDlm::Dlmlock::Update' }
+
+    trait :locked do
+      host
+    end
   end
 end

data/test/factories/dlmlock_events.rb

@@ -0,0 +1,13 @@
+FactoryBot.define do
+  factory :dlmlock_event, class: ::ForemanDlm::DlmlockEvent do
+    dlmlock
+    event_type { 'release' }
+    host
+
+    trait :old_event do
+      after(:build) do |event|
+        event.created_at = 2.weeks.ago
+      end
+    end
+  end
+end

data/test/factories/host.rb

@@ -0,0 +1,7 @@
+FactoryBot.modify do
+  factory :host do
+    trait :with_dlm_facet do
+      association :dlm_facet, :factory => :dlm_facet, :strategy => :build
+    end
+  end
+end

data/test/integration/foreman_dlm/dlmlocks_test.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'test_plugin_helper'
+require 'integration_test_helper'
+
+class DlmlocksTest < IntegrationTestWithJavascript
+  setup do
+    User.current = users(:admin)
+  end
+
+  test 'the index page works' do
+    FactoryBot.create_list(:dlmlock, 10)
+    assert_index_page(foreman_dlm_dlmlocks_path, 'Distributed Locks')
+  end
+
+  test 'the search bar has autocomplete' do
+    skip if Gem::Version.new(Foreman::Version.new.notag) < Gem::Version.new('1.20')
+    FactoryBot.create_list(:dlmlock, 10)
+
+    visit foreman_dlm_dlmlocks_path
+
+    search_bar = page.first('#search-bar')
+    search_bar.first('input').set('ho')
+
+    list = search_bar.find_all('a.dropdown-item').map(&:text)
+    assert_includes list, 'host'
+  end
+end

data/test/jobs/refresh_dlmlock_status_test.rb

@@ -0,0 +1,10 @@
+require 'test_plugin_helper'
+
+class RefreshDlmlockStatusTest < ActiveJob::TestCase
+  let(:host) { FactoryBot.create(:host, :managed) }
+
+  test 'should refresh dlmlock status' do
+    Host::Managed.any_instance.expects(:refresh_dlmlock_status).once
+    perform_enqueued_jobs { ForemanDlm::RefreshDlmlockStatus.perform_later(host.id) }
+  end
+end