1from test.test_support import findfile, run_unittest, TESTFN
2import unittest
3import os
4import io
5
6import aifc
7
8
9class AIFCTest(unittest.TestCase):
10
11    def setUp(self):
12        self.f = self.fout = None
13        self.sndfilepath = findfile('Sine-1000Hz-300ms.aif')
14
15    def tearDown(self):
16        if self.f is not None:
17            self.f.close()
18        if self.fout is not None:
19            try:
20                self.fout.close()
21            except (aifc.Error, AttributeError):
22                pass
23        try:
24            os.remove(TESTFN)
25        except OSError:
26            pass
27
28    def test_skipunknown(self):
29        #Issue 2245
30        #This file contains chunk types aifc doesn't recognize.
31        self.f = aifc.open(self.sndfilepath)
32
33    def test_params(self):
34        f = self.f = aifc.open(self.sndfilepath)
35        self.assertEqual(f.getnchannels(), 2)
36        self.assertEqual(f.getsampwidth(), 2)
37        self.assertEqual(f.getframerate(), 48000)
38        self.assertEqual(f.getnframes(), 14400)
39        self.assertEqual(f.getcomptype(), 'NONE')
40        self.assertEqual(f.getcompname(), 'not compressed')
41        self.assertEqual(f.getparams(), (2, 2, 48000, 14400, 'NONE', 'not compressed'))
42
43    def test_read(self):
44        f = self.f = aifc.open(self.sndfilepath)
45        self.assertEqual(f.tell(), 0)
46        self.assertEqual(f.readframes(2), '\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
47        f.rewind()
48        pos0 = f.tell()
49        self.assertEqual(pos0, 0)
50        self.assertEqual(f.readframes(2), '\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
51        pos2 = f.tell()
52        self.assertEqual(pos2, 2)
53        self.assertEqual(f.readframes(2), '\x17t\x17t"\xad"\xad')
54        f.setpos(pos2)
55        self.assertEqual(f.readframes(2), '\x17t\x17t"\xad"\xad')
56        f.setpos(pos0)
57        self.assertEqual(f.readframes(2), '\x00\x00\x00\x00\x0b\xd4\x0b\xd4')
58
59    def test_write(self):
60        f = self.f = aifc.open(self.sndfilepath)
61        fout = self.fout = aifc.open(TESTFN, 'wb')
62        fout.aifc()
63        fout.setparams(f.getparams())
64        for frame in range(f.getnframes()):
65            fout.writeframes(f.readframes(1))
66        fout.close()
67        fout = self.fout = aifc.open(TESTFN, 'rb')
68        f.rewind()
69        self.assertEqual(f.getparams(), fout.getparams())
70        self.assertEqual(f.readframes(5), fout.readframes(5))
71
72    def test_compress(self):
73        f = self.f = aifc.open(self.sndfilepath)
74        fout = self.fout = aifc.open(TESTFN, 'wb')
75        fout.aifc()
76        fout.setnchannels(f.getnchannels())
77        fout.setsampwidth(f.getsampwidth())
78        fout.setframerate(f.getframerate())
79        fout.setcomptype('ULAW', 'foo')
80        for frame in range(f.getnframes()):
81            fout.writeframes(f.readframes(1))
82        fout.close()
83        self.assertLess(
84            os.stat(TESTFN).st_size,
85            os.stat(self.sndfilepath).st_size*0.75,
86            )
87        fout = self.fout = aifc.open(TESTFN, 'rb')
88        f.rewind()
89        self.assertEqual(f.getparams()[0:3], fout.getparams()[0:3])
90        self.assertEqual(fout.getcomptype(), 'ULAW')
91        self.assertEqual(fout.getcompname(), 'foo')
92        # XXX: this test fails, not sure if it should succeed or not
93        # self.assertEqual(f.readframes(5), fout.readframes(5))
94
95    def test_close(self):
96        class Wrapfile(object):
97            def __init__(self, file):
98                self.file = open(file, 'rb')
99                self.closed = False
100            def close(self):
101                self.file.close()
102                self.closed = True
103            def __getattr__(self, attr): return getattr(self.file, attr)
104        testfile = Wrapfile(self.sndfilepath)
105        f = self.f = aifc.open(testfile)
106        self.assertEqual(testfile.closed, False)
107        f.close()
108        self.assertEqual(testfile.closed, True)
109        testfile = open(TESTFN, 'wb')
110        fout = aifc.open(testfile, 'wb')
111        self.assertFalse(testfile.closed)
112        with self.assertRaises(aifc.Error):
113            fout.close()
114        self.assertTrue(testfile.closed)
115        fout.close() # do nothing
116
117
118class AIFCLowLevelTest(unittest.TestCase):
119
120    def test_read_written(self):
121        def read_written(self, what):
122            f = io.BytesIO()
123            getattr(aifc, '_write_' + what)(f, x)
124            f.seek(0)
125            return getattr(aifc, '_read_' + what)(f)
126        for x in (-1, 0, 0.1, 1):
127            self.assertEqual(read_written(x, 'float'), x)
128        for x in (float('NaN'), float('Inf')):
129            self.assertEqual(read_written(x, 'float'), aifc._HUGE_VAL)
130        for x in (b'', b'foo', b'a' * 255):
131            self.assertEqual(read_written(x, 'string'), x)
132        for x in (-0x7FFFFFFF, -1, 0, 1, 0x7FFFFFFF):
133            self.assertEqual(read_written(x, 'long'), x)
134        for x in (0, 1, 0xFFFFFFFF):
135            self.assertEqual(read_written(x, 'ulong'), x)
136        for x in (-0x7FFF, -1, 0, 1, 0x7FFF):
137            self.assertEqual(read_written(x, 'short'), x)
138        for x in (0, 1, 0xFFFF):
139            self.assertEqual(read_written(x, 'ushort'), x)
140
141    def test_read_raises(self):
142        f = io.BytesIO(b'\x00')
143        self.assertRaises(EOFError, aifc._read_ulong, f)
144        self.assertRaises(EOFError, aifc._read_long, f)
145        self.assertRaises(EOFError, aifc._read_ushort, f)
146        self.assertRaises(EOFError, aifc._read_short, f)
147
148    def test_write_long_string_raises(self):
149        f = io.BytesIO()
150        with self.assertRaises(ValueError):
151            aifc._write_string(f, b'too long' * 255)
152
153
154def test_main():
155    run_unittest(AIFCTest)
156    run_unittest(AIFCLowLevelTest)
157
158
159if __name__ == "__main__":
160    unittest.main()
161