Skip to content

Commit 6d6b266

Browse files
authored
fix some issues with message delete service (#30)
* fix some issues with message delete service
1 parent 64cff8a commit 6d6b266

File tree

4 files changed

+21
-19
lines changed

4 files changed

+21
-19
lines changed

agrirouter/auth/response.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@ def verify(self, public_key) -> None:
5757
finally:
5858
self._was_verified = True
5959

60-
6160
@staticmethod
6261
def decode_token(token: Union[str, bytes]) -> AuthorizationToken:
6362
if type(token) == str:

agrirouter/messaging/parameters/service.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -214,15 +214,15 @@ class FeedDeleteParameters(MessageParameters):
214214
def __init__(self,
215215
*,
216216
message_ids: list = None,
217-
receivers: list = None,
217+
senders: list = None,
218218
validity_period: ValidityPeriod = None,
219219
application_message_seq_no: int,
220220
application_message_id: str,
221221
team_set_context_id: str = None,
222222
onboarding_response: BaseOnboardingResonse,
223223
):
224224
self.message_ids = message_ids if message_ids else []
225-
self.receivers = receivers if receivers else []
225+
self.senders = senders if senders else []
226226
self.validity_period = validity_period
227227
super(FeedDeleteParameters, self).__init__(
228228
application_message_seq_no=application_message_seq_no,
@@ -243,17 +243,17 @@ def add_message_ids(self, message_id):
243243
def extend_message_ids(self, message_ids):
244244
self.message_ids.extend(message_ids)
245245

246-
def get_receivers(self):
247-
return deepcopy(self.receivers)
246+
def get_senders(self):
247+
return deepcopy(self.senders)
248248

249-
def set_receivers(self, receivers: list):
250-
self.receivers = receivers
249+
def set_senders(self, senders: list):
250+
self.senders = senders
251251

252-
def add_receivers(self, receiver):
253-
self.receivers.append(receiver)
252+
def add_senders(self, receiver):
253+
self.senders.append(receiver)
254254

255-
def extend_receivers(self, receivers):
256-
self.receivers.extend(receivers)
255+
def extend_senders(self, senders):
256+
self.senders.extend(senders)
257257

258258
def get_validity_period(self):
259259
return self.validity_period

agrirouter/messaging/services/messaging.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
from agrirouter.generated.messaging.request.payload.account.endpoints_pb2 import ListEndpointsQuery
22
from agrirouter.generated.messaging.request.payload.endpoint.capabilities_pb2 import CapabilitySpecification
33
from agrirouter.generated.messaging.request.payload.endpoint.subscription_pb2 import Subscription
4-
from agrirouter.generated.messaging.request.payload.feed.feed_requests_pb2 import MessageConfirm, MessageQuery
4+
from agrirouter.generated.messaging.request.payload.feed.feed_requests_pb2 import MessageConfirm, MessageDelete, \
5+
MessageQuery
56
from agrirouter.generated.messaging.request.request_pb2 import RequestEnvelope
67
from agrirouter.messaging.encode import encode_message
78
from agrirouter.messaging.enums import TechnicalMessageType
@@ -107,16 +108,18 @@ def encode(parameters: FeedDeleteParameters) -> EncodedMessage:
107108
application_message_seq_no=parameters.get_application_message_seq_no(),
108109
team_set_context_id=parameters.get_team_set_context_id(),
109110
mode=RequestEnvelope.Mode.Value("DIRECT"),
110-
technical_message_type=TechnicalMessageType.FEED_CONFIRM.value
111+
technical_message_type=TechnicalMessageType.FEED_DELETE.value
111112
)
112113

113-
message_confirm = MessageConfirm(
114-
message_ids=parameters.get_message_ids()
114+
message_delete = MessageDelete(
115+
message_ids=parameters.get_message_ids(),
116+
senders=parameters.get_senders(),
117+
validity_period=parameters.get_validity_period()
115118
)
116119

117120
message_payload_parameters = MessagePayloadParameters(
118-
type_url=TypeUrl.get_type_url(MessageConfirm),
119-
value=message_confirm.SerializeToString()
121+
type_url=TypeUrl.get_type_url(MessageDelete),
122+
value=message_delete.SerializeToString()
120123
)
121124

122125
message_content = encode_message(message_header_parameters, message_payload_parameters)

tests/auth_test/test_response.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import pytest
66

77
from agrirouter.auth.response import AuthResponse
8-
from tests.constants import valid_response_signature, valid_response_token, public_key
8+
from tests.constants import valid_response_signature, valid_response_token, AR_PUBLIC_KEY
99

1010

1111
def test_decode_token():
@@ -31,7 +31,7 @@ def test_verify(authorization):
3131

3232
with pytest.raises(PermissionError):
3333
auth_response.is_valid
34-
auth_response.verify(public_key)
34+
auth_response.verify(AR_PUBLIC_KEY)
3535
assert auth_response.is_valid
3636

3737

0 commit comments

Comments
 (0)