elm/exercises/grade-school/GradeSchoolTests.elm

module Main exposing (..)

import ElmTest exposing (..)
import GradeSchool exposing (addStudent, studentsInGrade, allStudents)


tests : Test
tests =
    suite "GradeSchool"
        [ test "add student"
            (assertEqual [ "Aimee" ]
                (GradeSchool.empty
                    |> addStudent 2 "Aimee"
                    |> studentsInGrade 2
                )
            )
        , test "add more students in same class"
            (assertEqual [ "Blair", "James", "Paul" ]
                (GradeSchool.empty
                    |> addStudent 2 "James"
                    |> addStudent 2 "Blair"
                    |> addStudent 2 "Paul"
                    |> studentsInGrade 2
                )
            )
        , test "add students to different grades"
            (assertEqual [ [ "Chelsea" ], [ "Logan" ] ]
                (let
                    school =
                        GradeSchool.empty
                            |> addStudent 3 "Chelsea"
                            |> addStudent 7 "Logan"
                 in
                    [ studentsInGrade 3 school, studentsInGrade 7 school ]
                )
            )
        , test "get students in a grade"
            (assertEqual [ "Bradley", "Franklin" ]
                (GradeSchool.empty
                    |> addStudent 5 "Franklin"
                    |> addStudent 5 "Bradley"
                    |> addStudent 1 "Jeff"
                    |> studentsInGrade 5
                )
            )
        , test "get all students in the school"
            (assertEqual [ ( 3, [ "Kyle" ] ), ( 4, [ "Christopher", "Jennifer" ] ), ( 6, [ "Kareem" ] ) ]
                (GradeSchool.empty
                    |> addStudent 4 "Jennifer"
                    |> addStudent 6 "Kareem"
                    |> addStudent 4 "Christopher"
                    |> addStudent 3 "Kyle"
                    |> allStudents
                )
            )
        , test "get students in a non-existent grade"
            (assertEqual [] (studentsInGrade 1 GradeSchool.empty))
        ]


main : Program Never
main =
    runSuite tests