Merge pull request #33 from ksss/belongs_to-implicitly_returns_nil

Support annotation `implicitly-returns-nil`
ksss · Jan 31, 2025 · 095dab2 · 095dab2
2 parents 29541de + 607d517
commit 095dab2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lib/orthoses/active_record/belongs_to.rb b/lib/orthoses/active_record/belongs_to.rb
@@ -23,7 +23,7 @@ def call
               opt = "#{type}?"
 
               [
-                "def #{ref.name}: () -> #{opt}",
+                "%a{implicitly-returns-nil} def #{ref.name}: () -> #{type}",
                 "def #{ref.name}=: (#{opt}) -> #{opt}",
                 "def reload_#{ref.name}: () -> #{opt}",
               ].tap do |ary|

diff --git a/lib/orthoses/active_record/belongs_to_test.rb b/lib/orthoses/active_record/belongs_to_test.rb
@@ -1,3 +1,8 @@
+begin
+  require 'test_helper'
+rescue LoadError
+end
+
 module BelongsToTest
   LOADER = ->(){
     class User < ActiveRecord::Base
@@ -40,7 +45,8 @@ class BelongsToTest::Post < ::ActiveRecord::Base
 
     expect = <<~RBS
       module BelongsToTest::Post::GeneratedAssociationMethods
-        def user: () -> BelongsToTest::User?
+        %a{implicitly-returns-nil}
+        def user: () -> BelongsToTest::User
 
         def user=: (BelongsToTest::User?) -> BelongsToTest::User?