queue unit test

befehl.go

@@ -168,7 +168,7 @@ func (instance *Instance) executePayloadOnHosts(payload []byte, hostsFilePath st
 		go func() {
 			instance.runPayload(&wg, host, payload, sshConfig)
 			<-hostsChan
-			remaining := queue.decrementCounter(hostCnt)
+			remaining := queue.decrementCounter()
 			color.Magenta(fmt.Sprintf("Remaining: %d / %d\n", remaining, hostCnt))
 		}()
 	}

queue.go

@@ -14,6 +14,6 @@ func (q *queue) New(hostCnt int64) *queue {
 	}
 }
 
-func (q *queue) decrementCounter(total int) int64 {
+func (q *queue) decrementCounter() int64 {
 	return atomic.AddInt64(&q.count, -1)
 }

queue_test.go

@@ -0,0 +1,23 @@
+package befehl
+
+import (
+	"testing"
+)
+
+func TestNew(t *testing.T) {
+	q := new(queue).New(42)
+	if q.count != 42 {
+		t.Fatalf("count value of %d is unexpected", q.count)
+	}
+}
+
+func TestDecrementCounter(t *testing.T) {
+	q := new(queue).New(42)
+	newCount := q.decrementCounter()
+	if q.count != 41 {
+		t.Fatalf("count value of %d is unexpected", q.count)
+	}
+	if newCount != 41 {
+		t.Fatalf("newCount of %d is unexpected", newCount)
+	}
+}