ModelAttribute: Add `.classes` method

This allows people to get a list of the classes defined in
`ModelAttribute`. The classes are turned into constant-cased strings.
Not sure if that's useful, but this is how it was described in the
ticket.

Refs #4401

1 files changed, 16 insertions, 0 deletions

diff --git a/spec/lib/model_attribute_spec.rb b/spec/lib/model_attribute_spec.rb
index faea0b483..9e10bf40f 100644
--- a/spec/lib/model_attribute_spec.rb
+++ b/spec/lib/model_attribute_spec.rb
@@ -14,6 +14,22 @@ RSpec.describe ModelAttribute do
     end
   end
 
+  describe ".classes" do
+    it "returns the list of classes of ModelAttributes in .all" do
+      ModelAttribute.instance_variable_set(:@__all__, [
+        ModelAttribute.new(:route, :name, :string),
+        ModelAttribute.new(:journey_pattern, :name, :string),
+        ModelAttribute.new(:time_table, :start_date, :date)
+      ])
+
+      expect(ModelAttribute.classes).to match_array([
+        'Route',
+        'JourneyPattern',
+        'TimeTable'
+      ])
+    end
+  end
+
   describe ".methods_by_class" do
     it "returns all ModelAttributes for a given class" do
       ModelAttribute.instance_variable_set(:@__all__, [