Happy path tests work. Need to push edge case tests.