from __future__ import print_function
from builtins import object
import numpy as np

from mock import *

from atrcopy import SegmentData, iter_parsers
from atrcopy import errors


class BaseContainerTest:
    base_path = None
    expected_mime = ""

    @pytest.mark.parametrize("ext", ['.gz', '.bz2', '.xz', '.dcm'])
    def test_container(self, ext):
        pathname = self.base_path + ext
        try:
            sample_data = np.fromfile(pathname, dtype=np.uint8)
        except OSError:
            pass
        else:
            rawdata = SegmentData(sample_data.copy())
            mime, parser = iter_parsers(rawdata)
            assert mime == self.expected_mime
            assert len(parser.image.files) == self.num_files_in_sample

class TestContainerAtariDosSDImage(BaseContainerTest):
    base_path = "../test_data/container_dos_sd_test1.atr"
    expected_mime = "application/vnd.atari8bit.atr"
    num_files_in_sample = 5

class TestContainerAtariDosEDImage(BaseContainerTest):
    base_path = "../test_data/container_dos_ed_test1.atr"
    expected_mime = "application/vnd.atari8bit.atr"
    num_files_in_sample = 5

class TestContainerAtariDosDDImage(BaseContainerTest):
    base_path = "../test_data/container_dos_dd_test1.atr"
    expected_mime = "application/vnd.atari8bit.atr"
    num_files_in_sample = 5