Skip to content

Commit

Permalink
fix some issues with message delete service (#30)
Browse files Browse the repository at this point in the history
* fix some issues with message delete service
  • Loading branch information
oliverrahner authored Feb 8, 2022
1 parent 64cff8a commit 6d6b266
Show file tree
Hide file tree
Showing 4 changed files with 21 additions and 19 deletions.
1 change: 0 additions & 1 deletion agrirouter/auth/response.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,6 @@ def verify(self, public_key) -> None:
finally:
self._was_verified = True


@staticmethod
def decode_token(token: Union[str, bytes]) -> AuthorizationToken:
if type(token) == str:
Expand Down
20 changes: 10 additions & 10 deletions agrirouter/messaging/parameters/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -214,15 +214,15 @@ class FeedDeleteParameters(MessageParameters):
def __init__(self,
*,
message_ids: list = None,
receivers: list = None,
senders: list = None,
validity_period: ValidityPeriod = None,
application_message_seq_no: int,
application_message_id: str,
team_set_context_id: str = None,
onboarding_response: BaseOnboardingResonse,
):
self.message_ids = message_ids if message_ids else []
self.receivers = receivers if receivers else []
self.senders = senders if senders else []
self.validity_period = validity_period
super(FeedDeleteParameters, self).__init__(
application_message_seq_no=application_message_seq_no,
Expand All @@ -243,17 +243,17 @@ def add_message_ids(self, message_id):
def extend_message_ids(self, message_ids):
self.message_ids.extend(message_ids)

def get_receivers(self):
return deepcopy(self.receivers)
def get_senders(self):
return deepcopy(self.senders)

def set_receivers(self, receivers: list):
self.receivers = receivers
def set_senders(self, senders: list):
self.senders = senders

def add_receivers(self, receiver):
self.receivers.append(receiver)
def add_senders(self, receiver):
self.senders.append(receiver)

def extend_receivers(self, receivers):
self.receivers.extend(receivers)
def extend_senders(self, senders):
self.senders.extend(senders)

def get_validity_period(self):
return self.validity_period
Expand Down
15 changes: 9 additions & 6 deletions agrirouter/messaging/services/messaging.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
from agrirouter.generated.messaging.request.payload.account.endpoints_pb2 import ListEndpointsQuery
from agrirouter.generated.messaging.request.payload.endpoint.capabilities_pb2 import CapabilitySpecification
from agrirouter.generated.messaging.request.payload.endpoint.subscription_pb2 import Subscription
from agrirouter.generated.messaging.request.payload.feed.feed_requests_pb2 import MessageConfirm, MessageQuery
from agrirouter.generated.messaging.request.payload.feed.feed_requests_pb2 import MessageConfirm, MessageDelete, \
MessageQuery
from agrirouter.generated.messaging.request.request_pb2 import RequestEnvelope
from agrirouter.messaging.encode import encode_message
from agrirouter.messaging.enums import TechnicalMessageType
Expand Down Expand Up @@ -107,16 +108,18 @@ def encode(parameters: FeedDeleteParameters) -> EncodedMessage:
application_message_seq_no=parameters.get_application_message_seq_no(),
team_set_context_id=parameters.get_team_set_context_id(),
mode=RequestEnvelope.Mode.Value("DIRECT"),
technical_message_type=TechnicalMessageType.FEED_CONFIRM.value
technical_message_type=TechnicalMessageType.FEED_DELETE.value
)

message_confirm = MessageConfirm(
message_ids=parameters.get_message_ids()
message_delete = MessageDelete(
message_ids=parameters.get_message_ids(),
senders=parameters.get_senders(),
validity_period=parameters.get_validity_period()
)

message_payload_parameters = MessagePayloadParameters(
type_url=TypeUrl.get_type_url(MessageConfirm),
value=message_confirm.SerializeToString()
type_url=TypeUrl.get_type_url(MessageDelete),
value=message_delete.SerializeToString()
)

message_content = encode_message(message_header_parameters, message_payload_parameters)
Expand Down
4 changes: 2 additions & 2 deletions tests/auth_test/test_response.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import pytest

from agrirouter.auth.response import AuthResponse
from tests.constants import valid_response_signature, valid_response_token, public_key
from tests.constants import valid_response_signature, valid_response_token, AR_PUBLIC_KEY


def test_decode_token():
Expand All @@ -31,7 +31,7 @@ def test_verify(authorization):

with pytest.raises(PermissionError):
auth_response.is_valid
auth_response.verify(public_key)
auth_response.verify(AR_PUBLIC_KEY)
assert auth_response.is_valid


Expand Down

0 comments on commit 6d6b266

Please sign in to comment.