escapeSingleQuotes_test.go

func escapeSingleQuote(slice []byte) []byte {
	l := len(slice)
	var shifts int
	for index := l - 1; index >= 0; index-- {
		if slice[index] == '\'' {
			slice = append(slice, byte(0))
			copy(slice[index+1:], slice[index:]) // move bytes to right
			shifts++
		}
	}
	return slice[:l+shifts]
}

func ensureBytes(t *testing.T, got, want []byte) {
	t.Helper()
	if !bytes.Equal(got, want) {
		t.Errorf("GOT: %q; WANT: %q", got, want)
	}
}

func TestEscapeSingleQuote(t *testing.T) {
	t.Run("nil", func(t *testing.T) {
		ensureBytes(t, escapeSingleQuote(nil), nil)
	})
	t.Run("single quote", func(t *testing.T) {
		ensureBytes(t, escapeSingleQuote([]byte{'\''}), []byte{'\'', '\''})
	})
	t.Run("two single quotes", func(t *testing.T) {
		ensureBytes(t, escapeSingleQuote([]byte{'\'', '\''}), []byte{'\'', '\'', '\'', '\''})
	})
}