sorbet-rails 0.7.4 → 0.7.25

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 33c9dda20b44067f6f788873f626fd9d2fdf5bbe8a62d8c374e2748b637f5993
-  data.tar.gz: e2d65d1ec7c2e03d7135d7de8540b859c6348d27382732893fff611a5b627d88
+  metadata.gz: 99ef9d97be5d8b7c9ce91c26827337bb0cf170a704d0e1c025a1a47964b8f0cd
+  data.tar.gz: 831bc3d6fc4aafd8e0f4ea315369a410b3dbd90669a8bbd5b5f059fe9ce6543b
 SHA512:
-  metadata.gz: 86a4cc741bef3972a5a8cb3a77d1eaa1f1fe4924d816c5ba681bf6fe5b5d91d0b6e94410a2a08fe15b8d696b80e50b7ff7f4c5a049e58a841fa9d2ecd83fd9c0
-  data.tar.gz: 6120f8d8fdf6e631f92cf6428774cd6df1a410d64282d96ed4921202c1715e2211e15823a510234849615cb7e3dc08368121fd663ad1ab9148ad337123799dca
+  metadata.gz: 7309e5f22a4f7979ff5665c0d17d2c22bf823d9bbb4d6468668d6df0ad5b6e4243a300974e5b289c8314adbfc40f04d889a3d4a69a1d932b443d5ae688a11256
+  data.tar.gz: '06496c5b1cf2b7e688fb100cba9d3dfae8baded69712b0a8b2e4adfbef27ec93dca88205813776c9905329ac17e118bb871c19a2d6abfbaff1dce006c27eda3d'

data/.github/dependabot.yml

@@ -0,0 +1,7 @@
+version: 2
+updates:
+  - package-ecosystem: "bundler"
+    # Files stored in repository root
+    directory: "/"
+    schedule:
+      interval: "daily"

data/.github/workflows/ci-master.yml

@@ -0,0 +1,28 @@
+name: ci-master
+
+on:
+  push:
+    branches: # Trigger only on the main branch to avoid duplicate runs on PR branches
+      - master
+
+jobs:
+  rspec:
+    runs-on: ubuntu-latest
+
+    strategy:
+      fail-fast: false
+      matrix:
+        ruby: [2.5, 2.6, 2.7]
+        rails: [5.2, 6.0, 6.1]
+
+    env:
+      RAILS_VERSION: ${{ matrix.rails }}
+      CI: true
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: ruby/setup-ruby@v1
+      with:
+        ruby-version: ${{ matrix.ruby }}
+        # bundler-cache: true -- this messes up some dynamic constants
+    - run: "./spec/bin/run_spec.sh"

data/.github/workflows/ci.yml

@@ -0,0 +1,26 @@
+name: ci
+
+on:
+  pull_request: # Trigger All PRs
+
+jobs:
+  rspec:
+    runs-on: ubuntu-latest
+
+    strategy:
+      fail-fast: false
+      matrix:
+        ruby: [2.5, 2.6, 2.7]
+        rails: [5.2, 6.0, 6.1]
+
+    env:
+      RAILS_VERSION: ${{ matrix.rails }}
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: ruby/setup-ruby@v1
+      with:
+        ruby-version: ${{ matrix.ruby }}
+        # bundler-cache: true -- this messes up some dynamic constants
+    - run: "./spec/bin/run_spec.sh"
+    - run: bundle exec srb tc

data/.github/workflows/publish-gem.yml

@@ -0,0 +1,27 @@
+name: publish-gem
+
+on:
+  # pull_request: # test with pull request
+  push:
+    branches: # Trigger only on the main branch to avoid duplicate runs on PR branches
+      - master
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    env:
+      RAILS_VERSION: 6.0
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: ruby/setup-ruby@v1
+      with:
+        ruby-version: 2.7
+        # bundler-cache: true -- this messes up some dynamic constants
+    - run: "./spec/bin/run_spec.sh"
+    - run: bundle exec srb tc
+    - name: Release Gem
+      uses: dawidd6/action-publish-gem@v1
+      with:
+        api_key: ${{secrets.RUBYGEMS_API_KEY}}

