|  | @@ -0,0 +1,40 @@
 | 
	
		
			
				|  |  | +package log_test
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +import (
 | 
	
		
			
				|  |  | +	"io/ioutil"
 | 
	
		
			
				|  |  | +	"os"
 | 
	
		
			
				|  |  | +	"testing"
 | 
	
		
			
				|  |  | +	"time"
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common"
 | 
	
		
			
				|  |  | +	"v2ray.com/core/common/buf"
 | 
	
		
			
				|  |  | +	. "v2ray.com/core/common/log"
 | 
	
		
			
				|  |  | +	. "v2ray.com/ext/assert"
 | 
	
		
			
				|  |  | +)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func TestFileLogger(t *testing.T) {
 | 
	
		
			
				|  |  | +	assert := With(t)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	f, err := ioutil.TempFile("", "vtest")
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +	path := f.Name()
 | 
	
		
			
				|  |  | +	common.Must(f.Close())
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	creator, err := CreateFileLogWriter(path)
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	handler := NewLogger(creator)
 | 
	
		
			
				|  |  | +	handler.Handle(&GeneralMessage{Content: "Test Log"})
 | 
	
		
			
				|  |  | +	time.Sleep(2 * time.Second)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	common.Must(common.Close(handler))
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	f, err = os.Open(path)
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	b, err := buf.ReadAllToBytes(f)
 | 
	
		
			
				|  |  | +	assert(err, IsNil)
 | 
	
		
			
				|  |  | +	assert(string(b), HasSubstring, "Test Log")
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	common.Must(f.Close())
 | 
	
		
			
				|  |  | +}
 |