TAO/tests/Sequence_Unit_Tests/unbounded_sequence_cdr_ut.cpp

   1 /**
   2  * @file
   3  *
   4  * @brief Unit test for unbounded sequences of object references.
   5  *
   6  * @author Carlos O'Ryan
   7  */
   8 #include "testing_object_reference_traits.hpp"
   9 #include "tao/Object_Reference_Traits_T.h"
  10 #include "testing_allocation_traits.hpp"
  11 #include "testing_range_checking.hpp"
  12
  13 #include "mock_reference.hpp"
  14
  15 #include "tao/Unbounded_Value_Sequence_T.h"
  16 #include "tao/Unbounded_Octet_Sequence_T.h"
  17 #include "tao/Unbounded_Object_Reference_Sequence_T.h"
  18 #include "tao/Unbounded_Basic_String_Sequence_T.h"
  19 #include "tao/Unbounded_BD_String_Sequence_T.h"
  20 #include "tao/Unbounded_Sequence_CDR_T.h"
  21 #include "tao/CDR.h"
  22
  23 #include "test_macros.h"
  24
  25
  26 using namespace TAO_VERSIONED_NAMESPACE_NAME::TAO;
  27
  28 typedef unbounded_object_reference_sequence<mock_reference, mock_reference_var> tested_sequence;
  29
  30 CORBA::Boolean operator<< (TAO_OutputCDR &strm, const tested_sequence &sequence)
  31 {
  32   return TAO::marshal_sequence(strm, sequence);
  33 }
  34
  35 CORBA::Boolean operator>> (TAO_InputCDR &strm, tested_sequence &sequence)
  36 {
  37   return TAO::demarshal_sequence(strm, sequence);
  38 }
  39
  40 struct Tester
  41 {
  42   typedef tested_sequence::value_type value_type;
  43   typedef tested_sequence::const_value_type const_value_type;
  44
  45   typedef tested_sequence::element_traits tested_element_traits;
  46   typedef tested_sequence::allocation_traits tested_allocation_traits;
  47   typedef TAO::details::range_checking<value_type,true> range;
  48
  49   value_type * alloc_and_init_buffer()
  50   {
  51     value_type * buf = tested_sequence::allocbuf(8);
  52     buf[0] = mock_reference::allocate(1);
  53     buf[1] = mock_reference::allocate(4);
  54     buf[2] = mock_reference::allocate(9);
  55     buf[3] = mock_reference::allocate(16);
  56
  57     return buf;
  58   }
  59
  60   int check_values(tested_sequence const & a)
  61   {
  62     CHECK_EQUAL( 1, a[0]->id());
  63     CHECK_EQUAL( 4, a[1]->id());
  64     CHECK_EQUAL( 9, a[2]->id());
  65     CHECK_EQUAL(16, a[3]->id());
  66     return 0;
  67   }
  68
  69   int test_stream()
  70   {
  71     value_type * buffer = alloc_and_init_buffer();
  72
  73     expected_calls s(mock_reference::marshal_calls);
  74     {
  75       tested_sequence a;
  76       a.replace(8, 4, buffer, false);
  77
  78       CHECK_EQUAL(CORBA::ULong(8), a.maximum());
  79       CHECK_EQUAL(CORBA::ULong(4), a.length());
  80       CHECK_EQUAL(buffer, a.get_buffer());
  81       CHECK_EQUAL(false, a.release());
  82       check_values(a);
  83
  84       TAO_OutputCDR stream;
  85       stream << a;
  86       FAIL_RETURN_IF_NOT(s.expect(4), s);
  87     }
  88     tested_sequence::freebuf(buffer);
  89     return 0;
  90   }
  91 };
  92
  93 int ACE_TMAIN(int,ACE_TCHAR*[])
  94 {
  95   int status = 0;
  96
  97   try
  98     {
  99       Tester x;
 100       status += x.test_stream ();
 101     }
 102   catch (const ::CORBA::Exception &ex)
 103     {
 104       ex._tao_print_exception("ERROR : unexpected CORBA exception caugth :");
 105       ++status;
 106     }
 107   return status;
 108 }