X-Git-Url: https://git.phdru.name/?p=sqlconvert.git;a=blobdiff_plain;f=tests%2Ftest_mysql2postgres.py;h=6782c5408eeef1b33335339da8e32a7ac9126ed8;hp=b81c6bf4245797900b38c6730c54dbf0753cfa59;hb=HEAD;hpb=3fa4c00fb1cb34b1b47914e4d92ba8da2cdc2c82

diff --git a/tests/test_mysql2postgres.py b/tests/test_mysql2postgres.py
index b81c6bf..6782c54 100644
--- a/tests/test_mysql2postgres.py
+++ b/tests/test_mysql2postgres.py
@@ -4,6 +4,7 @@ import pytest
 
 from sqlconvert.print_tokens import tlist2str
 from sqlconvert.process_mysql import unescape_strings
+from sqlconvert.process_tokens import escape_strings
 
 connection = getConnection()
 pytestmark = pytest.mark.skipif(connection.dbName != "postgres",
@@ -19,11 +20,13 @@ CREATE TABLE test (
 
 def test_mysql2postgres_string():
     connection.query(create_postgres_test_table)
-    parsed = parse("insert into test (id, test_str) "
-                   "values (1, '\"te\\'st\\\"')")[0]
+    parsed = parse("insert into test (id, test_str) values "
+                   "(1, '\"te\\'st\\\"\\n')")[0]
     unescape_strings(parsed)
+    escape_strings(parsed, 'postgres')
     query = tlist2str(parsed)
-    assert query == u"INSERT INTO test (id, test_str) VALUES (1, '\"te''st\"')"
+    assert query == u"INSERT INTO test (id, test_str) VALUES " \
+                    u"(1, E'\"te''st\"\\n')"
     connection.query(query)
     test_str = connection.queryOne("SELECT test_str FROM test WHERE id=1")[0]
-    assert test_str == u"\"te'st\""
+    assert test_str == u"\"te'st\"\n"