Follow-up: Update test for status icon check

Description

This is a follow-up issue for fix: Add Code Suggestion State Check for Status Bar Icon to address !1898 (comment 2078266739).

Issue should cover the fix in the test to avoid possible regressions