data/CONTRIBUTING.md

@@ -87,10 +87,9 @@ To run the tests against all supported Rails branches, use:
 You can also switch to a version of Rails with `RAILS_VERSION`:
 
 ```sh
-❯ RAILS_VERSION=5.0 ./spec/bin/run_spec.sh
-❯ RAILS_VERSION=5.1 ./spec/bin/run_spec.sh
 ❯ RAILS_VERSION=5.2 ./spec/bin/run_spec.sh
 ❯ RAILS_VERSION=6.0 ./spec/bin/run_spec.sh
+❯ RAILS_VERSION=6.1 ./spec/bin/run_spec.sh
 ```
 
 #### Debugging

data/Gemfile

@@ -13,14 +13,6 @@ case rails_version
 when "master"
   gem "rails", {github: "rails/rails"}
   gem "bundler", ">= 1.3.0"
-when "5.0"
-  gem "rails", "~> 5.0.7"
-  gem "bundler", ">= 2.0"
-  gem "sqlite3", "~> 1.3.6"
-when "5.1"
-  gem "rails", "~> 5.1.7"
-  gem "bundler", ">= 2.0"
-  gem "sqlite3", "~> 1.4.1"
 when "5.2"
   gem "rails", "~> 5.2.3"
   gem "bundler", ">= 2.0"
@@ -29,8 +21,12 @@ when "6.0"
   gem "rails", "~> 6.0.0"
   gem "bundler", ">= 2.0"
   gem "sqlite3", "~> 1.4.1"
+when "6.1"
+  gem "rails", "~> 6.1.0"
+  gem "bundler", ">= 2.0"
+  gem "sqlite3", "~> 1.4.1"
 else
-  gem "rails", "~> 5.2.3"
+  gem "rails", "~> 6.0.0"
   gem "bundler", ">= 2.0"
   gem "sqlite3", "~> 1.4.1"
 end

data/README.md

