mitkNodePredicateSourceTest.cpp

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#include "mitkNodePredicateSource.h"
#include "mitkStandaloneDataStorage.h"
#include "mitkTestingMacros.h"

int mitkNodePredicateSourceTest(int /* argc */, char * /*argv*/ [])
{
  MITK_TEST_BEGIN("Testing nodePredicateSource")

  // Create a DataStorage
  mitk::DataStorage::Pointer myDataStorage(mitk::StandaloneDataStorage::New().GetPointer());
  mitk::DataNode::Pointer godfather, grandMother, mother, daughter;
  godfather = mitk::DataNode::New();
  grandMother = mitk::DataNode::New();
  mother = mitk::DataNode::New();
  daughter = mitk::DataNode::New();

  myDataStorage->Add(godfather);
  myDataStorage->Add(grandMother);
  myDataStorage->Add(mother, grandMother);
  myDataStorage->Add(daughter, mother);
  mitk::NodePredicateSource::Pointer testPredicate = mitk::NodePredicateSource::New(grandMother, false, myDataStorage);
  ;

  MITK_TEST_CONDITION_REQUIRED(true, "Testing instantiation");
  MITK_TEST_CONDITION_REQUIRED(testPredicate->CheckNode(mother), "Node is derivative");
  MITK_TEST_CONDITION_REQUIRED(!testPredicate->CheckNode(godfather), "Node is not derivative");
  MITK_TEST_CONDITION_REQUIRED(!testPredicate->CheckNode(daughter),
                               "Node is derivative but only direct derivatives are wanted");

  testPredicate = mitk::NodePredicateSource::New(grandMother, true, myDataStorage);
  MITK_TEST_CONDITION_REQUIRED(testPredicate->CheckNode(daughter),
                               "Node is not direct derivative and all derivatives are wanted ");
  MITK_TEST_CONDITION_REQUIRED(!testPredicate->CheckNode(grandMother), "Self is not a derivative!");

  MITK_TEST_END();
}