initial commit [ci skip]

engine/replacer/replacer_test.go

@@ -0,0 +1,56 @@
+// Code generated automatically. DO NOT EDIT.
+
+// Copyright 2019 Drone.IO Inc. All rights reserved.
+// Use of this source code is governed by the Polyform License
+// that can be found in the LICENSE file.
+
+package replacer
+
+import (
+	"bytes"
+	"io"
+	"testing"
+
+	"github.com/drone-runners/drone-runner-docker/engine"
+)
+
+func TestReplace(t *testing.T) {
+	secrets := []*engine.Secret{
+		{Name: "DOCKER_USERNAME", Data: []byte("octocat"), Mask: false},
+		{Name: "DOCKER_PASSWORD", Data: []byte("correct-horse-batter-staple"), Mask: true},
+		{Name: "DOCKER_EMAIL", Data: []byte(""), Mask: true},
+	}
+
+	buf := new(bytes.Buffer)
+	w := New(&nopCloser{buf}, secrets)
+	w.Write([]byte("username octocat password correct-horse-batter-staple"))
+	w.Close()
+
+	if got, want := buf.String(), "username octocat password [secret:docker_password]"; got != want {
+		t.Errorf("Want masked string %s, got %s", want, got)
+	}
+}
+
+// this test verifies that if there are no secrets to scan and
+// mask, the io.WriteCloser is returned as-is.
+func TestReplaceNone(t *testing.T) {
+	secrets := []*engine.Secret{
+		{Name: "DOCKER_USERNAME", Data: []byte("octocat"), Mask: false},
+		{Name: "DOCKER_PASSWORD", Data: []byte("correct-horse-batter-staple"), Mask: false},
+	}
+
+	buf := new(bytes.Buffer)
+	w := &nopCloser{buf}
+	r := New(w, secrets)
+	if w != r {
+		t.Errorf("Expect buffer returned with no replacer")
+	}
+}
+
+type nopCloser struct {
+	io.Writer
+}
+
+func (*nopCloser) Close() error {
+	return nil
+}