add unit test for logic class