spec/formatter_spec.rb in cistern-2.8.0 vs spec/formatter_spec.rb in cistern-2.8.1

@@ -1,22 +1,29 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 
 class Inspector < Sample::Model
   identity :id
   attribute :name
+
+  has_many :employees, -> { Inspectors.new(associate_id: id).all }
+  has_many :managers, -> { Inspectors.new(associate_id: nil).all }
 end
 
 class Anon < Sample::Model
   attribute :chan
 end
 
 class Inspectors < Sample::Collection
+  attribute :associate_id
+
   model Inspector
 
   def all(options = {})
     merge_attributes(options)
-    load([{ id: 1, name: '2' }, { id: 3, name: '4' }])
+    load([{ id: 1, name: '2', managers: [{ id: 5, name: '5'}]}, { id: 3, name: '4' }, employees: [{ id: 9, name: '8'}]])
   end
 end
 
 describe Cistern::Formatter::Default do
   before { Cistern.formatter = described_class }
@@ -62,23 +69,36 @@
   it 'formats a model' do
     Cistern.formatter = Cistern::Formatter::Formatador
 
     expect(Inspector.new(id: 1, name: 'name').inspect).to eq('  <Inspector
     id=1,
-    name="name"
+    name="name",
+    employees=nil,
+    managers=nil
   >')
   end
 
   it 'formats a collection' do
     expect(Inspectors.new.all.inspect).to eq('  <Inspectors
+    associate_id=nil
     [
       <Inspector
         id=1,
-        name="2"
+        name="2",
+        employees=nil,
+        managers=[{"id"=>5, "name"=>"5"}]
       >,
       <Inspector
         id=3,
-        name="4"
+        name="4",
+        employees=nil,
+        managers=nil
+      >,
+      <Inspector
+        id=nil,
+        name=nil,
+        employees=[{"id"=>9, "name"=>"8"}],
+        managers=nil
       >
     ]
   >')
   end
 end