src/SALOMEDS/Test/SALOMEDSTest_AttributeTarget.cxx

   1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 /*!
  24  * Check all methods of SALOMEDS_AttributeTarget
  25  * Use code of SALOMEDS_AttributeTarget.cxx
  26  */
  27 void SALOMEDSTest::testAttributeTarget()
  28 {
  29   //Create or find the Study manager
  30   _PTR(StudyManager) sm ( new SALOMEDS_StudyManager(_sm) );
  31
  32   CPPUNIT_ASSERT(sm);
  33
  34   //Create a new study
  35   _PTR(Study) study = sm->NewStudy("Test");
  36
  37   CPPUNIT_ASSERT(study);
  38
  39   //Create Study Builder
  40   _PTR(StudyBuilder) studyBuilder = study->NewBuilder();
  41
  42   CPPUNIT_ASSERT(studyBuilder);
  43
  44   //Create a SObject with entry 0:1:1
  45   _PTR(SObject) so = study->CreateObjectID("0:1:1");
  46
  47   CPPUNIT_ASSERT(so);
  48
  49   //Create an attribute AttributeTarget
  50   _PTR(AttributeTarget) _attr = studyBuilder->FindOrCreateAttribute(so, "AttributeTarget");
  51
  52   //Check the attribute creation
  53   CPPUNIT_ASSERT(_attr);
  54
  55   //Check method Get
  56   CPPUNIT_ASSERT(_attr->Get().size() == 0);
  57
  58   _PTR(SObject) so1 = study->CreateObjectID("0:1:2");
  59   studyBuilder->Addreference(so1, so);
  60
  61   _PTR(SObject) so2 = study->CreateObjectID("0:1:3");
  62   studyBuilder->Addreference(so2, so1);
  63
  64   //Check method Add
  65   _attr->Add(so1);
  66
  67   _attr->Add(so2);
  68
  69   //Check method Get
  70   std::vector< _PTR(SObject) > v = _attr->Get();
  71
  72   CPPUNIT_ASSERT(v.size() == 2);
  73
  74   CPPUNIT_ASSERT(v[0]->GetID() == "0:1:2");
  75
  76   CPPUNIT_ASSERT(v[1]->GetID() == "0:1:3");
  77
  78   //Check method Remove
  79   _attr->Remove(so1);
  80
  81   v = _attr->Get();
  82
  83   CPPUNIT_ASSERT(v.size() == 1);
  84
  85   CPPUNIT_ASSERT(v[0]->GetID() == "0:1:3");
  86
  87   sm->Close(study);
  88 }
  89
  90
  91