Fixed OS tests to assert after updating its diff entities
created a fn 'get_selected_entities' to avoid duplication
updated fn name and updated exception under else part
Merge pull request #582 from sghai/fix-assertion

Added a fn to assert the updated entities under OS