ModelAttribute: Add `.methods_by_class` method

Allows us to get all `ModelAttribute`s given a certain class name.

Needed an `#==` method to be able to test object equality more easily so
I added one in.

I don't like that the `klass` is a lowercase symbol and the `klass`
attribute to `.methods_by_class` is a constantized string. Want to
correct that and make them uniform. Only did it this way because that's
how it was suggested in the ticket.

Refs #4401

1 files changed, 12 insertions, 0 deletions

diff --git a/lib/model_attribute.rb b/lib/model_attribute.rb
index 98bb54b67..8f1ecabed 100644
--- a/lib/model_attribute.rb
+++ b/lib/model_attribute.rb
@@ -9,6 +9,12 @@ class ModelAttribute
     all << new(klass, name, data_type)
   end
 
+  def self.methods_by_class(klass)
+    all.select do |model_attr|
+      model_attr.klass == klass.downcase.to_sym
+    end
+  end
+
   def initialize(klass, name, data_type)
     @klass = klass
     @name = name
@@ -59,4 +65,10 @@ class ModelAttribute
   def code
     "#{@klass}##{@name}"
   end
+
+  def ==(other)
+    klass == other.klass &&
+      name == other.name &&
+      data_type == other.data_type
+  end
 end