/*********************                                                        */
/*! \file stacking_map_black.h
 ** \verbatim
 ** Original author: Morgan Deters
 ** Major contributors: none
 ** Minor contributors (to current version): none
 ** This file is part of the CVC4 project.
 ** Copyright (c) 2009-2013  New York University and The University of Iowa
 ** See the file COPYING in the top-level source directory for licensing
 ** information.\endverbatim
 **
 ** \brief Black box testing of CVC4::theory::uf::morgan::StackingMap
 **
 ** Black box testing of CVC4::theory::uf::morgan::StackingMap.
 **/

#include <cxxtest/TestSuite.h>

#include "context/context.h"
#include "expr/node.h"
#include "theory/uf/morgan/stacking_map.h"

using namespace CVC4;
using namespace CVC4::theory::uf::morgan;
using namespace CVC4::context;

using namespace std;

/**
 * Test the StackingMap.
 */
class StackingMapBlack : public CxxTest::TestSuite {
  Context* d_ctxt;
  StackingMap<TNode, TNodeHashFunction>* d_map;
  NodeManager* d_nm;
  NodeManagerScope* d_scope;

  Node a, b, c, d, e, f, g;

public:

  void setUp() {
    d_ctxt = new Context;
    d_nm = new NodeManager(d_ctxt, NULL);
    d_scope = new NodeManagerScope(d_nm);
    d_map = new StackingMap<TNode, TNodeHashFunction>(d_ctxt);

    a = d_nm->mkVar("a", d_nm->realType());
    b = d_nm->mkVar("b", d_nm->realType());
    c = d_nm->mkVar("c", d_nm->realType());
    d = d_nm->mkVar("d", d_nm->realType());
    e = d_nm->mkVar("e", d_nm->realType());
    f = d_nm->mkVar("f", d_nm->realType());
    g = d_nm->mkVar("g", d_nm->realType());
  }

  void tearDown() {
    g = Node::null();
    f = Node::null();
    e = Node::null();
    d = Node::null();
    c = Node::null();
    b = Node::null();
    a = Node::null();

    delete d_map;
    delete d_scope;
    delete d_nm;
    delete d_ctxt;
  }

  void testSimpleContextual() {
    TS_ASSERT(d_map->find(a).isNull());
    TS_ASSERT(d_map->find(b).isNull());
    TS_ASSERT(d_map->find(c).isNull());
    TS_ASSERT(d_map->find(d).isNull());
    TS_ASSERT(d_map->find(e).isNull());
    TS_ASSERT(d_map->find(f).isNull());
    TS_ASSERT(d_map->find(g).isNull());

    d_map->set(a, b);

    TS_ASSERT(d_map->find(a) == b);
    TS_ASSERT(d_map->find(b).isNull());
    TS_ASSERT(d_map->find(c).isNull());
    TS_ASSERT(d_map->find(d).isNull());
    TS_ASSERT(d_map->find(e).isNull());
    TS_ASSERT(d_map->find(f).isNull());
    TS_ASSERT(d_map->find(g).isNull());

    d_ctxt->push();
    {
      TS_ASSERT(d_map->find(a) == b);
      TS_ASSERT(d_map->find(b).isNull());
      TS_ASSERT(d_map->find(c).isNull());
      TS_ASSERT(d_map->find(d).isNull());
      TS_ASSERT(d_map->find(e).isNull());
      TS_ASSERT(d_map->find(f).isNull());
      TS_ASSERT(d_map->find(g).isNull());

      d_map->set(c, d);
      d_map->set(f, e);

      TS_ASSERT(d_map->find(a) == b);
      TS_ASSERT(d_map->find(b).isNull());
      TS_ASSERT(d_map->find(c) == d);
      TS_ASSERT(d_map->find(d).isNull());
      TS_ASSERT(d_map->find(e).isNull());
      TS_ASSERT(d_map->find(f) == e);
      TS_ASSERT(d_map->find(g).isNull());

      d_ctxt->push();
      {

        TS_ASSERT(d_map->find(a) == b);
        TS_ASSERT(d_map->find(b).isNull());
        TS_ASSERT(d_map->find(c) == d);
        TS_ASSERT(d_map->find(d).isNull());
        TS_ASSERT(d_map->find(e).isNull());
        TS_ASSERT(d_map->find(f) == e);
        TS_ASSERT(d_map->find(g).isNull());

        d_map->set(a, c);
        d_map->set(f, f);
        d_map->set(e, d);
        d_map->set(c, Node::null());
        d_map->set(g, a);

        TS_ASSERT(d_map->find(a) == c);
        TS_ASSERT(d_map->find(b).isNull());
        TS_ASSERT(d_map->find(c).isNull());
        TS_ASSERT(d_map->find(d).isNull());
        TS_ASSERT(d_map->find(e) == d);
        TS_ASSERT(d_map->find(f) == f);
        TS_ASSERT(d_map->find(g) == a);

      }
      d_ctxt->pop();

      TS_ASSERT(d_map->find(a) == b);
      TS_ASSERT(d_map->find(b).isNull());
      TS_ASSERT(d_map->find(c) == d);
      TS_ASSERT(d_map->find(d).isNull());
      TS_ASSERT(d_map->find(e).isNull());
      TS_ASSERT(d_map->find(f) == e);
      TS_ASSERT(d_map->find(g).isNull());
    }
    d_ctxt->pop();

    TS_ASSERT(d_map->find(a) == b);
    TS_ASSERT(d_map->find(b).isNull());
    TS_ASSERT(d_map->find(c).isNull());
    TS_ASSERT(d_map->find(d).isNull());
    TS_ASSERT(d_map->find(e).isNull());
    TS_ASSERT(d_map->find(f).isNull());
    TS_ASSERT(d_map->find(g).isNull());
  }
};