taskinterface.gtest.cpp

Go to the documentation of this file.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use these files except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Copyright 2012 - Gonzalo Iglesias, Adrià de Gispert, William Byrne

#include <googletesting.h>

#ifndef GMAINTEST
#include "main.custom_assert.hpp"
#include "main.logger.hpp"
#endif

#include "taskinterface.hpp"

namespace googletesting {

class DataTest1 {
 public:
  bool modifiedbytask1;
  bool modifiedbytask2;
  bool modifiedbytask3;
};

template<class Data>
class Task1: public uu::TaskInterface<Data> {
 public:
  bool run ( Data& d ) {
    d.modifiedbytask1 = true;
    return false;
  };

};

template<class Data>
class Task2: public uu::TaskInterface<Data> {
 public:
  bool run ( Data& d ) {
    d.modifiedbytask2 = true;
    return false;
  };
};

template<class Data>
class Task3: public uu::TaskInterface<Data> {
 public:
  bool run ( Data& d ) {
    d.modifiedbytask3 = true;
    return false;
  };
};

TEST ( TaskInterface, chainoftasks1 ) {
  Task1<DataTest1> t1;
  t1.appendTask ( new Task2<DataTest1> );
  t1.appendTask ( new Task3<DataTest1> );
  DataTest1 d;
  d.modifiedbytask1 = d.modifiedbytask2 = d.modifiedbytask3 = false;
  t1.chainrun ( d );
  EXPECT_EQ ( d.modifiedbytask1, true );
  EXPECT_EQ ( d.modifiedbytask2, true );
  EXPECT_EQ ( d.modifiedbytask3, true );
}

template<class Data>
class Task2b: public uu::TaskInterface<Data> {
 public:
  bool run ( Data& d ) {
    d.modifiedbytask2 = true;
    return true;
  };
};

TEST ( TaskInterface, chainoftasks2 ) {
  Task1<DataTest1> t1;
  t1.appendTask ( new Task2b<DataTest1> );
  t1.appendTask ( new Task3<DataTest1> );
  DataTest1 d;
  d.modifiedbytask1 = d.modifiedbytask2 = d.modifiedbytask3 = false;
  bool result = t1.chainrun ( d );
  EXPECT_EQ ( result, true );
  EXPECT_EQ ( d.modifiedbytask1, true );
  EXPECT_EQ ( d.modifiedbytask2, true );
  EXPECT_EQ ( d.modifiedbytask3, false );
}

TEST ( TaskInterface, chainoftasks3 ) {
  Task1<DataTest1> t1;
  t1 ( new Task2b<DataTest1> )
  ( new Task3<DataTest1> );
  DataTest1 d;
  d.modifiedbytask1 = d.modifiedbytask2 = d.modifiedbytask3 = false;
  bool result = t1.chainrun ( d );
  EXPECT_EQ ( result, true );
  EXPECT_EQ ( d.modifiedbytask1, true );
  EXPECT_EQ ( d.modifiedbytask2, true );
  EXPECT_EQ ( d.modifiedbytask3, false );
}


TEST ( TaskInterface, chainoftasks4 ) {
  Task1<DataTest1> t1;
  t1 ( NULL )
  ( new Task3<DataTest1> );
  DataTest1 d;
  d.modifiedbytask1 = d.modifiedbytask2 = d.modifiedbytask3 = false;
  bool result = t1.chainrun ( d );
  EXPECT_EQ ( result, false );
  EXPECT_EQ ( d.modifiedbytask1, true );
  EXPECT_EQ ( d.modifiedbytask2, false );
  EXPECT_EQ ( d.modifiedbytask3, true );
}

class DataTest2 {
 public:
  unsigned idx;
  std::vector<unsigned> v;
};

template<class Data>
class Task4: public uu::TaskInterface<Data> {
 public:
  bool run ( Data& d ) {
    d.v.push_back ( d.idx );
    return false;
  };

};

TEST ( TaskInterface, idx ) {
  Task4<DataTest2> t4;
  DataTest2 d;
  for ( unsigned k = 0; k < 5; k += 2 ) {
    d.idx = k;
    t4.run ( d );
  }
  ASSERT_EQ ( d.v.size(), 3 );
  EXPECT_EQ ( d.v[0], 0 );
  EXPECT_EQ ( d.v[1], 2 );
  EXPECT_EQ ( d.v[2], 4 );
}

template<class Data>
class PostEditTask: public uu::TaskInterface<Data> {
 private:
  uu::TaskInterface<Data> *fullsystem_;
 public:
  PostEditTask ( uu::TaskInterface<Data> *fs ) : fullsystem_ ( fs ) {};
  bool run ( Data& d ) {
    //Run first time on sentence x
    fullsystem_->chainrun ( d );
    // Do stuff...
    //Run second time on sentence x. d now contains certain modifications that could change the behaviour and output of the fullsystem.
    fullsystem_->chainrun ( d );
    return false;
  };
  inline ~PostEditTask() {
    if ( fullsystem_ ) delete fullsystem_;
  };

};

TEST ( TaskInterface, idx_chainrun ) {
  PostEditTask<DataTest2> t5 ( new Task4<DataTest2>() );
  DataTest2 d;
  for ( unsigned k = 0; k < 3; k += 2 ) {
    d.idx = k;
    t5.run ( d );
  }
  ASSERT_EQ ( d.v.size(), 4 );
  EXPECT_EQ ( d.v[0], 0 );
  EXPECT_EQ ( d.v[1], 0 );
  EXPECT_EQ ( d.v[2], 2 );
  EXPECT_EQ ( d.v[3], 2 );
}

};

#ifndef GMAINTEST

int main ( int argc, char **argv ) {
  ::testing::InitGoogleTest ( &argc, argv );
  return RUN_ALL_TESTS();
};

#endif