File: test_icom_clone.py

package info (click to toggle)
chirp 1%3A20221106%2Bpy3-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 11,144 kB
  • sloc: python: 119,334; ansic: 296; sh: 184; makefile: 41
file content (117 lines) | stat: -rw-r--r-- 4,210 bytes parent folder | download | duplicates (2)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
# Copyright 2019 Dan Smith <dsmith@danplanet.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from builtins import bytes

import glob
import os
import logging
import unittest

from chirp import directory
directory.safe_import_drivers()
from chirp.drivers import icf
from chirp import memmap
from tests import icom_clone_simulator


class BaseIcomCloneTest():
    def setUp(self):
        self.radio = directory.get_radio(self.RADIO_IDENT)(None)
        self.simulator = icom_clone_simulator.FakeIcomRadio(self.radio)
        self.radio.set_pipe(self.simulator)

    def image_filename(self, filename):
        tests_base = os.path.dirname(os.path.abspath(__file__))
        return os.path.join(tests_base, 'images', filename)

    def load_from_test_image(self, filename):
        self.simulator.load_from_file(self.image_filename(filename))

    def image_version(self, filename):
        end = self.radio.get_memsize()

        with open(self.image_filename(filename), 'rb') as f:
            data = f.read()
        if data[-8:-6] == b'%02x' % self.radio.get_model()[0]:
            return 2
        elif data[end-16:end] == b'IcomCloneFormat3':
            return 3


    def test_sync_in(self):
        test_file = self.IMAGE_FILE
        self.load_from_test_image(test_file)
        self.radio.sync_in()

        img_ver = self.image_version(test_file)
        if img_ver == 2:
            endstring = b''.join(b'%02x' % ord(c)
                                 for c in self.radio._model[:2])
            self.assertEqual(endstring + b'0001', self.radio._mmap[-8:])
        elif img_ver == 3:
            self.assertEqual(b'IcomCloneFormat3', self.radio._mmap[-16:])
        elif img_ver is None:
            self.assertEqual(self.radio.get_memsize(), len(self.radio._mmap))

    def test_sync_out(self):
        self.radio._mmap = memmap.MemoryMapBytes(
            bytes(b'\x00') * self.radio.get_memsize())
        self.radio._mmap[50] = bytes(b'abcdefgh')
        self.radio.sync_out()
        self.assertEqual(b'abcdefgh', self.simulator._memory[50:58])


class TestRawRadioData(unittest.TestCase):
    def test_get_payload(self):
        radio = directory.get_radio('Icom_IC-2730A')(None)
        payload = radio.get_payload(bytes(b'\x00\x10\xFE\x00'), True, True)
        self.assertEqual(b'\x00\x10\xFF\x0E\x00\xF2', payload)

        payload = radio.get_payload(bytes(b'\x00\x10\xFE\x00'), True, False)
        self.assertEqual(b'\x00\x10\xFF\x0E\x00', payload)

    def test_process_frame_payload(self):
        radio = directory.get_radio('Icom_IC-2730A')(None)
        data = radio.process_frame_payload(bytes(b'\x00\x10\xFF\x0E\x00'))
        self.assertEqual(b'\x00\x10\xFE\x00', data)


class TestAdapterMeta(type):
    def __new__(cls, name, parents, dct):
        return super(TestAdapterMeta, cls).__new__(cls, name, parents, dct)


test_file_glob = os.path.join(os.path.dirname(os.path.abspath(__file__)),
                              '..', 'images',
                              'Icom_*.img')
import sys
for image_file in glob.glob(test_file_glob):
    base, _ext = os.path.splitext(os.path.basename(image_file))

    try:
        radio = directory.get_radio(base)
    except Exception:
        continue

    if issubclass(radio, icf.IcomRawCloneModeRadio):
        # The simulator does not behave like a raw radio
        continue

    class_name = 'Test_%s' % base
    sys.modules[__name__].__dict__[class_name] = \
        TestAdapterMeta(class_name,
                        (BaseIcomCloneTest, unittest.TestCase),
                        dict(RADIO_IDENT=base, IMAGE_FILE=image_file))