def METHOD_NAME(upgrade_buckets, mcg_obj_session):
    """
    Test that all buckets in cluster are in OPTIMAL state after upgrade.
    """
    for bucket in mcg_obj_session.read_system().get("buckets"):
        assert bucket.get("mode") == BS_OPTIMAL