AthenaOutputStream_test.cxx

Go to the documentation of this file.

/** @file AthenaOutputStream_test.cxx
 * @brief  unit test for AtRndSvc
 *
 * @author Paolo Calafiura <[email protected]> -ATLAS Collaboration
 * $Id: AthenaOutputStream_test.cxx,v 1.1.1.1 2004/09/28 01:51:05 liwd Exp $
 **/
 
#include <cassert>
#include <iostream>
#include <vector>
#include "TestTools/initGaudi.h"
#include "ToyConversion/FooBar.h"
#include "GaudiKernel/IAlgorithm.h"
#include "GaudiKernel/IAlgManager.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IDataSelector.h"
#include "GaudiKernel/SmartIF.h"
#include "AthenaKernel/IClassIDSvc.h"
#include "BesServices/AthenaOutputStream.h"
#include "StoreGate/StoreGateSvc.h"
 
using std::cerr;
using std::cout;
using std::endl;
using std::vector;
using namespace Athena_test;
 
int main() {
  cout << "*** AthenaOutputStream_test starts ***" <<endl;
  ISvcLocator* pSvcLoc(0);
  if (!initGaudi("AthenaOutputStream_test.txt", pSvcLoc)) {
    cerr << "This test can not be run" << endl;
    return 0;
  }  
  assert(pSvcLoc);
 
  //locate necessary stuff
  const bool CREATEIF(true);
  IClassIDSvc* pCLIDSvc(0);
  assert( (pSvcLoc->service("ClassIDSvc", pCLIDSvc, CREATEIF)).isSuccess() );
  assert( pCLIDSvc );
 
  StoreGateSvc* pStore(0);
  assert( (pSvcLoc->service("StoreGateSvc", pStore, CREATEIF)).isSuccess() );
  assert( pStore );
 
  SmartIF<IAlgManager> algMan(IID_IAlgManager, pSvcLoc);
  assert( algMan.isValid() );
  IAlgorithm* pAlg(0);
  assert( (algMan->createAlgorithm( "AthenaOutputStream", "AthenaOutputStream", pAlg)).isSuccess() );
 
  assert( (pAlg->sysInitialize()).isSuccess() );
  assert( (pAlg->initialize()).isSuccess() );
 
  assert( (pStore->record(new Foo(), "uno")).isSuccess());
  assert( (pStore->record(new Foo(), "due")).isSuccess());
  assert( (pStore->record(new Bar(), "uno")).isSuccess());
  assert( (pStore->record(new Bar(), "due")).isSuccess());
  
  AthenaOutputStream* pStream(dynamic_cast<AthenaOutputStream*>(pAlg));
  assert( pStream );
 
  //fill the vector of selected objects
  pStream->collectAllObjects();
 
  //  cout << pStream->selectedObjects()->end() - 
  //    pStream->selectedObjects()->begin() <<endl;
  // verify that we got the right objects in the list
  //  this of course depends on AthenaOutputStream_test.txt
  assert( 4 == (pStream->selectedObjects()->end() - 
        pStream->selectedObjects()->begin()) );
  
  pStream->clearSelection();
  assert( 0 == (pStream->selectedObjects()->end() - 
        pStream->selectedObjects()->begin()) );
 
 
  //all done
  cout << "*** AthenaOutputStream_test OK ***" <<endl;
  return 0;
}