test/UnitTests/PDU_Factory.cpp

   1 #include <iostream>
   2 #include "gtest/gtest.h"
   3
   4 #include "KDIS/KDefines.h"
   5 #include "KDIS/Extras/PDU_Factory.h"
   6 #include "KDIS/PDU/Entity_Info_Interaction/Entity_State_PDU.h"
   7
   8 using namespace KDIS;
   9 using namespace UTILS;
  10 using namespace PDU;
  11
  12 class PDU_Factory_Filter_Test : public PDU_Factory_Filter
  13 {
  14 public:
  15
  16     int m_ApplyFilterCalled;
  17     int m_AppplyFilterBeforeDecodingPDUBodyCalled;
  18
  19     KBOOL m_ApplyFilterReturnValue;
  20     KBOOL m_ApplyFilterBeforeDecodingPDUBodyReturnValue;
  21
  22     PDU_Factory_Filter_Test() :
  23         m_ApplyFilterCalled(0),
  24         m_AppplyFilterBeforeDecodingPDUBodyCalled(0),
  25         m_ApplyFilterReturnValue(true),
  26         m_ApplyFilterBeforeDecodingPDUBodyReturnValue(true)
  27     {
  28     };
  29
  30     ~PDU_Factory_Filter_Test() {};
  31
  32     virtual KBOOL ApplyFilter( const KDIS::PDU::Header * H )
  33     {
  34         ++m_ApplyFilterCalled;
  35         return m_ApplyFilterReturnValue;
  36     };
  37
  38     virtual KBOOL ApplyFilterBeforeDecodingPDUBody(const KDIS::PDU::Header * H)
  39     {
  40         ++m_AppplyFilterBeforeDecodingPDUBodyCalled;
  41         return m_ApplyFilterBeforeDecodingPDUBodyReturnValue;
  42     }
  43 };
  44
  45 TEST( PDU_Factory, PDU_Factory_Filter_FiltersAreUsed )
  46 {
  47     PDU_Factory factory;
  48     PDU_Factory_Filter_Test * filter = new PDU_Factory_Filter_Test();
  49     factory.AddFilter( filter );
  50     Entity_State_PDU pdu;
  51     KDataStream stream = pdu.Encode();
  52
  53     ASSERT_EQ(0, filter->m_ApplyFilterCalled);
  54     ASSERT_EQ(0, filter->m_AppplyFilterBeforeDecodingPDUBodyCalled);
  55
  56     std::unique_ptr<Header> header = factory.Decode(stream);
  57
  58     EXPECT_EQ(1, filter->m_ApplyFilterCalled);
  59     EXPECT_EQ(1, filter->m_AppplyFilterBeforeDecodingPDUBodyCalled);
  60 }
  61
  62 TEST( PDU_Factory, PDU_Factory_Filter_ApplyFilterFalse_RejectsPDU )
  63 {
  64     PDU_Factory factory;
  65     PDU_Factory_Filter_Test * filter = new PDU_Factory_Filter_Test();
  66     filter->m_ApplyFilterReturnValue = false;
  67     factory.AddFilter(filter);
  68     Entity_State_PDU pdu;
  69     KDataStream stream = pdu.Encode();
  70
  71     std::unique_ptr<Header> header = factory.Decode(stream);
  72     EXPECT_FALSE(header.get());
  73 }
  74
  75 TEST( PDU_Factory, PDU_Factory_Filter_ApplyFilterBeforeDecodingPDUBody_RejectsPDU )
  76 {
  77     PDU_Factory factory;
  78     PDU_Factory_Filter_Test * filter = new PDU_Factory_Filter_Test();
  79     filter->m_ApplyFilterBeforeDecodingPDUBodyReturnValue = false;
  80     factory.AddFilter(filter);
  81     Entity_State_PDU pdu;
  82     KDataStream stream = pdu.Encode();
  83
  84     std::unique_ptr<Header> header = factory.Decode(stream);
  85     EXPECT_FALSE(header.get());
  86 }