1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
import os
import unittest
from chirp.drivers import th9800
class Fake9800:
ident = (b'\x54\x48\x39\x38\x30\x30\xff\xff'
b'\x56\x31\x34\x78\xff\xff\xff\xff')
def __init__(self):
self.writebuf = b''
# Pre-load the identify dance
self.readbuf = b'A' + self.ident
def write(self, data):
assert isinstance(data, bytes)
self.writebuf += data
if data.startswith(b'R'):
self.readbuf += b'W\x00\x00\x00' + (b'\x00' * 0x80)
elif data.startswith(b'W'):
self.writebuf += data
self.readbuf += b'A'
elif data == b'A':
assert len(self.readbuf) == 0
self.readbuf += b'A'
elif data == b'ENDW':
self.readbuf += b'123'
def read(self, n):
buf = self.readbuf[:n]
self.readbuf = self.readbuf[n:]
return buf
class TestTH9800(unittest.TestCase):
def test_identify(self):
f = Fake9800()
r = th9800.TYTTH9800Radio(f)
ident = th9800._identify(r)
self.assertEqual(f.ident, ident)
self.assertEqual(b'\x02PROGRAM\x02A', f.writebuf)
def test_download(self):
f = Fake9800()
r = th9800.TYTTH9800Radio(f)
r.sync_in()
def test_upload(self):
f = Fake9800()
img = os.path.join(os.path.dirname(__file__),
'..', 'images', 'TYT_TH-9800.img')
r = th9800.TYTTH9800Radio(img)
r.set_pipe(f)
r.sync_out()
|
