Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions smpclient/requests/enumeration_management.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
from smp import enumeration_management as smpem


class _EnumGroupBase:
_ErrorV1 = smpem.EnumManagementErrorV1
_ErrorV2 = smpem.EnumManagementErrorV2


class CountSupportedGroups(smpem.GroupCountRequest, _EnumGroupBase):
_Response = smpem.GroupCountResponse


class ListSupportedGroups(smpem.ListOfGroupsRequest, _EnumGroupBase):
_Response = smpem.ListOfGroupsResponse


class GroupId(smpem.GroupIdRequest, _EnumGroupBase):
_Response = smpem.GroupIdResponse


class GroupDetails(smpem.GroupDetailsRequest, _EnumGroupBase):
_Response = smpem.GroupDetailsResponse
35 changes: 35 additions & 0 deletions tests/test_requests.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from typing import Tuple, Type

import pytest
from smp import enumeration_management as smpem
from smp import error as smperr
from smp import file_management as smpfs
from smp import header as smphdr
Expand All @@ -18,6 +19,12 @@
from smp.user import intercreate as smpic

from smpclient.generics import SMPRequest, TEr1, TEr2, TRep
from smpclient.requests.enumeration_management import (
CountSupportedGroups,
GroupDetails,
GroupId,
ListSupportedGroups,
)
from smpclient.requests.file_management import (
FileClose,
FileDownload,
Expand Down Expand Up @@ -55,6 +62,34 @@
@pytest.mark.parametrize(
"test_tuple",
(
(
smpem.GroupCountRequest(),
CountSupportedGroups(),
smpem.GroupCountResponse,
smpem.EnumManagementErrorV1,
smpem.EnumManagementErrorV2,
),
(
smpem.ListOfGroupsRequest(),
ListSupportedGroups(),
smpem.ListOfGroupsResponse,
smpem.EnumManagementErrorV1,
smpem.EnumManagementErrorV2,
),
(
smpem.GroupIdRequest(),
GroupId(),
smpem.GroupIdResponse,
smpem.EnumManagementErrorV1,
smpem.EnumManagementErrorV2,
),
(
smpem.GroupDetailsRequest(),
GroupDetails(),
smpem.GroupDetailsResponse,
smpem.EnumManagementErrorV1,
smpem.EnumManagementErrorV2,
),
(
smpimg.ImageStatesReadRequest(),
ImageStatesRead(),
Expand Down
Loading