#!/usr/bin/env python

"""
test parsing
"""

# imports
import speedtest.parse
import os
import shutil
import tempfile
import unittest

here = os.path.dirname(os.path.abspath(__file__))

class ParsingTest(unittest.TestCase):

    def test_parse_columns(self):
        """test columns parsing"""

        columns = ['REPOSITORY',
                   'TAG',
                   'IMAGE ID',
                   'CREATED',
                   'VIRTUAL SIZE']

        # load data
        output = open(os.path.join(here, 'data', 'docker_images.out')).read()

        # parse the data
        data = speedtest.parse.parse_columns(output, columns)
        # (Pdb) data[-1]
        # {'REPOSITORY': 'steeef/sensu-centos', 'CREATED': '2 years ago', 'IMAGE': 'b6f14e', 'VIRTUAL': '1.284 GB', 'TAG': 'latest', 'ID': 'cbffa0', 'SIZE': ''}
        self.assertEqual(data[-1].get("IMAGE ID"), "b6f14ecbffa0")

if __name__ == '__main__':
    unittest.main()