| /* |
| * |
| * Copyright (c) 2014-2017 Nest Labs, Inc. |
| * All rights reserved. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| /** |
| * @file |
| * This file defines an automated test suite for testing |
| * functionalities of the Nest Weave Reliable Messaging Protocol |
| * (WRMP). |
| * |
| */ |
| |
| #ifndef TEST_WRMP_H_ |
| #define TEST_WRMP_H_ |
| |
| #include <Weave/Core/WeaveCore.h> |
| #include <Weave/Core/WeaveServerBase.h> |
| #include <Weave/Profiles/WeaveProfiles.h> |
| |
| #if WEAVE_CONFIG_ENABLE_RELIABLE_MESSAGING |
| using namespace ::nl::Inet; |
| using namespace ::nl::Weave; |
| enum |
| { |
| kWeaveProfile_Test = 101 |
| }; |
| |
| enum |
| { |
| kWeaveTestMessageType_Generate_Response = 1, |
| kWeaveTestMessageType_No_Response = 2, |
| kWeaveTestMessageType_Periodic = 3, |
| kWeaveTestMessageType_Request_Periodic = 4, |
| kWeaveTestMessageType_Request_Throttle = 5, |
| kWeaveTestMessageType_Request_DD = 6, |
| kWeaveTestMessageType_DD_Test = 7, |
| kWeaveTestMessageType_SetDropAck = 8, |
| kWeaveTestMessageType_ClearDropAck = 9, |
| kWeaveTestMessageType_Lost_Ack = 10, |
| kWeaveTestMessageType_RequestCloseEC = 11, |
| kWeaveTestMessageType_CloseEC = 12, |
| kWeaveTestMessageType_AllowDup = 13, |
| kWeaveTestMessageType_DontAllowDup = 14, |
| kWeaveTestMessageType_EchoRequestForDup = 15, |
| kWeaveTestMessageType_Response = 16, |
| }; |
| |
| typedef enum |
| { |
| TEST_PASS = 0, |
| TEST_FAIL = 1, |
| } testStatus_t; |
| |
| class WRMPTestClient |
| { |
| public: |
| WRMPTestClient(); |
| |
| const WeaveFabricState *FabricState; // [READ ONLY] Fabric state object |
| WeaveExchangeManager *ExchangeMgr; // [READ ONLY] Exchange manager object |
| |
| WEAVE_ERROR Init(WeaveExchangeManager *exchangeMgr, uint64_t nodeId, IPAddress nodeAddr, uint16_t port, InterfaceId sendIntfId); |
| WEAVE_ERROR Shutdown(); |
| |
| WEAVE_ERROR SendEchoRequest(WeaveConnection *con, PacketBuffer *payload); |
| WEAVE_ERROR SendEchoRequest(uint64_t nodeId, IPAddress nodeAddr, PacketBuffer *payload); |
| |
| WEAVE_ERROR SendEchoRequest(PacketBuffer *payload); |
| WEAVE_ERROR SendEchoRequest(PacketBuffer *payload, uint16_t sendFlags); |
| typedef void (*EchoFunct)(uint64_t nodeId, IPAddress nodeAddr, PacketBuffer *payload); |
| EchoFunct OnEchoResponseReceived; |
| //WEAVE_ERROR GeneratePeriodicMessage(void); |
| |
| ExchangeContext *ExchangeCtx; // The exchange context for the most recently started exchange. |
| |
| static void HandleResponse(ExchangeContext *ec, const IPPacketInfo *pktInfo, const WeaveMessageInfo *msgInfo, uint32_t profileId, uint8_t msgType, PacketBuffer *payload); |
| |
| }; |
| |
| |
| class WRMPTestServer : public WeaveServerBase |
| { |
| public: |
| WRMPTestServer(); |
| |
| WEAVE_ERROR Init(WeaveExchangeManager *exchangeMgr); |
| WEAVE_ERROR Shutdown(); |
| WEAVE_ERROR GeneratePeriodicMessage(int MaxCount, ExchangeContext *ec); |
| |
| typedef void (*EchoFunct)(uint64_t nodeId, IPAddress nodeAddr, PacketBuffer *payload); |
| EchoFunct OnEchoRequestReceived; |
| |
| static void HandleRcvdMessage(ExchangeContext *ec, const IPPacketInfo *pktInfo, const WeaveMessageInfo *msgInfo, uint32_t profileId, uint8_t msgType, PacketBuffer *payload); |
| static void HandleRcvdMessageForAckDelete(ExchangeContext *ec, const IPPacketInfo *pktInfo, const WeaveMessageInfo *msgInfo, uint32_t profileId, uint8_t msgType, PacketBuffer *payload); |
| |
| static void HandleThrottleFlow(ExchangeContext *ec, const IPPacketInfo *pktInfo, |
| const WeaveMessageInfo *msgInfo, uint32_t profileId, uint8_t msgType, PacketBuffer *payload); |
| }; |
| #endif // WEAVE_CONFIG_ENABLE_RELIABLE_MESSAGING |
| #endif // TEST_WRMP_H_ |