Add attribute accessor for .pluck

Fixes #57
theablefew · Mar 11, 2024 · 6e5b448 · 6e5b448
1 parent 4db16d3
commit 6e5b448
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/stretchy/common.rb b/lib/stretchy/common.rb
@@ -6,6 +6,9 @@ def inspect
             "#<#{self.class.name} #{attributes.map { |k,v| "#{k}: #{v.blank? ? 'nil' : v}" }.join(', ')}>"
         end
 
+        def [](attribute)
+            self.send(attribute)
+        end
 
         class_methods do
 

diff --git a/spec/stretchy/record_spec.rb b/spec/stretchy/record_spec.rb
@@ -73,7 +73,16 @@
             expect(described_class.default_size).to eq(100)
           end
 
+        end
+      end
 
+      context 'attributes' do
+
+        it 'accesses attributes' do
+          expect(described_class.new(title: "hello")[:title]).to eq("hello")
+        end
+        it 'plucks' do
+          expect(described_class.all.pluck(:title)).to be_a(Array)
         end
       end