From 0e39222fa2f83b8e2d985ff808a583dc266d60b0 Mon Sep 17 00:00:00 2001 From: Vladimir Byko-Ianko Date: Fri, 31 Jul 2020 16:57:19 +0300 Subject: [PATCH] [tracking] Tests on crash fix in case of Protocol::DecodeHeader({}). --- tracking/tracking_tests/protocol_test.cpp | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/tracking/tracking_tests/protocol_test.cpp b/tracking/tracking_tests/protocol_test.cpp index 631013e5f8..15c9aa7872 100644 --- a/tracking/tracking_tests/protocol_test.cpp +++ b/tracking/tracking_tests/protocol_test.cpp @@ -27,9 +27,23 @@ UNIT_TEST(Protocol_DecodeHeader) TEST_EQUAL(packet.size(), 7, ()); TEST_EQUAL(Protocol::PacketType(packet[0]), Protocol::PacketType::CurrentAuth, ()); - auto header = Protocol::DecodeHeader(packet); - CHECK_EQUAL(header.first, Protocol::PacketType::CurrentAuth, ()); - CHECK_EQUAL(header.second, id_str.size(), ()); + { + auto header = Protocol::DecodeHeader(packet); + TEST_EQUAL(header.first, Protocol::PacketType::CurrentAuth, ()); + TEST_EQUAL(header.second, id_str.size(), ()); + } + + { + auto header = Protocol::DecodeHeader({}); + TEST_EQUAL(header.first, Protocol::PacketType::Error, ()); + TEST_EQUAL(header.second, 0, ()); + } + + { + auto header = Protocol::DecodeHeader({7, 9}); + TEST_EQUAL(header.first, Protocol::PacketType::Error, ()); + TEST_EQUAL(header.second, 2, ()); + } } UNIT_TEST(Protocol_CreateDataPacket)