fix: set default timeout to 300 seconds

runpod · Jan 7, 2025 · ce68609 · ce68609
1 parent e97554d
commit ce68609
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -104,6 +104,9 @@ func init() {
 		log.Info("Parsed test config", zap.Any("testConfig", testConfig))
 		for i, test := range testConfig {
 			test.ID = i
+			if test.Timeout == 0 {
+				test.Timeout = 300
+			}
 		}
 	}
 }
@@ -197,7 +200,6 @@ func (h *Handler) JobTake(c *gin.Context) {
 
 	if currentTest >= len(testConfig) {
 		sendResultsToGraphQL("COMPLETED", nil)
-		time.Sleep(time.Duration(10) * time.Second)
 		h.log.Error("No more tests", zap.Int("current_test", currentTest))
 
 		c.JSON(500, gin.H{