src/connectivity/wlan/lib/mlme/cpp/tests/tim_element_unittest.cc

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <gtest/gtest.h>
#include <wlan/common/tim_element.h>

namespace wlan {
namespace common {

static TimHeader tim_header(uint8_t offset) {
  TimHeader hdr;
  hdr.dtim_count = 1;
  hdr.dtim_period = 2;
  hdr.bmp_ctrl.set_group_traffic_ind(0);
  hdr.bmp_ctrl.set_offset(offset);
  return hdr;
}

TEST(TimElement, IsTrafficBuffered) {
  const uint8_t bitmap[] = {0x12};

  EXPECT_FALSE(IsTrafficBuffered(0, tim_header(0), bitmap));
  EXPECT_TRUE(IsTrafficBuffered(1, tim_header(0), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(2, tim_header(0), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(3, tim_header(0), bitmap));
  EXPECT_TRUE(IsTrafficBuffered(4, tim_header(0), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(5, tim_header(0), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(100, tim_header(0), bitmap));

  // Offset of 1 means "skip 16 bits"
  EXPECT_FALSE(IsTrafficBuffered(15, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(16, tim_header(1), bitmap));
  EXPECT_TRUE(IsTrafficBuffered(17, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(18, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(19, tim_header(1), bitmap));
  EXPECT_TRUE(IsTrafficBuffered(20, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(21, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(22, tim_header(1), bitmap));
  EXPECT_FALSE(IsTrafficBuffered(100, tim_header(1), bitmap));
}

TEST(TimElement, FindAndParseOk) {
  // Present & valid TIM
  std::vector<uint8_t> buf({0, 3, 'f', 'o', 'o',    // SSID
                            5, 5, 1, 2, 3, 10, 20,  // TIM
                            7, 3, 'A', 'B', 'C'});  // Country
  auto tim = FindAndParseTim(buf);
  ASSERT_TRUE(tim);
  EXPECT_EQ(tim->header.dtim_count, 1);
  EXPECT_EQ(tim->header.dtim_period, 2);
  EXPECT_EQ(tim->header.bmp_ctrl.val(), 3);
  ASSERT_EQ(tim->bitmap.size(), 2u);
  ASSERT_EQ(tim->bitmap[0], 10);
  ASSERT_EQ(tim->bitmap[1], 20);
}

TEST(TimElement, FindAndParseAbsent) {
  auto tim = FindAndParseTim(std::vector<uint8_t>({0, 3, 'f', 'o', 'o', 7, 3, 'A', 'B', 'C'}));
  ASSERT_FALSE(tim);
}

TEST(TimElement, FindAndParseInvalid) {
  auto tim =
      FindAndParseTim(std::vector<uint8_t>({0, 3, 'f', 'o', 'o', 5, 2, 1, 2, 7, 3, 'A', 'B', 'C'}));
  ASSERT_FALSE(tim);
}

}  // namespace common
}  // namespace wlan