added support for overring the team number in tests
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