Merge pull request #5 from svenvc/master

P3FormattedStatement printOn: fix
svenvc · Mar 7, 2024 · 16c4cc5 · 16c4cc5
2 parents 29d48c2 + 769ed55
commit 16c4cc5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/P3-Tests/P3FormattedStatementTest.class.st b/P3-Tests/P3FormattedStatementTest.class.st
@@ -211,6 +211,24 @@ P3FormattedStatementTest >> testNull [
 	client execute: 'DROP TABLE table1'
 ]
 
+{ #category : #tests }
+P3FormattedStatementTest >> testPrintOn [
+"client format: returns a P3FormattedStatement. Use result to test printOn:"
+
+	| sql p3fs result|
+
+	sql := 'INSERT INTO table1 (id, name) VALUES ($1, $2)'.
+
+	p3fs := client format: sql.
+
+	result := String streamContents: [ :stream | p3fs printOn: stream ].
+
+	self assert: result contents equals: ('a P3FormattedStatement({1})' format: { sql })
+
+
+
+]
+
 { #category : #tests }
 P3FormattedStatementTest >> testSimple [
 	| statement result |

diff --git a/P3/P3FormattedStatement.class.st b/P3/P3FormattedStatement.class.st
@@ -139,7 +139,7 @@ P3FormattedStatement >> printObject: object on: stream [
 { #category : #printing }
 P3FormattedStatement >> printOn: stream [
 	super printOn: stream.
-	stream nextPut: $(; << sql; nextPut: $)
+	stream nextPut: $(; << source; nextPut: $)
 ]
 
 { #category : #'printing-dispatched' }