[Test] Update test code. CUnit test framework was updated, have to update test code to adapt.