'Verify that after state change, confirmation message is hidden.
Arguments:
unit_page (Object)  Unit container page.
operation (String)  Publish or discard changes operation.'

def verify_state_change(self, unit_page, operation):

    self.container.verify_publish_title(self.DRAFT_STATUS)
    if (operation == 'publish'):
        unit_page.publish_action.click()
    else:
        unit_page.discard_changes()
    self.container.verify_publish_title(self.PUBLISHED_LIVE_STATUS)
    self.container.verify_confirmation_message(message=self.message_move.format(display_name=self.source_component_display_name), verify_hidden=True)
