commit 68b84df9949983d8a0e55183749f1f6d2a0bb670
author Daniel Zheng <zhengdaniel@google.com> Thu Apr 13 11:39:26 2023 -0700
committer Daniel Zheng <zhengdaniel@google.com> Wed Apr 19 12:30:38 2023 -0700
tree d19c85116cf48e8820448bbec2c787fa712e4b3f
parent 76aa257850e511a49bde24588d3500d4cccf4d07

Added test for version check

Test: fastboot_test
Bug: 194686221
Change-Id: I886c0a91be3b4f4bf7f2b8c76704aa30a352ee5b

fastboot/task_test.cpp

diff --git a/fastboot/task_test.cpp b/fastboot/task_test.cpp
index 400e27f..52d1fc3 100644
--- a/fastboot/task_test.cpp
+++ b/fastboot/task_test.cpp
@@ -80,3 +80,21 @@
         ASSERT_EQ(task->GetImageName(), expected_values[i][3]);
     }
 }
+
+TEST_F(ParseTest, VERSION_CHECK_CORRRECT) {
+    std::vector<std::string> correct_versions = {
+            "version 1.0",
+            "version 22.00",
+    };
+
+    std::vector<std::string> bad_versions = {"version",        "version .01", "version x1",
+                                             "version 1.0.1",  "version 1.",  "s 1.0",
+                                             "version 1.0 2.0"};
+
+    for (auto& version : correct_versions) {
+        ASSERT_TRUE(CheckFastbootInfoRequirements(android::base::Split(version, " "))) << version;
+    }
+    for (auto& version : bad_versions) {
+        ASSERT_FALSE(CheckFastbootInfoRequirements(android::base::Split(version, " "))) << version;
+    }
+}
\ No newline at end of file