add unit tests to head_refs #700

o2sh · Aug 15, 2022 · 0b7111a · 0b7111a
1 parent 8fe3e42
commit 0b7111a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/info/head_refs.rs b/src/info/head_refs.rs
@@ -42,3 +42,20 @@ impl Serialize for HeadRefs {
         state.end()
     }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_head_refs() {
+        let head = HeadRefs::new("be561d5".into(), vec!["main".into(), "origin/main".into()]);
+        assert_eq!(format!("{}", head), "be561d5 (main, origin/main)")
+    }
+
+    #[test]
+    fn test_head_refs_with_no_refs() {
+        let head = HeadRefs::new("be561d5".into(), vec![]);
+        assert_eq!(format!("{}", head), "be561d5")
+    }
+}