Add OrderCycle#supplied_variants

This commit is contained in:
Rohan Mitchell
2015-11-24 13:33:46 +11:00
parent ccb23cd186
commit ea6974d438
2 changed files with 8 additions and 0 deletions

View File

@@ -135,6 +135,10 @@ class OrderCycle < ActiveRecord::Base
to_a # http://stackoverflow.com/q/15110166
end
def supplied_variants
self.exchanges.incoming.map(&:variants).flatten.uniq.reject(&:deleted?)
end
def distributed_variants
self.exchanges.outgoing.map(&:variants).flatten.uniq.reject(&:deleted?)
end

View File

@@ -211,6 +211,10 @@ describe OrderCycle do
@oc.variants.count.should == 4
end
it "reports on the variants supplied" do
@oc.supplied_variants.should match_array [@p0.master]
end
it "reports on the variants distributed" do
@oc.distributed_variants.should match_array [@p1.master, @p2.master, @p2_v]
end