-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
137 additions
and
96 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,48 +1,51 @@ | ||
from unittest import TestCase | ||
|
||
from tmtccmd.util import ObjectIdU32 | ||
from tmtccmd.util.obj_id import ObjectIdU8, ObjectIdU16 | ||
from tmtccmd.util.obj_id import ( | ||
ComponentIdU16, | ||
ComponentIdU32, | ||
ComponentIdU8, | ||
) | ||
|
||
|
||
class TestObjectId(TestCase): | ||
def test_basic(self): | ||
obj_id0 = ObjectIdU32(1, "Some Name") | ||
obj_id0 = ComponentIdU32(1, "Some Name") | ||
self.assertEqual(str(obj_id0), "Object ID Some Name with ID 0x00000001") | ||
self.assertEqual( | ||
obj_id0.__repr__(), "ObjectIdU32(object_id=1, name='Some Name')" | ||
obj_id0.__repr__(), "ComponentIdU32(object_id=1, name='Some Name')" | ||
) | ||
self.assertEqual(obj_id0.as_bytes, bytes([0x00, 0x00, 0x00, 0x01])) | ||
self.assertEqual(obj_id0.as_hex_string, "0x00000001") | ||
self.assertEqual(int(obj_id0), 1) | ||
obj_id1 = ObjectIdU32(1, "Other Name") | ||
obj_id1 = ComponentIdU32(1, "Other Name") | ||
self.assertEqual(obj_id0, obj_id1) | ||
obj_from_raw = ObjectIdU32.from_bytes(obj_id0.as_bytes) | ||
obj_from_raw = ComponentIdU32.from_bytes(obj_id0.as_bytes) | ||
self.assertEqual(obj_from_raw, obj_id0) | ||
with self.assertRaises(ValueError): | ||
ObjectIdU32.from_bytes(bytes()) | ||
ComponentIdU32.from_bytes(bytes()) | ||
with self.assertRaises(ValueError): | ||
ObjectIdU32.from_bytes(bytes([0, 1, 2])) | ||
ComponentIdU32.from_bytes(bytes([0, 1, 2])) | ||
with self.assertRaises(ValueError): | ||
obj_id1.obj_id = -1 | ||
|
||
def test_diff_types(self): | ||
obj_id_u8 = ObjectIdU8(1, "U8 ID 0") | ||
obj_id_u8 = ComponentIdU8(1, "U8 ID 0") | ||
self.assertEqual(obj_id_u8.as_bytes, bytes([1])) | ||
self.assertEqual(obj_id_u8.as_hex_string, "0x01") | ||
self.assertEqual(obj_id_u8.byte_len, 1) | ||
obj_id_u16 = ObjectIdU16(2, "U16 ID 2") | ||
obj_id_u16 = ComponentIdU16(2, "U16 ID 2") | ||
self.assertEqual(obj_id_u16.as_bytes, bytes([0, 2])) | ||
self.assertEqual(obj_id_u16.as_hex_string, "0x0002") | ||
self.assertEqual(obj_id_u16.byte_len, 2) | ||
obj_id_u32 = ObjectIdU32(1, "U32 ID 1") | ||
obj_id_u32 = ComponentIdU32(1, "U32 ID 1") | ||
test_dict = dict() | ||
test_dict.update({obj_id_u8: obj_id_u8.name}) | ||
test_dict.update({obj_id_u16: obj_id_u16.name}) | ||
test_dict.update({obj_id_u32: obj_id_u32.name}) | ||
self.assertEqual(len(test_dict), 3) | ||
obj_id_u8_from_raw = ObjectIdU8.from_bytes(obj_id_u8.as_bytes) | ||
obj_id_u8_from_raw = ComponentIdU8.from_bytes(obj_id_u8.as_bytes) | ||
self.assertEqual(obj_id_u8_from_raw, obj_id_u8) | ||
obj_id_u16_from_raw = ObjectIdU16.from_bytes(obj_id_u16.as_bytes) | ||
obj_id_u16_from_raw = ComponentIdU16.from_bytes(obj_id_u16.as_bytes) | ||
self.assertEqual(obj_id_u16_from_raw, obj_id_u16) | ||
obj_id_u32_from_raw = ObjectIdU32.from_bytes(obj_id_u32.as_bytes) | ||
obj_id_u32_from_raw = ComponentIdU32.from_bytes(obj_id_u32.as_bytes) | ||
self.assertEqual(obj_id_u32_from_raw, obj_id_u32) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,16 +1,20 @@ | ||
from tmtccmd.util.obj_id import ObjectIdDictT, ObjectIdU32 | ||
from tmtccmd.util.obj_id import ComponentIdMapping, ObjectIdU32 | ||
|
||
|
||
INVALID_ID = bytes([0xFF, 0xFF, 0xFF, 0xFF]) | ||
|
||
|
||
def get_core_object_ids() -> ObjectIdDictT: | ||
def get_base_component_id_mapping() -> ComponentIdMapping: | ||
"""These are the object IDs for the tmtccmd core. The core will usually take care of | ||
inserting these into the object manager during the program initialization. | ||
:return Dictionary of the core object IDs | ||
""" | ||
invalid_id = ObjectIdU32.from_bytes(obj_id_as_bytes=INVALID_ID) | ||
invalid_id = ObjectIdU32.from_bytes(raw=INVALID_ID) | ||
invalid_id.name = "Invalid ID" | ||
object_id_dict = {INVALID_ID: invalid_id} | ||
return object_id_dict | ||
|
||
|
||
def get_core_object_ids() -> ComponentIdMapping: | ||
return get_base_component_id_mapping() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,2 +1,14 @@ | ||
from .obj_id import ObjectIdU32, ObjectIdU16, ObjectIdU8, ObjectIdBase, ObjectIdDictT | ||
from .obj_id import ( | ||
ComponentIdU32, | ||
ComponentIdU16, | ||
ComponentIdU8, | ||
ComponentIdBase, | ||
ComponentIdMapping, | ||
ObjectIdU32, | ||
ObjectIdU16, | ||
ObjectIdU8, | ||
ObjectIdBase, | ||
ObjectIdMapping, | ||
ObjectIdDictT, | ||
) | ||
from .retval import RetvalDictT |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.