Python helper.capture_stdout函数代码示例

 def test_show_user_config_with_defaults(self):
     with capture_stdout() as output:
         self.run_command('config', '-d')
     output = yaml.load(output.getvalue())
     self.assertEqual(output['option'], 'value')
     self.assertEqual(output['library'], 'lib')
     self.assertEqual(output['import']['timid'], False)

    def test_list_unicode_query(self):
        self.item.title = u'na\xefve'
        self.item.store()
        self.lib._connection().commit()

        with capture_stdout() as stdout:
            self._run_list([u'na\xefve'])
        out = stdout.getvalue()
        self.assertTrue(u'na\xefve' in out.decode(stdout.encoding))

    def test_write_metadata_field(self):
        item = self.add_item_fixture()
        item.read()
        old_title = item.title

        item.title = "new title"
        item.store()

        with capture_stdout() as stdout:
            self.write_cmd()

        self.assertTrue("{0} -> new title".format(old_title) in stdout.getvalue())

    def test_print_tracks_output_as_tracks(self):
        """Test the output of the "print tracks" choice, as singletons."""
        self.matcher.matching = AutotagStub.BAD

        with capture_stdout() as output:
            with control_stdin("\n".join(["t", "s", "p", "s"])):
                # as Tracks; Skip; Print tracks; Skip
                self.importer.run()

        # Manually build the string for comparing the output.
        tracklist = u"Print tracks? " u"02. Tag Title 2 - Tag Artist (0:01)"
        self.assertIn(tracklist, output.getvalue())

    def test_print_tracks_output(self):
        """Test the output of the "print tracks" choice."""
        self.matcher.matching = AutotagStub.BAD

        with capture_stdout() as output:
            with control_stdin('\n'.join(['p', 's'])):
                # Print tracks; Skip
                self.importer.run()

        # Manually build the string for comparing the output.
        tracklist = ('Print tracks? '
                     '01. Tag Title 1 - Tag Artist (0:01)\n'
                     '02. Tag Title 2 - Tag Artist (0:01)')
        self.assertIn(tracklist, output.getvalue())

    def test_non_metadata_field_unchanged(self):
        """Changing a non-"tag" field like `bitrate` and writing should
        have no effect.
        """
        # An item that starts out "clean".
        item = self.add_item_fixture()
        item.read()

        # ... but with a mismatched bitrate.
        item.bitrate = 123
        item.store()

        with capture_stdout() as stdout:
            self.write_cmd()

        self.assertEqual(stdout.getvalue(), '')

 def test_list_album_outputs_something(self):
     with capture_stdout() as stdout:
         self._run_list(album=True)
     self.assertGreater(len(stdout.getvalue()), 0)

 def test_list_item_path(self):
     with capture_stdout() as stdout:
         self._run_list(fmt=u'$path')
     self.assertEqual(stdout.getvalue().strip(), u'xxx/yyy')

 def test_list_outputs_item(self):
     with capture_stdout() as stdout:
         self._run_list()
     self.assertIn(u'the title', stdout.getvalue())

 def test_list_album_uses_album_artist(self):
     with capture_stdout() as stdout:
         self._run_list(album=True)
     self.assertNotIn(u'the artist', stdout.getvalue())
     self.assertIn(u'the album artist', stdout.getvalue())

 def test_list_item_format_artist(self):
     with capture_stdout() as stdout:
         self._run_list(fmt=u'$artist')
     self.assertIn(u'the artist', stdout.getvalue())

 def test_config_paths_with_cli(self):
     with capture_stdout() as output:
         self.run_command('--config', self.cli_config_path, 'config', '-p')
     paths = output.getvalue().split('\n')
     self.assertEqual(len(paths), 3)
     self.assertEqual(paths[0], self.cli_config_path)

 def test_show_user_config_with_cli(self):
     with capture_stdout() as output:
         self.run_command('--config', self.cli_config_path, 'config')
     output = yaml.load(output.getvalue())
     self.assertEqual(output['library'], 'lib')
     self.assertEqual(output['option'], 'cli overwrite')

 def test_show_user_config(self):
     with capture_stdout() as output:
         self.run_command('config')
     output = yaml.load(output.getvalue())
     self.assertEqual(output['option'], 'value')

 def test_list_uses_track_artist(self):
     with capture_stdout() as stdout:
         self._run_list()
     self.assertIn("the artist", stdout.getvalue())
     self.assertNotIn("the album artist", stdout.getvalue())