Remove prints, update test

ansible · Oct 31, 2024 · e475c48 · e475c48
1 parent 4f6d051
commit e475c48
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 8 deletions.
diff --git a/pkg/types/main.go b/pkg/types/main.go
@@ -38,10 +38,6 @@ func (cfg NodeCfg) Init() error {
 	} else {
 		submitIDRegex := regexp.MustCompile(`^[.\-_a-zA-Z0-9]*$`)
 		match := submitIDRegex.FindSubmatch([]byte(cfg.ID))
-		for _, m := range match {
-			fmt.Println("HERE: " + string(m))
-			fmt.Println(cfg.ID)
-		}
 		if match == nil {
 			return fmt.Errorf("node id can only contain a-z, A-Z, 0-9 or special characters . - _ but received: %s", cfg.ID)
 		}

diff --git a/pkg/workceptor/remote_work.go b/pkg/workceptor/remote_work.go
@@ -183,9 +183,6 @@ func (rw *remoteUnit) startRemoteUnit(ctx context.Context, conn net.Conn, reader
 	}
 	submitIDRegex := regexp.MustCompile(`([.\-_a-zA-Z0-9]+)~([a-zA-Z0-9]+){8}`)
 	match := submitIDRegex.FindSubmatch([]byte(response))
-	for _, m := range match {
-		fmt.Println("HERE: " + string(m))
-	}
 	if match == nil || len(match) != 3 {
 		return fmt.Errorf("could not parse response: %s", strings.TrimRight(response, "\n"))
 	}

diff --git a/pkg/workceptor/workceptor_test.go b/pkg/workceptor/workceptor_test.go
@@ -48,7 +48,7 @@ func TestAllocateUnit(t *testing.T) {
 		return mockWorkUnit
 	}
 
-	mockNetceptor.EXPECT().NodeID().Return("test")
+	mockNetceptor.EXPECT().NodeID().Return("test").Times(4)
 	w, err := workceptor.New(ctx, mockNetceptor, "/tmp")
 	if err != nil {
 		t.Errorf("Error while creating Workceptor: %v", err)