ActionUnitTest.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "gtest/gtest.h"

#include "MooseMesh.h"
#include "FEProblem.h"
#include "AppFactory.h"
#include "MooseMain.h"

class ActionUnitTest : public ::testing::Test
{
public:
  ActionUnitTest(const std::string & app_name)
    : _app(Moose::createMooseApp(app_name, 0, nullptr)),
      _factory(_app->getFactory()),
      _action_factory(_app->getActionFactory())
  {
    buildMinimalObjects();
  }

  void SetUp() override
  {
    buildActions();
    runActions();
  }

protected:
  virtual void buildActions() = 0;

  void buildMinimalObjects()
  {
    InputParameters mesh_params = _factory.getValidParams("GeneratedMesh");
    mesh_params.set<MooseEnum>("dim") = "3";
    mesh_params.set<unsigned int>("nx") = 2;
    mesh_params.set<unsigned int>("ny") = 2;
    mesh_params.set<unsigned int>("nz") = 2;
    _mesh = _factory.createUnique<MooseMesh>("GeneratedMesh", "name1", mesh_params);
    _mesh->setMeshBase(_mesh->buildMeshBaseObject());
    _mesh->buildMesh();

    InputParameters problem_params = _factory.getValidParams("FEProblem");
    problem_params.set<MooseMesh *>("mesh") = _mesh.get();
    problem_params.set<std::string>(MooseBase::name_param) = "name2";
    _fe_problem = _factory.create<FEProblem>("FEProblem", "problem", problem_params);

    _fe_problem->createQRules(libMesh::QGAUSS, libMesh::FIRST, libMesh::FIRST, libMesh::FIRST);

    _app->actionWarehouse().problemBase() = _fe_problem;
  }

  virtual void runActions() = 0;

  template <typename T>
  T & addObject(const std::string & type, const std::string & name, InputParameters & params);

  std::unique_ptr<MooseMesh> _mesh;
  std::shared_ptr<MooseApp> _app;
  Factory & _factory;
  ActionFactory & _action_factory;
  std::shared_ptr<FEProblem> _fe_problem;
};

template <typename T>
T &
ActionUnitTest::addObject(const std::string & type,
                          const std::string & name,
                          InputParameters & params)
{
  auto objects = _fe_problem->addObject<T>(type, name, params);
  mooseAssert(objects.size() == 1, "Doesn't work with threading");
  return *objects[0];
}