mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 19:45:28 +00:00
17 lines
275 B
Go
17 lines
275 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
|
||
|
|
"github.com/projectdiscovery/gologger/levels"
|
||
|
|
)
|
||
|
|
|
||
|
|
// CaptureWriter captures log output for testing
|
||
|
|
type CaptureWriter struct {
|
||
|
|
Buffer *bytes.Buffer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *CaptureWriter) Write(data []byte, level levels.Level) {
|
||
|
|
w.Buffer.Write(data)
|
||
|
|
}
|