from __future__ import unicode_literals
from __future__ import absolute_import
from tests import unittest

import mock
from six import StringIO

from fig import progress_stream 


class ProgressStreamTestCase(unittest.TestCase):

    def test_stream_output(self):
        output = [
            '{"status": "Downloading", "progressDetail": {"current": '
            '31019763, "start": 1413653874, "total": 62763875}, '
            '"progress": "..."}',
        ]
        events = progress_stream.stream_output(output, StringIO())
        self.assertEqual(len(events), 1)