test/node/test_farmers.rb in zold-0.17.6 vs test/node/test_farmers.rb in zold-0.17.7

- old
+ new

@@ -26,24 +26,31 @@ require_relative '../test__helper' require_relative '../../lib/zold/node/farmers' require_relative '../../lib/zold/verbose_thread' class FarmersTest < Zold::Test + # Types to test + TYPES = [ + Zold::Farmers::Plain, + Zold::Farmers::Spawn, + Zold::Farmers::Fork + ].freeze + def test_calculates_next_score before = Zold::Score.new(host: 'some-host', port: 9999, invoice: 'NOPREFIX4@ffffffffffffffff', strength: 3) - [Zold::Farmers::Plain, Zold::Farmers::Fork].each do |farmer_class| + TYPES.each do |farmer_class| farmer = farmer_class.new(log: test_log) after = farmer.up(before) assert_equal(1, after.value) assert(!after.expired?) assert_equal('some-host', after.host) assert_equal(9999, after.port) end end def test_calculates_large_score - [Zold::Farmers::Plain, Zold::Farmers::Fork].each do |type| + TYPES.each do |type| log = TestLogger.new(test_log) thread = Thread.start do farmer = type.new(log: log) farmer.up(Zold::Score.new(host: 'a', port: 1, invoice: 'NOPREFIX4@ffffffffffffffff', strength: 20)) end @@ -52,10 +59,10 @@ thread.join end end def test_kills_farmer - [Zold::Farmers::Plain, Zold::Farmers::Fork].each do |type| + TYPES.each do |type| farmer = type.new(log: test_log) thread = Thread.start do Zold::VerboseThread.new(test_log).run do farmer.up(Zold::Score.new(host: 'some-host', invoice: 'NOPREFIX4@ffffffffffffffff', strength: 32)) end