updated unit tests for business logic

Signed-off-by: hemant99911 <hemant99911@gmail.com>