added support for overring the team number in tests
diff --git a/aos/common/network/team_number.cc b/aos/common/network/team_number.cc
index 3e9aeb6..38b169a 100644
--- a/aos/common/network/team_number.cc
+++ b/aos/common/network/team_number.cc
@@ -11,7 +11,10 @@
namespace network {
namespace {
+uint16_t override_team = 0;
+
uint16_t *DoGetTeamNumber() {
+ if (override_team != 0) return &override_team;
const in_addr &address = configuration::GetOwnIPAddress();
static uint16_t r =
(((address.s_addr & 0xFF00) >> 8) * 100) +
@@ -27,5 +30,7 @@
return *once.Get();
}
+void OverrideTeamNumber(uint16_t team) { override_team = team; }
+
} // namespace network
} // namespace aos
diff --git a/aos/common/network/team_number.h b/aos/common/network/team_number.h
index f250c85..08103ed 100644
--- a/aos/common/network/team_number.h
+++ b/aos/common/network/team_number.h
@@ -11,6 +11,12 @@
// repeatedly.
uint16_t GetTeamNumber();
+// Overrides the team number returned from GetTeamNumber(). Must be called
+// before GetTeamNumber() is ever called.
+// Overriding to team 0 won't work.
+// Intended only for tests.
+void OverrideTeamNumber(uint16_t team);
+
} // namespace network
} // namespace aos