Add tests for T::Struct and T::Enum

Shopify · Dec 2, 2021 · c4f90c9 · c4f90c9
1 parent 3f3ff27
commit c4f90c9
Showing 1 changed file with 65 additions and 0 deletions.
diff --git a/test/rbi/rewriters/flatten_scopes_test.rb b/test/rbi/rewriters/flatten_scopes_test.rb
@@ -117,5 +117,70 @@ def m1; end
         module A::B::C::D; end
       RBI
     end
+
+    def test_flatten_scopes_with_t_struct
+      rbi = RBI::Parser.parse_string(<<~RBI)
+        module A
+          class B < T::Struct
+            const :foo, String
+
+            class C
+              module D; end
+            end
+          end
+        end
+
+        module E; end
+      RBI
+
+      rbi.flatten_scopes!
+
+      assert_equal(<<~RBI, rbi.string)
+        module A; end
+        module E; end
+
+        class A::B < T::Struct
+          const :foo, String
+        end
+
+        class A::B::C; end
+        module A::B::C::D; end
+      RBI
+    end
+
+    def test_flatten_scopes_with_t_enum
+      rbi = RBI::Parser.parse_string(<<~RBI)
+        module A
+          class B
+            class C
+              class D < T::Enum
+                enums do
+                  Foo = new
+                  Bar = new
+                end
+              end
+            end
+          end
+        end
+
+        module E; end
+      RBI
+
+      rbi.flatten_scopes!
+
+      assert_equal(<<~RBI, rbi.string)
+        module A; end
+        module E; end
+        class A::B; end
+        class A::B::C; end
+
+        class A::B::C::D < T::Enum
+          enums do
+            Foo = new
+            Bar = new
+          end
+        end
+      RBI
+    end
   end
 end