print direction

badetitou · Sep 9, 2022 · 74666d9 · 74666d9
1 parent ccffe9c
commit 74666d9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/MermaidPharo-Writter-Flowchart-Tests/MeWritterVisitorFlowchartTest.class.st b/src/MermaidPharo-Writter-Flowchart-Tests/MeWritterVisitorFlowchartTest.class.st
@@ -26,3 +26,15 @@ MeWritterVisitorFlowchartTest >> testWriteFlowchartDiagram [
 		          flowchartDiagram accept: writter ].
 	self assert: result equals: 'flowchart'
 ]
+
+{ #category : #initialization }
+MeWritterVisitorFlowchartTest >> testWriteFlowchartDiagramWithDirection [
+
+	| flowchartDiagram result |
+	flowchartDiagram := MeFlowchartDiagram new.
+	flowchartDiagram direction: MeFCDirection TB.
+	result := String streamContents: [ :stream | 
+		          writter outputStream: stream.
+		          flowchartDiagram accept: writter ].
+	self assert: result equals: 'flowchart TB'
+]
diff --git a/src/MermaidPharo-Writter-Flowchart/MeWritterVisitorFlowchart.class.st b/src/MermaidPharo-Writter-Flowchart/MeWritterVisitorFlowchart.class.st
@@ -6,5 +6,10 @@ Class {
 
 { #category : #visiting }
 MeWritterVisitorFlowchart >> visitFlowchartDiagram: aFlowchartDiagram [
-	self <<< 'flowchart'
+
+	self <<< 'flowchart'.
+	aFlowchartDiagram direction ifNotNil: [ :direction | 
+		self
+			<<< ' ';
+			<<< direction directionIdentifier ]
 ]