return absl::FailedPreconditionError("Unexpected listener history.");
      }
    }
    return absl::OkStatus();
  }

  std::vector<XlaJitCompilationActivity> GetListenerHistory() {
    return activity_history_;
  }

  void ClearListenerHistory() { activity_history_.clear(); }

 private:
  std::vector<XlaJitCompilationActivity> activity_history_;
};

  EXPECT_EQ(stats.compile_count, 5);
  EXPECT_EQ(stats.cumulative_compile_time_us, 5 * 4);

  // TODO(b/255826209): Use ::testing::EqualsProto once b/135192747 is fixed.
  const auto& actual_activities = listener_ptr->GetListenerHistory();
  EXPECT_EQ(actual_activities.size(), expected_activities.size());
  for (size_t i = 0; i < actual_activities.size(); ++i) {
    EXPECT_EQ(actual_activities[i].SerializeAsString(),