Merge pull request #239 from psu-libraries/add-unit-to-admin

Add unit to admin

Author: Smullz622

app/models/job.rb

@@ -23,6 +23,7 @@ def completed?
         field :status
         field :owner_type
         field :owner_id
+        field :page_count
         field :created_at
         field :finished_at
       end
@@ -31,6 +32,7 @@ def completed?
         field :uuid
         field :type
         field :status
+        field :page_count
         field :llm_model
         field :prompt
         field :alt_text

app/models/rails_admin.rb

@@ -4,6 +4,7 @@
   config.included_models = [
     'GuiUser',
     'ApiUser',
-    'Job'
+    'Job',
+    'Unit'
   ]
 end

app/models/unit.rb

@@ -7,4 +7,28 @@ class Unit < ApplicationRecord
   validates :name, presence: true
   validates :daily_page_limit, presence: true, numericality: { only_integer: true, greater_than_or_equal_to: 0 }
   validates :overall_page_limit, presence: true, numericality: { only_integer: true, greater_than_or_equal_to: 0 }
+
+  RailsAdmin.config do |config|
+    config.model 'Unit' do
+      list do
+        field :id
+        field :name
+        field :daily_page_limit
+        field :overall_page_limit
+        field :created_at
+      end
+
+      edit do
+        field :name
+        field :daily_page_limit
+        field :overall_page_limit
+        field :created_at do
+          read_only true
+        end
+        field :updated_at do
+          read_only true
+        end
+      end
+    end
+  end
 end

spec/requests/admin/rails_admin_spec.rb

@@ -33,6 +33,11 @@
       get '/admin/job', headers: admin_headers
       expect(response).to have_http_status(:ok)
     end
+
+    it 'allows access to Unit model' do
+      get '/admin/unit', headers: admin_headers
+      expect(response).to have_http_status(:ok)
+    end
   end
 
   context 'when user is not an admin' do