Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | # Copyright 2020 The Pigweed Authors |
| 3 | # |
| 4 | # Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| 5 | # use this file except in compliance with the License. You may obtain a copy of |
| 6 | # the License at |
| 7 | # |
| 8 | # https://www.apache.org/licenses/LICENSE-2.0 |
| 9 | # |
| 10 | # Unless required by applicable law or agreed to in writing, software |
| 11 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 12 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 13 | # License for the specific language governing permissions and limitations under |
| 14 | # the License. |
| 15 | """Tests creating pw_rpc client.""" |
| 16 | |
| 17 | import unittest |
| 18 | |
Wyatt Hepler | 8ce9013 | 2020-12-03 10:57:20 -0800 | [diff] [blame] | 19 | from pw_status import Status |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 20 | from pw_rpc_protos.internal.packet_pb2 import PacketType, RpcPacket |
Wyatt Hepler | 8ce9013 | 2020-12-03 10:57:20 -0800 | [diff] [blame] | 21 | |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 22 | from pw_rpc import packets |
| 23 | |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 24 | _TEST_REQUEST = RpcPacket(type=PacketType.REQUEST, |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 25 | channel_id=1, |
| 26 | service_id=2, |
| 27 | method_id=3, |
| 28 | payload=RpcPacket(status=321).SerializeToString()) |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 29 | |
| 30 | |
| 31 | class PacketsTest(unittest.TestCase): |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 32 | """Tests for packet encoding and decoding.""" |
Wyatt Hepler | dd4cce9 | 2020-07-17 12:24:23 -0700 | [diff] [blame] | 33 | def test_encode_request(self): |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 34 | data = packets.encode_request((1, 2, 3), RpcPacket(status=321)) |
| 35 | packet = RpcPacket() |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 36 | packet.ParseFromString(data) |
| 37 | |
Wyatt Hepler | dd4cce9 | 2020-07-17 12:24:23 -0700 | [diff] [blame] | 38 | self.assertEqual(_TEST_REQUEST, packet) |
| 39 | |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 40 | def test_encode_response(self): |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 41 | response = RpcPacket(type=PacketType.RESPONSE, |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 42 | channel_id=1, |
| 43 | service_id=2, |
| 44 | method_id=3, |
| 45 | payload=RpcPacket(status=321).SerializeToString()) |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 46 | |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 47 | data = packets.encode_response((1, 2, 3), RpcPacket(status=321)) |
| 48 | packet = RpcPacket() |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 49 | packet.ParseFromString(data) |
| 50 | |
| 51 | self.assertEqual(response, packet) |
| 52 | |
Wyatt Hepler | dd4cce9 | 2020-07-17 12:24:23 -0700 | [diff] [blame] | 53 | def test_encode_cancel(self): |
| 54 | data = packets.encode_cancel((9, 8, 7)) |
| 55 | |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 56 | packet = RpcPacket() |
Wyatt Hepler | dd4cce9 | 2020-07-17 12:24:23 -0700 | [diff] [blame] | 57 | packet.ParseFromString(data) |
| 58 | |
| 59 | self.assertEqual( |
| 60 | packet, |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 61 | RpcPacket(type=PacketType.CANCEL_SERVER_STREAM, |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 62 | channel_id=9, |
| 63 | service_id=8, |
| 64 | method_id=7)) |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 65 | |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 66 | def test_encode_client_error(self): |
| 67 | data = packets.encode_client_error(_TEST_REQUEST, Status.NOT_FOUND) |
| 68 | |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 69 | packet = RpcPacket() |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 70 | packet.ParseFromString(data) |
| 71 | |
| 72 | self.assertEqual( |
| 73 | packet, |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 74 | RpcPacket(type=PacketType.CLIENT_ERROR, |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 75 | channel_id=1, |
| 76 | service_id=2, |
| 77 | method_id=3, |
| 78 | status=Status.NOT_FOUND.value)) |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 79 | |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 80 | def test_decode(self): |
Wyatt Hepler | dd4cce9 | 2020-07-17 12:24:23 -0700 | [diff] [blame] | 81 | self.assertEqual(_TEST_REQUEST, |
| 82 | packets.decode(_TEST_REQUEST.SerializeToString())) |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 83 | |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 84 | def test_for_server(self): |
| 85 | self.assertTrue(packets.for_server(_TEST_REQUEST)) |
| 86 | |
| 87 | self.assertFalse( |
| 88 | packets.for_server( |
Wyatt Hepler | a2970c5 | 2021-02-02 14:47:22 -0800 | [diff] [blame^] | 89 | RpcPacket(type=PacketType.RESPONSE, |
Wyatt Hepler | a56ab15 | 2020-08-12 17:06:31 -0700 | [diff] [blame] | 90 | channel_id=1, |
| 91 | service_id=2, |
| 92 | method_id=3, |
| 93 | payload=RpcPacket(status=321).SerializeToString()))) |
Wyatt Hepler | 0f26235 | 2020-07-29 09:51:27 -0700 | [diff] [blame] | 94 | |
Wyatt Hepler | 5cfde33 | 2020-06-30 08:53:44 -0700 | [diff] [blame] | 95 | |
| 96 | if __name__ == '__main__': |
| 97 | unittest.main() |