[rust] Grade School

1 files changed, 75 insertions, 0 deletions

diff --git a/rust/grade-school/tests/grade-school.rs b/rust/grade-school/tests/grade-school.rs
new file mode 100644
index 0000000..3f9f060
--- /dev/null
+++ b/rust/grade-school/tests/grade-school.rs
@@ -0,0 +1,75 @@
+use grade_school as school;
+
+fn some_strings(v: &[&str]) -> Option<Vec<String>> {
+    Some(v.iter().map(|s| s.to_string()).collect())
+}
+
+#[test]
+fn test_grades_for_empty_school() {
+    let s = school::School::new();
+    assert_eq!(s.grades(), vec![]);
+}
+
+#[test]
+fn test_grades_for_one_student() {
+    let mut s = school::School::new();
+    s.add(2, "Aimee");
+    assert_eq!(s.grades(), vec![2]);
+}
+
+#[test]
+fn test_grades_for_several_students_are_sorted() {
+    let mut s = school::School::new();
+    s.add(2, "Aimee");
+    s.add(7, "Logan");
+    s.add(4, "Blair");
+    assert_eq!(s.grades(), vec![2, 4, 7]);
+}
+
+#[test]
+fn test_grades_when_several_students_have_the_same_grade() {
+    let mut s = school::School::new();
+    s.add(2, "Aimee");
+    s.add(2, "Logan");
+    s.add(2, "Blair");
+    assert_eq!(s.grades(), vec![2]);
+}
+
+#[test]
+fn test_grade_for_empty_school() {
+    let s = school::School::new();
+    assert_eq!(s.grade(1), None);
+}
+
+#[test]
+fn test_grade_when_no_students_have_that_grade() {
+    let mut s = school::School::new();
+    s.add(7, "Logan");
+    assert_eq!(s.grade(1), None);
+}
+
+#[test]
+fn test_grade_for_one_student() {
+    let mut s = school::School::new();
+    s.add(2, "Aimee");
+    assert_eq!(s.grade(2), some_strings(&["Aimee"]));
+}
+
+#[test]
+fn test_grade_returns_students_sorted_by_name() {
+    let mut s = school::School::new();
+    s.add(2, "James");
+    s.add(2, "Blair");
+    s.add(2, "Paul");
+    assert_eq!(s.grade(2), some_strings(&["Blair", "James", "Paul"]));
+}
+
+#[test]
+fn test_add_students_to_different_grades() {
+    let mut s = school::School::new();
+    s.add(3, "Chelsea");
+    s.add(7, "Logan");
+    assert_eq!(s.grades(), vec![3, 7]);
+    assert_eq!(s.grade(3), some_strings(&["Chelsea"]));
+    assert_eq!(s.grade(7), some_strings(&["Logan"]));
+}