require 'rails_helper' describe DotHelper do describe "with example Agents" do class Agents::DotFoo < Agent default_schedule "2pm" def check create_event :payload => {} end end class Agents::DotBar < Agent cannot_be_scheduled! def check create_event :payload => {} end end before do stub(Agents::DotFoo).valid_type?("Agents::DotFoo") { true } stub(Agents::DotBar).valid_type?("Agents::DotBar") { true } end describe "#agents_dot" do before do @agents = [ @foo = Agents::DotFoo.new(name: "foo").tap { |agent| agent.user = users(:bob) agent.save! }, @bar1 = Agents::DotBar.new(name: "bar1").tap { |agent| agent.user = users(:bob) agent.sources << @foo agent.save! }, @bar2 = Agents::DotBar.new(name: "bar2").tap { |agent| agent.user = users(:bob) agent.sources << @foo agent.propagate_immediately = true agent.disabled = true agent.save! }, @bar3 = Agents::DotBar.new(name: "bar3").tap { |agent| agent.user = users(:bob) agent.sources << @bar2 agent.save! }, ] @foo.reload @bar2.reload end it "generates a DOT script" do expect(agents_dot(@agents)).to match(%r{ \A digraph \x20 "Agent \x20 Event \x20 Flow" \{ node \[ [^\]]+ \]; edge \[ [^\]]+ \]; (?\w+) \[label=foo\]; \k -> (?\w+) \[style=dashed\]; \k -> (?\w+) \[color="\#999999"\]; \k \[label=bar1\]; \k \[label=bar2,style="rounded,dashed",color="\#999999",fontcolor="\#999999"\]; \k -> (?\w+) \[style=dashed,color="\#999999"\]; \k \[label=bar3\]; \} \z }x) end it "generates a richer DOT script" do expect(agents_dot(@agents, rich: true)).to match(%r{ \A digraph \x20 "Agent \x20 Event \x20 Flow" \{ node \[ [^\]]+ \]; edge \[ [^\]]+ \]; (?\w+) \[label=foo,tooltip="Dot \x20 Foo",URL="#{Regexp.quote(agent_path(@foo))}"\]; \k -> (?\w+) \[style=dashed\]; \k -> (?\w+) \[color="\#999999"\]; \k \[label=bar1,tooltip="Dot \x20 Bar",URL="#{Regexp.quote(agent_path(@bar1))}"\]; \k \[label=bar2,tooltip="Dot \x20 Bar",URL="#{Regexp.quote(agent_path(@bar2))}",style="rounded,dashed",color="\#999999",fontcolor="\#999999"\]; \k -> (?\w+) \[style=dashed,color="\#999999"\]; \k \[label=bar3,tooltip="Dot \x20 Bar",URL="#{Regexp.quote(agent_path(@bar3))}"\]; \} \z }x) end end end describe "DotHelper::DotDrawer" do describe "#id" do it "properly escapes double quotaion and backslash" do expect(DotHelper::DotDrawer.draw(foo: "") { id('hello\\"') }).to eq('"hello\\\\\\""') end end end end