# -*- Mode: Python; test-case-name: morituri.test.test_image_cue -*-
# vi:si:et:sw=4:sts=4:ts=4

import os
import unittest

import gobject
gobject.threads_init()

from morituri.image import image
from morituri.common import task

def h(i):
    return "0x%08x" % i

class TrackSingleTestCase(unittest.TestCase):
    def setUp(self):
        self.image = image.Image(os.path.join(os.path.dirname(__file__),
            'track-single.cue'))
        self.runner = task.SyncRunner()
        self.image.setup(self.runner)

    def testAudioRipCRC(self):
        crctask = image.AudioRipCRCTask(self.image) 
        self.runner.run(crctask, verbose=False)

        self.assertEquals(len(crctask.crcs), 4)
        self.assertEquals(h(crctask.crcs[0]), '0x00000000')
        self.assertEquals(h(crctask.crcs[1]), '0x793fa868')
        self.assertEquals(h(crctask.crcs[2]), '0x8dd37c26')
        self.assertEquals(h(crctask.crcs[3]), '0x00000000')

    def testLength(self):
        tracks = self.image.cue.tracks
        self.assertEquals(self.image.getTrackLength(tracks[0]), 2)
        self.assertEquals(self.image.getTrackLength(tracks[1]), 2)
        self.assertEquals(self.image.getTrackLength(tracks[2]), 2)
        self.assertEquals(self.image.getTrackLength(tracks[3]), 4)

    def testCDDB(self):
        self.assertEquals(self.image.getCDDBDiscId(), "08000004")

    def testAccurateRip(self):
        self.assertEquals(self.image.getAccurateRipIds(), (
            "00000016", "0000005b"))

class TracSeparateTestCase(unittest.TestCase):
    def setUp(self):
        self.image = image.Image(os.path.join(os.path.dirname(__file__),
            'track-separate.cue'))
        self.runner = task.SyncRunner()
        self.image.setup(self.runner)

    def testAudioRipCRC(self):
        crctask = image.AudioRipCRCTask(self.image) 
        self.runner.run(crctask, verbose=False)

        self.assertEquals(len(crctask.crcs), 4)
        self.assertEquals(h(crctask.crcs[0]), '0xaf18681e')
        self.assertEquals(h(crctask.crcs[1]), '0xd63dc2d2')
        self.assertEquals(h(crctask.crcs[2]), '0xd63dc2d2')
        self.assertEquals(h(crctask.crcs[3]), '0x7271db39')

    def testLength(self):
        tracks = self.image.cue.tracks
        self.assertEquals(self.image.getTrackLength(tracks[0]), 10)
        self.assertEquals(self.image.getTrackLength(tracks[1]), 10)
        self.assertEquals(self.image.getTrackLength(tracks[2]), 10)
        self.assertEquals(self.image.getTrackLength(tracks[3]), 10)

    def testCDDB(self):
        self.assertEquals(self.image.getCDDBDiscId(), "08000004")

    def testAccurateRip(self):
        self.assertEquals(self.image.getAccurateRipIds(), (
            "00000064", "00000191"))

class AudioLengthTestCase(unittest.TestCase):
    def testLength(self):
        path = os.path.join(os.path.dirname(__file__), 'track.flac')
        t = image.AudioLengthTask(path)
        runner = task.SyncRunner()
        runner.run(t, verbose=False)
        self.assertEquals(t.length, 5880)

class AccurateRipResponseTestCase(unittest.TestCase):
    def testResponse(self):
        path = os.path.join(os.path.dirname(__file__),
            'dBAR-011-0010e284-009228a3-9809ff0b.bin')
        data = open(path, "rb").read()
        response = image.AccurateRipResponse(data)

        self.assertEquals(response.trackCount, 11)
        self.assertEquals(response.discId1, "0010e284")
        self.assertEquals(response.discId2, "009228a3")
        self.assertEquals(response.cddbDiscId, "9809ff0b")

        for i in range(11):
            self.assertEquals(response.confidences[i], 35)
        self.assertEquals(response.crcs[0], "beea32c8")
        self.assertEquals(response.crcs[10], "acee98ca")