@@ -1,6 +1,6 @@
 # sorbet-rails
 [![Gem Version](https://badge.fury.io/rb/sorbet-rails.svg)](https://badge.fury.io/rb/sorbet-rails)
-[![Build Status](https://travis-ci.com/chanzuckerberg/sorbet-rails.svg?branch=master)](https://travis-ci.com/chanzuckerberg/sorbet-rails)
+[![Build Status](https://github.com/chanzuckerberg/sorbet-rails/actions/workflows/ci-master.yml/badge.svg)](https://github.com/chanzuckerberg/sorbet-rails)
 [![codecov](https://codecov.io/gh/chanzuckerberg/sorbet-rails/branch/master/graph/badge.svg)](https://codecov.io/gh/chanzuckerberg/sorbet-rails)
 
 A set of tools to make the [Sorbet](https://sorbet.org) typechecker work with Ruby on Rails seamlessly.
@@ -9,6 +9,8 @@ This gem adds a few Rake tasks to generate Ruby Interface (RBI) files for dynami
 
 `sorbet-rails` supports Rails 5+ or later.
 
+**Notice**: we automatically creates a new release on Rubygems with every merges to the master branch
+
 **Notice**: we no longer support Rails 4.2. [Version 0.5.6](https://github.com/chanzuckerberg/sorbet-rails/releases/tag/v0.5.6) is the last version supporting Rails 4.2.
 
 ## Initial Setup
@@ -206,7 +208,7 @@ end
 ❯ bundle exec  rake rails_rbi:custom
 ```
 
-`sorbet-rails` adds `TypedParams` to extact typed controller parameters.
+`sorbet-rails` adds `TypedParams` to extract typed controller parameters.
 
 TypedParams takes a parameter definition, which is a subclass of `T::Struct` and coerces the `params` object into an instance of that subclass using [sorbet-coerce](https://github.com/chanzuckerberg/sorbet-coerce):
 ```ruby
@@ -311,7 +313,7 @@ It is worth going through the list of files that is ignored and resolve them (an
 
 ### Overriding generated signatures if needed
 
-`sorbet-rails` relies on Rails reflection to generate signatures. There are features this gem doesn't support yet such as [serialize](https://github.com/chanzuckerberg/sorbet-rails/issues/49) and [attribute custom types](https://github.com/chanzuckerberg/sorbet-rails/issues/16). The gem also doesn't know the signature of any methods you have overridden. However, it is possible to override the signatures that `sorbet-rails` generates.
+`sorbet-rails` relies on Rails reflection to generate signatures. There are features this gem doesn't support yet such as [attribute custom types](https://github.com/chanzuckerberg/sorbet-rails/issues/16). The gem also doesn't know the signature of any methods you have overridden. However, it is possible to override the signatures that `sorbet-rails` generates.
 
 For example, here is how to override the signature for a method in a model:
 
@@ -409,8 +411,16 @@ Model.unscoped.scoping do … end
 
 ### `select` with a block
 
-The [`select` method](https://apidock.com/rails/v4.0.2/ActiveRecord/QueryMethods/select) in Rails has two modes: it can be given a list of symbols, in which case rails will only return the given columns from the database, or it can be given a block, in which case it acts like [`Enumerable.select`](https://ruby-doc.org/core-2.6.4/Enumerable.html) and returns an array. We have chosen to support the first use case. If you want to pass a block to `select`, you can simply call `to_a` before you do. Note that this would be done within the `select` call anyway, so the performance penalty will be minimal.
+The [`select` method](https://apidock.com/rails/v4.0.2/ActiveRecord/QueryMethods/select) in Rails has two modes: it can be given a list of symbols, in which case rails will only return the given columns from the database, or it can be given a block, in which case it acts like [`Enumerable.select`](https://ruby-doc.org/core-2.6.4/Enumerable.html) and returns an array.
+
+We have chosen to support `select` with a block as the primary use case, since it is a more prevalent use of this method. We provide `select_columns` as an alternative if you want to select a list of columns.
 
+```
+Model.select { |record| record.id > 1} # valid, returns an array
+Model.select_columns(:id, :name) # valid, returns an association
+
+Model.select(:id, :name) # sorbet error, use `select_column` instead
+```
 ### `flatten` an array of relation
 
 When you call `flatten` on an array of ActiveRecord::Relation, sorbet [doesn't recognize](https://github.com/sorbet/sorbet/issues/2767) that it will flatten the relation and return an array of model. The work around is to call `to_a` on the relation first.
@@ -524,6 +534,8 @@ These are the currently-supported gems and their symbolized names:
 | [Paperclip]  | `:paperclip` |
 | [AttrJson]  | `:attr_json` |
 | [FlagShihTzu]  | `:flag_shih_tzu` |
+| [AASM]  | `:aasm` |
+| [Money-Rails]  | `:money_rails` |
 
 You can also configure the core model plugins if needed. The default plugins are defined in the [config](https://github.com/chanzuckerberg/sorbet-rails/blob/master/lib/sorbet-rails/config.rb). For the full list of plugin symbols, check out [here](https://github.com/chanzuckerberg/sorbet-rails/blob/master/lib/sorbet-rails/model_plugins/plugins.rb).
 
@@ -537,6 +549,8 @@ You can also configure the core model plugins if needed. The default plugins are
 [Paperclip]: https://github.com/thoughtbot/paperclip
 [AttrJson]: https://github.com/jrochkind/attr_json
 [FlagShihTzu]: https://github.com/pboling/flag_shih_tzu
+[AASM]: https://github.com/aasm/aasm
+[Money-Rails]: https://github.com/RubyMoney/money-rails
 
 ### Customize Generation Class
 

data/Rakefile

@@ -10,13 +10,22 @@ namespace :update_spec do
 
   desc "Generate Rails apps for all versions."
   task :all do |t, args|
+    Rake::Task['update_spec:v6_1'].invoke
     Rake::Task['update_spec:v6_0'].invoke
     Rake::Task['update_spec:v5_2'].invoke
-    Rake::Task['update_spec:v5_1'].invoke
-    Rake::Task['update_spec:v5_0'].invoke
   end
 
-  desc "Delete the Rails 6 spec directory and regenerate it."
+  desc "Delete the Rails 6.1 spec directory and regenerate it."
+  task :v6_1 do |t, args|
+    Bundler.with_clean_env do
+      FileUtils.rm_rf 'spec/support/v6.1' if File.directory?('spec/support/v6.1')
+      system("gem install rails -v 6.1.4")
+      system("rails _6.1.4_ -v")
+      system("RAILS_VERSION='6.1' rails _6.1.4_ new --template spec/generators/rails-template.rb spec/support/v6.0 --skip-javascript --skip-action-cable --skip-test --skip-sprockets --skip-spring --skip-bootsnap --skip-listen")
+    end
+  end
+
+  desc "Delete the Rails 6.0 spec directory and regenerate it."
   task :v6_0 do |t, args|
     Bundler.with_clean_env do
       FileUtils.rm_rf 'spec/support/v6.0' if File.directory?('spec/support/v6.0')
@@ -36,29 +45,9 @@ namespace :update_spec do
     end
   end
 
-  desc "Delete the Rails 5.1 spec directory and regenerate it."
-  task :v5_1 do |t, args|
-    Bundler.with_clean_env do
-      FileUtils.rm_rf 'spec/support/v5.1' if File.directory?('spec/support/v5.1')
-      system("gem install rails -v 5.1.7")
-      system("rails _5.1.7_ -v")
-      system("RAILS_VERSION='5.1' rails _5.1.7_ new --template spec/generators/rails-template.rb spec/support/v5.1 --skip-javascript --skip-action-cable --skip-test --skip-sprockets --skip-spring --skip-listen")
-    end
-  end
-
-  desc "Delete the Rails 5.0 spec directory and regenerate it."
-  task :v5_0 do |t, args|
-    Bundler.with_clean_env do
-      FileUtils.rm_rf 'spec/support/v5.0' if File.directory?('spec/support/v5.0')
-      system("gem install rails -v 5.0.7")
-      system("rails _5.0.7_ -v")
-      system("RAILS_VERSION='5.0' rails _5.0.7_ new --template spec/generators/rails-template.rb spec/support/v5.0 --skip-javascript --skip-action-cable --skip-test --skip-sprockets --skip-spring --skip-listen")
-    end
-  end
-
   desc "Update sorbet_test_cases.rb in all the Rails apps in spec/support."
   task :sorbet_test_cases do |t, args|
-    ['v6.0', 'v5.2', 'v5.1', 'v5.0'].each do |version|
+    ['v6.1', 'v6.0', 'v5.2'].each do |version|
       FileUtils.cp("spec/generators/sorbet_test_cases.rb", "spec/support/#{version}/sorbet_test_cases.rb")
     end
   end

data/lib/sorbet-rails/active_record_rbi_formatter.rb

@@ -273,6 +273,18 @@ class SorbetRails::ActiveRecordRbiFormatter
         class_method: class_method,
       )
     end
+
+    boolean_methods = %w(any? many? none? one?)
+    boolean_methods.each do |boolean_method|
+      class_rbi.create_method(
+        boolean_method,
+        parameters: [
+          Parameter.new("&block", type: "T.nilable(T.proc.params(e: #{type}).returns(T.nilable(T::Boolean)))"),
+        ],
+        return_type: "T::Boolean",
+        class_method: class_method,
+      )
+    end
   end
 
   sig {
@@ -288,14 +300,5 @@ class SorbetRails::ActiveRecordRbiFormatter
       return_type: "T::Boolean",
       class_method: class_method,
     )
-
-    boolean_methods = %w(any? many? none? one?)
-    boolean_methods.each do |boolean_method|
-      class_rbi.create_method(
-        boolean_method,
-        return_type: "T::Boolean",
-        class_method: class_method,
-      )
-    end
   end
 end

data/lib/sorbet-rails/dependent_gem_rbis/activerecord.rbi

@@ -42,3 +42,7 @@ module ActiveModel::Validations
     def validators_on(*attributes); end
   end
 end
+
+class ActiveRecord::Type::Serialized
+  def coder; end
+end

data/lib/sorbet-rails/dependent_gem_rbis/parlour.rbi

@@ -1,4 +1,4 @@
-# typed: strong
+# typed: ignore
 module Kernel
 end
 

data/lib/sorbet-rails/gem_plugins/aasm_plugin.rb

@@ -0,0 +1,50 @@
+# typed: false
+class AasmPlugin < SorbetRails::ModelPlugins::Base
+  sig { override.params(root: Parlour::RbiGenerator::Namespace).void }
+  def generate(root)
+    return unless @model_class.include?(::AASM)
+
+    model_rbi = root.create_class(
+      model_class_name
+    )
+
+    aasm_events = @model_class.aasm.events.map(&:name)
+    aasm_states = @model_class.aasm.states.map(&:name)
+
+    # If you have an event like :bazify, you get these methods:
+    # - `may_bazify?`
+    # - `bazify`
+    # - `bazify!`
+    # - `bazify_without_validation!`
+    aasm_events.each do |event|
+      model_rbi.create_method(
+        "may_#{event}?",
+        return_type: 'T::Boolean'
+      )
+
+      model_rbi.create_method(
+        event.to_s,
+        return_type: 'T::Boolean'
+      )
+
+      model_rbi.create_method(
+        "#{event}!",
+        return_type: 'T::Boolean'
+      )
+
+      model_rbi.create_method(
+        "#{event}_without_validation!",
+        return_type: 'T::Boolean'
+      )
+    end
+
+    # - If you have a state like :baz, you get these methods:
+    # - `baz?`
+    aasm_states.each do |state|
+      model_rbi.create_method(
+        "#{state}?",
+        return_type: 'T::Boolean'
+      )
+    end
+  end
+end

data/lib/sorbet-rails/gem_plugins/money_rails_plugin.rb

@@ -0,0 +1,76 @@
+# typed: strict
+class MoneyRailsPlugin < SorbetRails::ModelPlugins::Base
+  sig { params(attribute: T.any(Symbol, String)).returns(T::Boolean) }
+  def allows_nil(attribute)
+    validators = model_class.validators_on(attribute)
+    return false if validators.empty?
+
+    validators.any? do |validator|
+      validator.respond_to?(:options) &&
+        validator.options[:allow_nil]
+    end
+  end
+
+  sig { override.params(root: Parlour::RbiGenerator::Namespace).void }
+  def generate(root)
+    return unless exists_class_method?(:monetize)
+
+    # money-rails gives us a hash mapping the monetized attribute to its subunit
+    # equivalent, eg: `{"price": "price_cents"}`
+    monetized_attributes = T.let(
+      T.unsafe(model_class).monetized_attributes,
+      T::Hash[String,String]
+    )
+
+    return if monetized_attributes.empty?
+
+    module_name = model_module_name("GeneratedMoneyRailsMethods")
+    module_rbi = root.create_module(module_name)
+
+    model_class_rbi = root.create_class(model_class_name)
+    model_class_rbi.create_include(module_name)
+
+    monetized_attributes.each do |name, subunit_name|
+      validators = model_class.validators_on(name)
+
+      money_attribute_type = allows_nil(name) ? "T.nilable(::Money)" : "::Money"
+      subunit_attribute_type = allows_nil(name) ? "T.nilable(Integer)" : "Integer"
+
+      module_rbi.create_method(
+        name,
+        return_type: money_attribute_type
+      )
+
+      module_rbi.create_method(
+        subunit_name,
+        return_type: subunit_attribute_type
+      )
+
+      module_rbi.create_method(
+        "#{name}_money_before_type_cast",
+        return_type: 'T.untyped'
+      )
+
+      module_rbi.create_method(
+        "#{name}=",
+        parameters:  [
+           Parlour::RbiGenerator::Parameter.new("value", type: money_attribute_type)
+        ],
+        return_type: nil
+      )
+
+      module_rbi.create_method(
+        "#{subunit_name}=",
+        parameters:  [
+           Parlour::RbiGenerator::Parameter.new("value", type: subunit_attribute_type)
+        ],
+        return_type: nil
+      )
+
+      module_rbi.create_method(
+        "currency_for_#{name}",
+        return_type: "T.nilable(::Money::Currency)"
+      )
+    end
+  end
+end

data/lib/sorbet-rails/job_rbi_formatter.rb

@@ -80,7 +80,7 @@ module SorbetRails
 
     sig { returns(String) }
     def generate_rbi
-      puts "-- Generate sigs for mailer #{@job_class.name} --"
+      puts "-- Generate sigs for ActiveJob #{@job_class.name} --"
       populate_rbi
       @rbi_generator.rbi
     end

data/lib/sorbet-rails/model_plugins/active_record_querying.rb

@@ -25,7 +25,8 @@ class SorbetRails::ModelPlugins::ActiveRecordQuerying < SorbetRails::ModelPlugin
     # <Model>::Relation class.
     # rails/activerecord/lib/active_record/querying.rb
     model_query_relation_methods = [
-      :select, :reselect, :order, :reorder, :group, :limit, :offset, :joins, :left_joins, :left_outer_joins,
+      # :select,
+      :reselect, :order, :reorder, :group, :limit, :offset, :joins, :left_joins, :left_outer_joins,
       :where, :rewhere, :preload, :extract_associated, :eager_load, :includes, :from, :lock, :readonly, :or,
       :having, :create_with, :distinct, :references, :none, :unscope, :optimizer_hints, :merge, :except, :only,
     ]
@@ -39,6 +40,29 @@ class SorbetRails::ModelPlugins::ActiveRecordQuerying < SorbetRails::ModelPlugin
         builtin_query_method: true,
       ) if exists_class_method?(method_name)
     end
+    if exists_class_method?("select")
+      # `select` can be used with a block to return an array, or with a list of column name
+      # to return a relation object that can be chained.
+      # I've seen usage of `select` with a block more often than with a list of column name.
+      # here we define select as taking a block, and add a `select_column` method for the other usage
+      add_relation_query_method(
+        root,
+        "select",
+        parameters: [
+          Parameter.new("&block", type: "T.proc.params(e: #{self.model_class_name}).returns(T::Boolean)"),
+        ],
+        builtin_query_method: true,
+        custom_return_value: "T::Array[#{self.model_class_name}]",
+      )
+      add_relation_query_method(
+        root,
+        "select_columns", # select_column is injected by sorbet-rails
+        parameters: [
+          Parameter.new("*args", type: "T.any(String, Symbol, T::Array[T.any(String, Symbol)])"),
+        ],
+        builtin_query_method: true,
+      )
+    end
 
     add_relation_query_method(
       root,

data/lib/sorbet-rails/model_plugins/active_record_serialized_attribute.rb

@@ -60,9 +60,10 @@ class SorbetRails::ModelPlugins::ActiveRecordSerializedAttribute < SorbetRails::
       # ActiveSupport::JSON.encode/ActiveSupport::JSON.decode
       # note that Hash keys are Strings since this is JSON
       'T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[String, T.untyped], Integer, String)'
-    else
-      # unknown uses YAML.load/YAML.dump
+    when 'Object', ''
       'T.any(T::Array[T.untyped], T::Boolean, Float, T::Hash[T.untyped, T.untyped], Integer, String)'
+    else
+      serialization_coder.to_s
     end
   end
 end

data/lib/sorbet-rails/model_plugins/base.rb

@@ -35,7 +35,14 @@ module SorbetRails::ModelPlugins
       column_type = @model_class.type_for_attribute(column_name)
       return unless column_type.is_a?(ActiveRecord::Type::Serialized)
 
-      column_type.coder.try(:object_class) || Object
+      object_class = column_type.coder.try(:object_class)
+      if object_class
+        object_class
+      elsif column_type.coder.is_a? Class
+        column_type.coder
+      else
+        Object
+      end
     end
   end
 end

data/lib/sorbet-rails/model_plugins/custom_finder_methods.rb

@@ -39,7 +39,7 @@ class SorbetRails::ModelPlugins::CustomFinderMethods < SorbetRails::ModelPlugins
     # allow common cases find_by_id
     custom_module_rbi.create_method(
       "find_by_id",
-      parameters: [ Parameter.new("id", type: "Integer") ],
+      parameters: [ Parameter.new("id", type: "T.nilable(Integer)") ],
       return_type: "T.nilable(#{self.model_class_name})",
     )
     custom_module_rbi.create_method(

data/lib/sorbet-rails/model_plugins/plugins.rb

@@ -87,6 +87,12 @@ module SorbetRails::ModelPlugins
     when :flag_shih_tzu
       require('sorbet-rails/gem_plugins/flag_shih_tzu_plugin')
       FlagShihTzuPlugin
+    when :aasm
+      require('sorbet-rails/gem_plugins/aasm_plugin')
+      AasmPlugin
+    when :money_rails
+      require('sorbet-rails/gem_plugins/money_rails_plugin')
+      MoneyRailsPlugin
     else
       raise UnrecognizedPluginName.new(
         "Unrecognized plugin with name: #{plugin_name}. Please check available plugins in the

data/lib/sorbet-rails/model_utils.rb

@@ -90,9 +90,16 @@ module SorbetRails::ModelUtils
       # where, limit, etc and not something like a named scope. It should likely
       # only be set to `true` when called from the ActiveRecordQuerying plugin.
       builtin_query_method: T::Boolean,
+      custom_return_value: T.nilable(String),
     ).void
   }
-  def add_relation_query_method(root, method_name, parameters: nil, builtin_query_method: false)
+  def add_relation_query_method(
+    root,
+    method_name,
+    parameters: nil,
+    builtin_query_method: false,
+    custom_return_value: nil
+  )
     # a relation querying method will be available on
     # - model (as a class method)
     # - activerecord relation
@@ -114,14 +121,14 @@ module SorbetRails::ModelUtils
       relation_module_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: self.model_relation_class_name,
+        return_type: custom_return_value || self.model_relation_class_name,
       )
 
       assoc_relation_module_rbi = root.create_module(self.model_query_methods_returning_assoc_relation_module_name)
       assoc_relation_module_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: assoc_return_value,
+        return_type: custom_return_value || assoc_return_value,
       )
     else
       # force generating these methods because sorbet's hidden-definitions generate & override them
@@ -129,7 +136,7 @@ module SorbetRails::ModelUtils
       model_class_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: self.model_relation_class_name,
+        return_type: custom_return_value || self.model_relation_class_name,
         class_method: true,
       )
 
@@ -137,21 +144,21 @@ module SorbetRails::ModelUtils
       model_relation_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: self.model_relation_class_name,
+        return_type: custom_return_value || self.model_relation_class_name,
       )
 
       model_assoc_relation_rbi = root.create_class(self.model_assoc_relation_class_name)
       model_assoc_relation_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: assoc_return_value,
+        return_type: custom_return_value || assoc_return_value,
       )
 
       collection_proxy_rbi = root.create_class(self.model_assoc_proxy_class_name)
       collection_proxy_rbi.create_method(
         method_name,
         parameters: parameters,
-        return_type: assoc_return_value,
+        return_type: custom_return_value || assoc_return_value,
       )
     end
   end

data/lib/sorbet-rails/rails_mixins/custom_finder_methods.rb

@@ -13,6 +13,10 @@ module SorbetRails
       last(n)
     end
 
+    def select_columns(*args)
+      select(*args)
+    end
+
     # Redeclare these dynamic methods here otherwise they will be generated by
     # `srb rbi hidden-definitions`
     def find_by_id(id)

data/lib/sorbet-rails/railtie.rb

@@ -1,4 +1,4 @@
-# typed: true
+# typed: false
 require "rails"
 require "sorbet-runtime"
 require "sorbet-rails/config"

data/lib/sorbet-rails/routes_rbi_formatter.rb

@@ -39,6 +39,14 @@ class SorbetRails::RoutesRbiFormatter
     @parlour.root.create_class('ActionMailer::Base') do |mod|
       mod.create_include('GeneratedUrlHelpers')
     end
+
+    @parlour.root.create_class('ActionDispatch::IntegrationTest') do |mod|
+      mod.create_include('GeneratedUrlHelpers')
+    end
+
+    @parlour.root.create_class('ActionDispatch::SystemTestCase') do |mod|
+      mod.create_include('GeneratedUrlHelpers')
+    end
   end
 
   sig { params(routes: T.untyped, filter: T.untyped).void }