writer_test: Use new-style trollius yield From.

ArchiveTeam · Oct 9, 2014 · d84435e · d84435e
1 parent 430a4b8
commit d84435e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/wpull/writer_test.py b/wpull/writer_test.py
@@ -170,7 +170,7 @@ def test_new_file_and_clobber(self):
 
         with cd_tempdir() as temp_dir:
             app = Builder(args, unit_test=True).build()
-            exit_code = yield app.run()
+            exit_code = yield From(app.run())
 
             self.assertEqual(0, exit_code)
 
@@ -203,7 +203,7 @@ def test_file_continue(self):
                 out_file.write(b'START')
 
             app = Builder(args, unit_test=True).build()
-            exit_code = yield app.run()
+            exit_code = yield From(app.run())
 
             self.assertEqual(0, exit_code)
 
@@ -231,7 +231,7 @@ def test_timestamping_hit(self):
 
             builder = Builder(args, unit_test=True)
             app = builder.build()
-            exit_code = yield app.run()
+            exit_code = yield From(app.run())
 
             self.assertEqual(0, exit_code)
 
@@ -256,7 +256,7 @@ def test_timestamping_miss(self):
 
             builder = Builder(args, unit_test=True)
             app = builder.build()
-            exit_code = yield app.run()
+            exit_code = yield From(app.run())
 
             self.assertEqual(0, exit_code)