Improve PGNP API

1 files changed, 10 insertions, 4 deletions

diff --git a/tests/tests.cpp b/tests/tests.cpp
index d766abd..551e9d4 100644
--- a/tests/tests.cpp
+++ b/tests/tests.cpp
@@ -8,7 +8,9 @@ TEST_CASE("Valid PGN", "[valid/pgn1]") {
   REQUIRE_NOTHROW(pgn.FromFile("pgn_files/valid/pgn1.pgn"));
   REQUIRE_THROWS(pgn.STRCheck());
 
-  HalfMove *m = pgn.GetMoves();
+  HalfMove *m = new HalfMove();
+  pgn.GetMoves(m);
+  HalfMove *m_backup = m;
   REQUIRE(m->GetLength() == 6);
 
   SECTION("Main line move checks") {
@@ -31,7 +33,7 @@ TEST_CASE("Valid PGN", "[valid/pgn1]") {
   }
 
   SECTION("Main line color checks") {
-    m = pgn.GetMoves();
+    m=m_backup;
     CHECK_FALSE(m->isBlack);
 
     m = m->MainLine;
@@ -63,7 +65,9 @@ TEST_CASE("Valid PGN", "[valid/pgn2]") {
   PGN pgn;
   REQUIRE_NOTHROW(pgn.FromFile("pgn_files/valid/pgn2.pgn"));
   REQUIRE_THROWS(pgn.STRCheck());
-  REQUIRE(pgn.GetMoves()->GetLength() == 66);
+  HalfMove *m = new HalfMove();
+  pgn.GetMoves(m);
+  REQUIRE(m->GetLength() == 66);
   CHECK(pgn.GetResult() == "0-1");
 }
 
@@ -71,6 +75,8 @@ TEST_CASE("Seven Tag Roster", "[std/pgn1]") {
   PGN pgn;
   REQUIRE_NOTHROW(pgn.FromFile("pgn_files/str/pgn1.pgn"));
   REQUIRE_NOTHROW(pgn.STRCheck());
-  REQUIRE(pgn.GetMoves()->GetLength() == 85);
+  HalfMove *m = new HalfMove();
+  pgn.GetMoves(m);
+  REQUIRE(m->GetLength() == 85);
   CHECK(pgn.GetResult() == "1/2-1/2");
 }