File: test_cli.py

package info (click to toggle)
python-can 4.6.1-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,428 kB
  • sloc: python: 27,154; makefile: 32; sh: 16
file content (154 lines) | stat: -rw-r--r-- 5,524 bytes parent folder | download
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
import argparse
import unittest
from unittest.mock import patch

from can.cli import add_bus_arguments, create_bus_from_namespace


class TestCliUtils(unittest.TestCase):
    def test_add_bus_arguments(self):
        parser = argparse.ArgumentParser()
        add_bus_arguments(parser, filter_arg=True, prefix="test")

        parsed_args = parser.parse_args(
            [
                "--test-channel",
                "0",
                "--test-interface",
                "vector",
                "--test-timing",
                "f_clock=8000000",
                "brp=4",
                "tseg1=11",
                "tseg2=4",
                "sjw=2",
                "nof_samples=3",
                "--test-filter",
                "100:7FF",
                "200~7F0",
                "--test-bus-kwargs",
                "app_name=MyApp",
                "serial=1234",
            ]
        )

        self.assertNotIn("channel", parsed_args)
        self.assertNotIn("test_bitrate", parsed_args)
        self.assertNotIn("test_data_bitrate", parsed_args)
        self.assertNotIn("test_fd", parsed_args)

        self.assertEqual(parsed_args.test_channel, "0")
        self.assertEqual(parsed_args.test_interface, "vector")
        self.assertEqual(parsed_args.test_timing.f_clock, 8000000)
        self.assertEqual(parsed_args.test_timing.brp, 4)
        self.assertEqual(parsed_args.test_timing.tseg1, 11)
        self.assertEqual(parsed_args.test_timing.tseg2, 4)
        self.assertEqual(parsed_args.test_timing.sjw, 2)
        self.assertEqual(parsed_args.test_timing.nof_samples, 3)
        self.assertEqual(len(parsed_args.test_can_filters), 2)
        self.assertEqual(parsed_args.test_can_filters[0]["can_id"], 0x100)
        self.assertEqual(parsed_args.test_can_filters[0]["can_mask"], 0x7FF)
        self.assertEqual(parsed_args.test_can_filters[1]["can_id"], 0x200 | 0x20000000)
        self.assertEqual(
            parsed_args.test_can_filters[1]["can_mask"], 0x7F0 & 0x20000000
        )
        self.assertEqual(parsed_args.test_bus_kwargs["app_name"], "MyApp")
        self.assertEqual(parsed_args.test_bus_kwargs["serial"], 1234)

    def test_add_bus_arguments_no_prefix(self):
        parser = argparse.ArgumentParser()
        add_bus_arguments(parser, filter_arg=True)

        parsed_args = parser.parse_args(
            [
                "--channel",
                "0",
                "--interface",
                "vector",
                "--timing",
                "f_clock=8000000",
                "brp=4",
                "tseg1=11",
                "tseg2=4",
                "sjw=2",
                "nof_samples=3",
                "--filter",
                "100:7FF",
                "200~7F0",
                "--bus-kwargs",
                "app_name=MyApp",
                "serial=1234",
            ]
        )

        self.assertEqual(parsed_args.channel, "0")
        self.assertEqual(parsed_args.interface, "vector")
        self.assertEqual(parsed_args.timing.f_clock, 8000000)
        self.assertEqual(parsed_args.timing.brp, 4)
        self.assertEqual(parsed_args.timing.tseg1, 11)
        self.assertEqual(parsed_args.timing.tseg2, 4)
        self.assertEqual(parsed_args.timing.sjw, 2)
        self.assertEqual(parsed_args.timing.nof_samples, 3)
        self.assertEqual(len(parsed_args.can_filters), 2)
        self.assertEqual(parsed_args.can_filters[0]["can_id"], 0x100)
        self.assertEqual(parsed_args.can_filters[0]["can_mask"], 0x7FF)
        self.assertEqual(parsed_args.can_filters[1]["can_id"], 0x200 | 0x20000000)
        self.assertEqual(parsed_args.can_filters[1]["can_mask"], 0x7F0 & 0x20000000)
        self.assertEqual(parsed_args.bus_kwargs["app_name"], "MyApp")
        self.assertEqual(parsed_args.bus_kwargs["serial"], 1234)

    @patch("can.Bus")
    def test_create_bus_from_namespace(self, mock_bus):
        namespace = argparse.Namespace(
            test_channel="vcan0",
            test_interface="virtual",
            test_bitrate=500000,
            test_data_bitrate=2000000,
            test_fd=True,
            test_can_filters=[{"can_id": 0x100, "can_mask": 0x7FF}],
            test_bus_kwargs={"app_name": "MyApp", "serial": 1234},
        )

        create_bus_from_namespace(namespace, prefix="test")

        mock_bus.assert_called_once_with(
            channel="vcan0",
            interface="virtual",
            bitrate=500000,
            data_bitrate=2000000,
            fd=True,
            can_filters=[{"can_id": 0x100, "can_mask": 0x7FF}],
            app_name="MyApp",
            serial=1234,
            single_handle=True,
        )

    @patch("can.Bus")
    def test_create_bus_from_namespace_no_prefix(self, mock_bus):
        namespace = argparse.Namespace(
            channel="vcan0",
            interface="virtual",
            bitrate=500000,
            data_bitrate=2000000,
            fd=True,
            can_filters=[{"can_id": 0x100, "can_mask": 0x7FF}],
            bus_kwargs={"app_name": "MyApp", "serial": 1234},
        )

        create_bus_from_namespace(namespace)

        mock_bus.assert_called_once_with(
            channel="vcan0",
            interface="virtual",
            bitrate=500000,
            data_bitrate=2000000,
            fd=True,
            can_filters=[{"can_id": 0x100, "can_mask": 0x7FF}],
            app_name="MyApp",
            serial=1234,
            single_handle=True,
        )


if __name__ == "__main__":
    unittest.main()