[IMP] TEST CASE:  Improved the test case to meet the requiremnts of new features.