From ea8f7e8a80f314efafef9c728a98c19348b6958f Mon Sep 17 00:00:00 2001
From: Dan Norman <buzzcutnorman@gmail.com>
Date: Mon, 20 Nov 2023 09:31:46 -0800
Subject: [PATCH] update test to utilize SingerWriter class

---
 tests/_singerlib/test_messages.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/_singerlib/test_messages.py b/tests/_singerlib/test_messages.py
index b4a33db44..e10259497 100644
--- a/tests/_singerlib/test_messages.py
+++ b/tests/_singerlib/test_messages.py
@@ -8,7 +8,7 @@
 from pytz import timezone
 
 import singer_sdk._singerlib as singer
-from singer_sdk._singerlib.messages import format_message
+from singer_sdk.io_base import SingerWriter
 
 UTC = datetime.timezone.utc
 
@@ -19,22 +19,24 @@ def test_exclude_null_dict():
 
 
 def test_format_message():
+    singerwriter = SingerWriter()
     message = singer.RecordMessage(
         stream="test",
         record={"id": 1, "name": "test"},
     )
-    assert format_message(message) == (
+    assert singerwriter.format_message(message) == (
         '{"type": "RECORD", "stream": "test", "record": {"id": 1, "name": "test"}}'
     )
 
 
 def test_write_message():
+    singerwriter = SingerWriter()
     message = singer.RecordMessage(
         stream="test",
         record={"id": 1, "name": "test"},
     )
     with redirect_stdout(io.StringIO()) as out:
-        singer.write_message(message)
+        singerwriter.write_message(message)
 
     assert out.getvalue() == (
         '{"type": "RECORD", "stream": "test", "record": {"id": 1, "name": "test"}}\n'