diff --git a/Foreman/Models/Nodes/PassthroughNode.cs b/Foreman/Models/Nodes/PassthroughNode.cs index 23c9532..0d7c097 100644 --- a/Foreman/Models/Nodes/PassthroughNode.cs +++ b/Foreman/Models/Nodes/PassthroughNode.cs @@ -21,7 +21,7 @@ public enum Errors { public bool SimpleDraw { get; set; } public PassthroughNode(ProductionGraph graph, int nodeID, ItemQualityPair item) : base(graph, nodeID) { PassthroughItem = item; - SimpleDraw = true; + SimpleDraw = graph.DefaultToSimplePassthroughNodes; controller = new PassthroughNodeController(this); } diff --git a/ForemanTest/ItemQualityNodeDiagnosticsTests.cs b/ForemanTest/ItemQualityNodeDiagnosticsTests.cs index 5e62253..139e491 100644 --- a/ForemanTest/ItemQualityNodeDiagnosticsTests.cs +++ b/ForemanTest/ItemQualityNodeDiagnosticsTests.cs @@ -6,6 +6,7 @@ using ForemanTest.support; using Microsoft.VisualStudio.TestTools.UnitTesting; using System; +using System.Drawing; using System.Linq; namespace ForemanTest { @@ -77,6 +78,20 @@ public void Passthrough_ToString_DescribesPassthroughNode() { Assert.IsFalse(node.ToString().StartsWith("Supply node", StringComparison.Ordinal)); } + [TestMethod] + public void Passthrough_NewNode_UsesGraphDefaultSimpleDraw() { + var ctx = GraphSessionTestHelper.CreateContext(); + var graph = ctx.NewGraph(); + + graph.DefaultToSimplePassthroughNodes = false; + PassthroughNode plain = graph.CreatePassthroughNode(ctx.Item("belt"), Point.Empty); + Assert.IsFalse(plain.SimpleDraw); + + graph.DefaultToSimplePassthroughNodes = true; + PassthroughNode simplified = graph.CreatePassthroughNode(ctx.Item("iron"), Point.Empty); + Assert.IsTrue(simplified.SimpleDraw); + } + [TestMethod] public void Spoil_ToString_IncludesInputOutputAndQualities() { var ctx = GraphSessionTestHelper.CreateContext();