commit 1f2996ede0f352b25ab8a903f0988ee9b8ecd697
author Austin Schuh <austin.linux@gmail.com> Sun Mar 15 23:09:00 2020 -0700
committer Austin Schuh <austin.linux@gmail.com> Sun Mar 22 00:16:41 2020 -0700
tree 96e88013290c37524a52c2e2e4e1f6d0f664520d
parent 2fe4b71c1a058b0058f4d9f06f61dc818b9c8780

Add function to parse pi number from hostname

Change-Id: Ie3f01d98e368816a3e58abe3d7639d4243a44012

aos/network/team_number_test.cc

diff --git a/aos/network/team_number_test.cc b/aos/network/team_number_test.cc
index c11f0a0..68727e2 100644
--- a/aos/network/team_number_test.cc
+++ b/aos/network/team_number_test.cc
@@ -6,8 +6,8 @@
 namespace network {
 namespace testing {
 
-using team_number_internal::ParseRoborioTeamNumber;
 using team_number_internal::ParsePiTeamNumber;
+using team_number_internal::ParseRoborioTeamNumber;
 
 TEST(TeamNumberTest, Parse2015TeamNumber) {
   EXPECT_EQ(971u, *ParseRoborioTeamNumber("roboRIO-971"));
@@ -39,6 +39,14 @@
   EXPECT_FALSE(ParseRoborioTeamNumber("pi-"));
   EXPECT_FALSE(ParseRoborioTeamNumber("pi-971"));
   EXPECT_FALSE(ParseRoborioTeamNumber("pi-971a-1"));
+
+  EXPECT_EQ(1u, *ParsePiNumber("pi-971-1"));
+  EXPECT_EQ(22u, *ParsePiNumber("pi-8971-22"));
+
+  EXPECT_FALSE(ParsePiNumber("pi-8971-"));
+  EXPECT_FALSE(ParsePiNumber("pi"));
+  EXPECT_FALSE(ParsePiNumber("pi-"));
+  EXPECT_FALSE(ParsePiNumber("pi-971"));
 }
 
 }  // namespace testing