go on TDD

add unit tests