Improve tests

2 files changed, 29 insertions, 2 deletions

diff --git a/src/pgnp.cpp b/src/pgnp.cpp
index bb9fd29..5336cd6 100644
--- a/src/pgnp.cpp
+++ b/src/pgnp.cpp
@@ -239,7 +239,12 @@ int PGN::ParseNextTag(int start_loc) {
   tags[key] = value;
   tagkeys.push_back(key);
 
-  // TODO: Check that caracters if a ]
+  EOF_CHECK(valueloc + 1);
+  c = pgn_content[valueloc + 1];
+  if (c != ']') {
+    throw UnexpectedCharacter(c, ']', valueloc + 1);
+  }
+
   return (valueloc + 1); // +1 For the last char of the tag which is ']'
 }
 
@@ -247,7 +252,12 @@ HalfMove *PGN::GetMoves() { return (moves); }
 
 std::vector<std::string> PGN::GetTagList() { return tagkeys; }
 
-std::string PGN::GetTagValue(std::string key) { return tags[key]; }
+std::string PGN::GetTagValue(std::string key) {
+  if (tags.find(key) == tags.end()) {
+    throw InvalidTagName();
+  }
+  return tags[key];
+}
 
 void PGN::Dump() {
   std::cout << "---------- PGN DUMP ----------" << std::endl;
diff --git a/src/pgnp.hpp b/src/pgnp.hpp
index a89de73..0e9c7a5 100644
--- a/src/pgnp.hpp
+++ b/src/pgnp.hpp
@@ -2,6 +2,7 @@
 #include <exception>
 #include <fstream>
 #include <iostream>
+#include <sstream>
 #include <streambuf>
 #include <string>
 #include <unordered_map>
@@ -24,6 +25,7 @@ public:
 
   HalfMove();
   ~HalfMove();
+  /// @brief Get number of HalfMove in the MailLine
   int GetLength();
   /// @brief Dump move and all its variations
   void Dump();
@@ -66,6 +68,21 @@ struct UnexpectedEOF : public std::exception {
   const char *what() const throw() { return "Unexpected end of pgn file"; }
 };
 
+struct InvalidTagName : public std::exception {
+  const char *what() const throw() { return "Invalid tag name"; }
+};
+
+struct UnexpectedCharacter : public std::exception {
+  std::string msg;
+  UnexpectedCharacter(char actual, char required, int loc) {
+    std::stringstream ss;
+    ss << "Expected \'" << required << "\' at location " << loc
+       << " but read \'" << actual << "\'";
+    msg = ss.str();
+  }
+  const char *what() const throw() { return msg.c_str(); }
+};
+
 struct STRCheckFailed : public std::exception {
   const char *what() const throw() {
     return "Seven Tag Roster compliance check failed";