asrock: Add tests

bmc-toolbox · Oct 30, 2023 · e63714c · e63714c
1 parent 029809e
commit e63714c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/providers/asrockrack/mock_test.go b/providers/asrockrack/mock_test.go
@@ -82,6 +82,7 @@ func mockASRockBMC() *httptest.Server {
 	// fw update endpoints - in order of invocation
 	handler.HandleFunc("/api/maintenance/flash", bmcFirmwareUpgrade)
 	handler.HandleFunc("/api/maintenance/firmware", bmcFirmwareUpgrade)
+	handler.HandleFunc("/api/maintenance/firmware/firmware", bmcFirmwareUpgrade)
 	handler.HandleFunc("/api/maintenance/firmware/verification", bmcFirmwareUpgrade)
 	handler.HandleFunc("/api/maintenance/firmware/upgrade", bmcFirmwareUpgrade)
 	handler.HandleFunc("/api/maintenance/firmware/flash-progress", bmcFirmwareUpgrade)
@@ -211,7 +212,7 @@ func bmcFirmwareUpgrade(w http.ResponseWriter, r *http.Request) {
 			w.WriteHeader(http.StatusOK)
 
 		// 2. upload firmware
-		case "/api/maintenance/firmware":
+		case "/api/maintenance/firmware", "/api/maintenance/firmware/firmware":
 
 			// validate flash mode set
 			if !fwUpgradeState.FlashModeSet {