foreman/test/unit/ensure_no_cycle_test.rb @ acfbc458
acfbc458 | Marek Hulan | require 'test_helper'
|
|
class EnsureNoCycleTest < ActiveSupport::TestCase
|
|||
def setup
|
|||
base = []
|
|||
base.push edge(1, 2)
|
|||
base.push edge(2, 3)
|
|||
base.push edge(3, 4)
|
|||
base.push edge(3, 5)
|
|||
base.push edge(2, 6)
|
|||
@graph = ActiveRecord::Base::EnsureNoCycle.new(base, :source, :target)
|
|||
end
|
|||
def edge(source, target)
|
|||
OpenStruct.new(:source => source, :target => target)
|
|||
end
|
|||
test "#tsort_each_node iterates over all nodes" do
|
|||
found = []
|
|||
@graph.tsort_each_node { |node| found.push node }
|
|||
assert_equal [1, 2, 3, 4, 5, 6], found
|
|||
end
|
|||
test "#tsort_each_child(node) finds all children for node" do
|
|||
found = []
|
|||
assert @graph.tsort_each_child(1) { |node| found.push node }
|
|||
assert_equal [2], found
|
|||
found = []
|
|||
assert @graph.tsort_each_child(2) { |node| found.push node }
|
|||
assert_equal [3, 6], found
|
|||
found = []
|
|||
assert @graph.tsort_each_child(3) { |node| found.push node }
|
|||
assert_equal [4, 5], found
|
|||
found = []
|
|||
assert @graph.tsort_each_child(4) { |node| found.push node }
|
|||
assert_equal [], found
|
|||
end
|
|||
test "#ensure detects cycle and raises an exception" do
|
|||
record = edge(6, 1)
|
|||
record.errors = ActiveModel::Errors.new(record)
|
|||
assert_raises Foreman::CyclicGraphException do
|
|||
@graph.ensure(record)
|
|||
end
|
|||
assert_present record.errors[:base], 'cycle did not add error to record'
|
|||
end
|
|||
test "#ensure passes when record does not create cycle" do
|
|||
record = edge(2, 4)
|
|||
assert_nothing_raised do
|
|||
assert @graph.ensure(record)
|
|||
end
|
|||
end
|
|||
end
|