/*
 * Boost.Extension / construction test case
 *
 * (C) Copyright Jeremy Pack 2007
 * Distributed under the Boost Software License, Version 1.0. (See
 * accompanying file LICENSE_1_0.txt or copy at
 * http://www.boost.org/LICENSE_1_0.txt)
 *
 * See http://www.boost.org/ for latest version.
 */


#include <boost/extension/factory.hpp>
#include <boost/extension/factory_map.hpp>
#define BOOST_TEST_MAIN
#include <boost/test/unit_test.hpp>
#include <memory>
#include <map>

using namespace boost::extensions;


class Automobile {
public:
  virtual ~Automobile() {}
  Automobile(int speed) : speed_(speed) {}
  virtual int getSpeed() {return speed_;}
protected:
  int speed_;
};

class Van : virtual public Automobile {
public:
  virtual ~Van() {}
  Van(int speed) : Automobile(speed) {}
  virtual int getSpeed() {return speed_ / 2;}
};

class PickupTruck : virtual public Automobile {
public:
  virtual ~PickupTruck() {}
  PickupTruck(int speed) : Automobile(speed) {}
  virtual int getSpeed() {return speed_ / 3;}
};

class Racecar : virtual public Automobile {
public:
  virtual ~Racecar() {}
  Racecar(int speed) : Automobile(speed) {}
  virtual int getSpeed() {return 3 * speed_;}
};

class RacingVan : public Racecar, public Van
{
public:
  virtual ~RacingVan() {}
  RacingVan(int speed) : Automobile(speed),
  Racecar(speed), Van(speed) {}
  virtual int getSpeed() {return 2 * speed_;}
};




BOOST_AUTO_TEST_CASE(factory_argless) {
  factory<Automobile, int> f;
  f.set<PickupTruck>();
  BOOST_CHECK(f.is_valid());
  std::auto_ptr<Automobile> pickup(f.create(90));
  BOOST_CHECK_EQUAL(pickup->getSpeed(), 30);
}

BOOST_AUTO_TEST_CASE(map_argless)
{
  std::map<std::string, factory<Automobile, int> > m;
  m["A van"].set<Van>();
  m["A basic automobile"].set<Automobile>();
  m["A pickup truck"].set<PickupTruck>();
  std::auto_ptr<Automobile> van(m["A van"].create(30));
  BOOST_CHECK_EQUAL(van->getSpeed(), 15);
  BOOST_CHECK_EQUAL
    (m["An unloaded car!"].create(30),
     (Automobile*)0);
}
/*
namespace boost { namespace extensions {
template <>
Automobile * create_function<Automobile, Racecar>::create()
{
  return new Racecar(50);
}
}} // namespace boost::extensions

BOOST_AUTO_TEST_CASE(factory_template)
{
  factory<Automobile, int> f;
  f.set<Racecar>();
  BOOST_CHECK(f.is_valid());
  std::auto_ptr<Automobile> racecar(f.create(60));
  BOOST_CHECK_EQUAL(racecar->getSpeed(), 180);
}*/

BOOST_AUTO_TEST_CASE(factory_map_single_param)
{
  factory_map m;
  m.get<Automobile, std::string, int>()["A pickup truck"].set<PickupTruck>();
  m.get<Automobile, std::string, int>()["A racing van!"].set<RacingVan>();
  std::auto_ptr<Automobile> pickup
    (m.get<Automobile, std::string, int>()["A pickup truck"].create(60));
  BOOST_CHECK_EQUAL(pickup->getSpeed(), 20);
  std::auto_ptr<Automobile> racingVan
    (m.get<Automobile, std::string, int>()["A racing van!"].create(60));
  BOOST_CHECK_EQUAL(racingVan->getSpeed(), 120);
  std::auto_ptr<Automobile> car_ptr
    (m.get<Automobile, std::string, int>()["A nonexistent car!"].create(30));
  BOOST_CHECK_EQUAL(car_ptr.get(), (Automobile*)0);
}