Fix bug when removing test case keys

[anna.git] / include / anna / testing / TestCase.hpp

diff --git a/include/anna/testing/TestCase.hpp b/include/anna/testing/TestCase.hpp
index d17a9f8..bb6f05d 100644 (file)
--- a/include/anna/testing/TestCase.hpp
+++ b/include/anna/testing/TestCase.hpp
@@ -42,7 +42,7 @@ namespace testing {
 class TestCase {
 
   void assertInitialized() const noexcept(false);
-  std::string assertMessage(const anna::DataBlock &db, bool toEntity) noexcept(false);
+  void assertMessage(const anna::DataBlock &db, bool toEntity, std::string &key1, std::string &key2) noexcept(false);
 
 public:
 
@@ -124,7 +124,8 @@ public:
 
   // getters
   const unsigned int &getId() const { return a_id; }
-  const std::string &getKey() const { return a_key; }
+  const std::string &getKey1() const { return a_key1; }
+  const std::string &getKey2() const { return a_key2; }
   const std::string &getDescription() const { return a_description; }
 
   // setters
@@ -146,7 +147,8 @@ public:
 private:
   // private members:
   unsigned int a_id;
-  std::string a_key;
+  std::string a_key1;
+  std::string a_key2;
   std::string a_description;
   std::vector<TestStep*> a_steps;
   std::vector<TestStep*>::const_iterator a_stepsIt;