From 49fc2ac5e9b137bd84639c550773546bb0962759 Mon Sep 17 00:00:00 2001
From: Johnny Robeson <johnny@localmomentum.net>
Date: Sun, 18 Sep 2016 03:31:00 -0400
Subject: [PATCH] Use run_with_command in write_cmd in test_ui.py

---
 test/test_ui.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/test/test_ui.py b/test/test_ui.py
index 89e099f8a..31f3f37da 100644
--- a/test/test_ui.py
+++ b/test/test_ui.py
@@ -354,7 +354,7 @@ class WriteTest(unittest.TestCase, TestHelper):
         self.teardown_beets()
 
     def write_cmd(self, *args):
-        self.run_command('write', *args)
+        return self.run_with_output('write', *args)
 
     def test_update_mtime(self):
         item = self.add_item_fixture()
@@ -380,10 +380,9 @@ class WriteTest(unittest.TestCase, TestHelper):
         item.bitrate = 123
         item.store()
 
-        with capture_stdout() as stdout:
-            self.write_cmd()
+        output = self.write_cmd()
 
-        self.assertEqual(stdout.getvalue(), '')
+        self.assertEqual(output, '')
 
     def test_write_metadata_field(self):
         item = self.add_item_fixture()
@@ -393,11 +392,10 @@ class WriteTest(unittest.TestCase, TestHelper):
         item.title = u'new title'
         item.store()
 
-        with capture_stdout() as stdout:
-            self.write_cmd()
+        output = self.write_cmd()
 
         self.assertTrue(u'{0} -> new title'.format(old_title)
-                        in stdout.getvalue())
+                        in output)
 
 
 class MoveTest(_common.TestCase):