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_TRUE(xla_executable != nullptr);

  // Check if device_compiler was able to load the executable from the
  // persistent cache.
  std::vector<XlaJitCompilationActivity> activity_history =
      listener_->GetListenerHistory();
  EXPECT_EQ(activity_history.size(), 1);
  EXPECT_EQ(activity_history[0].cluster_name(), fn.name());
  EXPECT_EQ(activity_history[0].compile_count(), 1);
  EXPECT_FALSE(activity_history[0].used_persistent_cache());

  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(),