[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,18 @@
+# frozen_string_literal: true
+
+module ActiveResource
+  module Associations
+    module ActiveRecord
+      def self.included(base)
+        base.extend(ClassMethods)
+      end
+
+      module ClassMethods
+        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
+end

lib/active_resource/railtie.rb

@@ -9,16 +9,22 @@ class Railtie < Rails::Railtie
 
     initializer "active_resource.set_configs" do |app|
       ActiveSupport.on_load(:active_resource) do
-        app.config.active_resource.each do |k, v|
-          send "#{k}=", v
-        end
+        app.config.active_resource.each { |k, v| send "#{k}=", v }
       end
     end
 
     initializer "active_resource.add_active_job_serializer" do |app|
       if app.config.try(:active_job).try(:custom_serializers)
         require "active_resource/active_job_serializer"
-        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.send(
+          :include,
+          ActiveResource::Associations::ActiveRecord
+        )
       end
     end
   end

test/cases/active_record_association_test.rb

@@ -0,0 +1,40 @@
+# 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
+  end
+
+  class TestRecord < ActiveRecord::Base
+    include ActiveResource::Associations::ActiveRecord
+    belongs_to_resource :person
+    belongs_to_resource :book, class_name: "Product"
+  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