From 80d9b36901ff0b30c0faa528cadab0f1d67b71cf Mon Sep 17 00:00:00 2001
From: Kai-Philipp Nosper <kai-philipp.nosper@stud.h-da.de>
Date: Fri, 11 Feb 2022 19:08:08 +0100
Subject: [PATCH] Add test for stringstore

---
 src/stringstore.rs | 53 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/src/stringstore.rs b/src/stringstore.rs
index 3869ab7..5db7066 100644
--- a/src/stringstore.rs
+++ b/src/stringstore.rs
@@ -49,3 +49,56 @@ impl StringStore {
     }
 
 }
+
+#[cfg(test)]
+mod tests {
+    use super::StringStore;
+
+    #[test]
+    fn test_stringstore_intern_lookup() {
+        let mut ss = StringStore::new();
+        let s1 = "Hello";
+        let s2 = "World";
+
+        let id1 = ss.intern_or_lookup(s1);
+        assert_eq!(ss.lookup(id1).unwrap().as_str(), s1);
+
+        let id2 = ss.intern_or_lookup(s2);
+        assert_eq!(ss.lookup(id2).unwrap().as_str(), s2);
+        assert_eq!(ss.lookup(id1).unwrap().as_str(), s1);
+    }
+
+    #[test]
+    fn test_stringstore_no_duplicates() {
+        let mut ss = StringStore::new();
+        let s1 = "Hello";
+        let s2 = "World";
+
+        let id1_1 = ss.intern_or_lookup(s1);
+        assert_eq!(ss.lookup(id1_1).unwrap().as_str(), s1);
+
+        let id1_2 = ss.intern_or_lookup(s1);
+        assert_eq!(ss.lookup(id1_2).unwrap().as_str(), s1);
+
+        // Check that the string is the same
+        assert_eq!(id1_1, id1_2);
+
+        // Check that only one string is actually stored
+        assert_eq!(ss.strings.len(), 1);
+        assert_eq!(ss.sids.len(), 1);
+
+
+
+        let id2_1 = ss.intern_or_lookup(s2);
+        assert_eq!(ss.lookup(id2_1).unwrap().as_str(), s2);
+
+        let id2_2 = ss.intern_or_lookup(s2);
+        assert_eq!(ss.lookup(id2_2).unwrap().as_str(), s2);
+
+        // Check that the string is the same
+        assert_eq!(id2_1, id2_2);
+
+        assert_eq!(ss.strings.len(), 2);
+        assert_eq!(ss.sids.len(), 2);
+    }
+}