// 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
}