bit-serializer-test.cc

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2020 Universita' di Firenze, Italy
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Author: Tommaso Pecorella <tommaso.pecorella@unifi.it>
 */
#include <cstdarg>
#include <iostream>
#include <sstream>
#include "ns3/test.h"
#include "ns3/bit-serializer.h"
#include "ns3/bit-deserializer.h"

using namespace ns3;

class BitSerializerTest : public TestCase
{
public:
  virtual void DoRun (void);
  BitSerializerTest ();
};

BitSerializerTest::BitSerializerTest ()
  : TestCase ("BitSerializer")
{}

void BitSerializerTest::DoRun ()
{
  BitSerializer testBitSerializer1;

  testBitSerializer1.PushBits (0x55, 7);
  testBitSerializer1.PushBits (0x7, 3);
  testBitSerializer1.PushBits (0x0, 2);

  std::vector<uint8_t> result = testBitSerializer1.GetBytes ();
  NS_TEST_EXPECT_MSG_EQ ((result[0] == 0xab) && (result[1] == 0xc0), true,
                         "Incorrect serialization " << std::hex << +result[0] << +result[1] << " instead of " << 0xab << " " << 0xc0 << std::dec);

  BitSerializer testBitSerializer2;

  testBitSerializer2.PushBits (0x55, 7);
  testBitSerializer2.PushBits (0x7, 3);
  testBitSerializer2.PushBits (0x0, 2);

  testBitSerializer2.InsertPaddingAtEnd (false);

  result = testBitSerializer2.GetBytes ();
  NS_TEST_EXPECT_MSG_EQ ((result[0] == 0x0a) && (result[1] == 0xbc), true,
                         "Incorrect serialization " << std::hex << +result[0] << +result[1] << " instead of " << 0x0a << " " << 0xbc << std::dec);
}

class BitDeserializerTest : public TestCase
{
public:
  virtual void DoRun (void);
  BitDeserializerTest ();
};

BitDeserializerTest::BitDeserializerTest ()
  : TestCase ("BitDeserializer")
{}

void BitDeserializerTest::DoRun ()
{
  uint16_t nibble1;
  uint16_t nibble2;
  uint16_t nibble3;
  bool result;

  BitDeserializer testBitDeserializer1;
  uint8_t test1[2];
  test1[0] = 0xab;
  test1[1] = 0xc0;

  testBitDeserializer1.PushBytes (test1, 2);
  nibble1 = testBitDeserializer1.GetBits (7);
  nibble2 = testBitDeserializer1.GetBits (3);
  nibble3 = testBitDeserializer1.GetBits (2);
  result = (nibble1 == 0x55) && (nibble2 == 0x7) && (nibble3 == 0x0);

  NS_TEST_EXPECT_MSG_EQ (result, true,
                         "Incorrect deserialization " << std::hex << nibble1 << " " << nibble2 << " " << nibble3 <<
                         " << instead of " << " " << 0x55 << " " << 0x7 << " " << 0x0 << std::dec);

  BitDeserializer testBitDeserializer2;
  std::vector<uint8_t> test2;
  test2.push_back (0xab);
  test2.push_back (0xc0);

  testBitDeserializer2.PushBytes (test2);
  nibble1 = testBitDeserializer2.GetBits (7);
  nibble2 = testBitDeserializer2.GetBits (3);
  nibble3 = testBitDeserializer2.GetBits (2);

  result = (nibble1 == 0x55) && (nibble2 == 0x7) && (nibble3 == 0x0);

  NS_TEST_EXPECT_MSG_EQ (result, true,
                         "Incorrect deserialization " << std::hex << nibble1 << " " << nibble2 << " " << nibble3 <<
                         " << instead of " << " " << 0x55 << " " << 0x7 << " " << 0x0 << std::dec);

}

class BitSerializerTestSuite : public TestSuite
{
public:
  BitSerializerTestSuite ();
};


BitSerializerTestSuite::BitSerializerTestSuite ()
  : TestSuite ("bit-serializer", UNIT)
{
  AddTestCase (new BitSerializerTest, TestCase::QUICK);
  AddTestCase (new BitDeserializerTest, TestCase::QUICK);
}

static BitSerializerTestSuite g_bitSerializerTest;