test(main): add unit tests