add unit test for logic class

add this to your project. One test is fail. take care about it.