[Active Record] Add belongs_to_resource to active record

Author: basicBrogrammer

Gemfile

@@ -8,6 +8,8 @@ branch = ENV.fetch("BRANCH", "main")
 gem "activesupport", github: "rails/rails", branch: branch
 gem "activemodel", github: "rails/rails", branch: branch
 gem "activejob", github: "rails/rails", branch: branch
+gem "activerecord", github: "rails/rails", branch: branch
+gem "sqlite3"
 
 gem "rubocop"
 gem "rubocop-minitest"

lib/active_resource/associations/active_record.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module ActiveResource
+  module Associations
+    module ActiveRecord
+      def belongs_to_resource(name, class_name: nil)
+        klass = class_name&.constantize || name.to_s.classify.constantize
+        define_method(name) { klass.find(send("#{name}_id")) }
+      end
+    end
+  end
+end

lib/active_resource/railtie.rb

@@ -21,5 +21,11 @@ class Railtie < Rails::Railtie
         app.config.active_job.custom_serializers << ActiveResource::ActiveJobSerializer
       end
     end
+
+    initializer "active_resource.patch_active_record" do |app|
+      ActiveSupport.on_load(:active_record) do
+        ActiveRecord::Base.extend(ActiveResource::Associations::ActiveRecord)
+      end
+    end
   end
 end

test/cases/active_record_association_test.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "active_record"
+require "active_resource/associations/active_record"
+
+class ActiveRecordAssociationTest < ActiveSupport::TestCase
+  setup do
+    setup_response # find me in abstract_unit
+
+    ActiveRecord::Base.establish_connection(
+      adapter: "sqlite3",
+      database: ":memory:"
+    )
+    ActiveRecord::Schema.define do
+      self.verbose = false
+
+      create_table :test_records, force: true do |t|
+        t.string :name
+        t.belongs_to :person
+        t.belongs_to :book
+        t.timestamps
+      end
+    end
+
+    ActiveRecord::Base.extend(ActiveResource::Associations::ActiveRecord)
+
+    class TestRecord < ActiveRecord::Base
+      belongs_to_resource :person
+      belongs_to_resource :book, class_name: "Product"
+    end
+  end
+
+  def test_belongs_to_resource
+    record = TestRecord.create(name: "test", person_id: 1)
+    assert_equal record.person.name, "Matz"
+  end
+
+  def test_belongs_to_resource_with_class_name
+    record = TestRecord.create(name: "test", person_id: 1, book_id: 1)
+    assert_equal record.book.name, "Rails book"
+  end
+end