Internal change

PiperOrigin-RevId: 367348796
This commit is contained in:
hai007
2021-04-07 20:12:45 -07:00
committed by Copybara-Service
parent 65a88396b2
commit d4d2d9fc74
2 changed files with 24 additions and 2 deletions
+5 -2
View File
@@ -948,10 +948,13 @@ void BwuManager::CancelRetryUpgradeAlarm(const std::string& endpoint_id) {
void BwuManager::CancelAllRetryUpgradeAlarms() {
NEARBY_LOG(INFO, "CancelAllRetryUpgradeAlarms invoked");
for (const auto& item : retry_upgrade_alarms_) {
for (auto& item : retry_upgrade_alarms_) {
const std::string& endpoint_id = item.first;
CancelRetryUpgradeAlarm(endpoint_id);
CancelableAlarm& cancellable_alarm = item.second.first;
NEARBY_LOG(INFO, "CancelRetryUpgradeAlarm for %s", endpoint_id.c_str());
cancellable_alarm.Cancel();
}
retry_upgrade_alarms_.clear();
}
Medium BwuManager::GetEndpointMedium(const std::string& endpoint_id) {
+19
View File
@@ -85,6 +85,25 @@ TEST(BwuManagerTest, CanProcessBandwidthUpgradeFrames) {
bwu_manager.Shutdown();
}
TEST(BwuManagerTest, InitiateBwu_UpgradeFails_NoCrash) {
ClientProxy client;
std::string endpoint_id("EP_A");
Mediums mediums;
EndpointChannelManager ecm;
EndpointManager em{&ecm};
BwuManager bwu_manager{mediums, em, ecm, {}, {}};
parser::UpgradePathInfo upgrade_path_info;
upgrade_path_info.set_medium(parser::UpgradePathInfo::WEB_RTC);
bwu_manager.InitiateBwuForEndpoint(&client, endpoint_id);
ExceptionOr<OfflineFrame> bwu_failed_frame =
parser::FromBytes(parser::ForBwuFailure(upgrade_path_info));
bwu_manager.OnIncomingFrame(bwu_failed_frame.result(), endpoint_id, &client,
Medium::WEB_RTC);
bwu_manager.Shutdown();
}
} // namespace
} // namespace connections
} // namespace nearby