mirror of
				https://github.com/juanfont/headscale.git
				synced 2025-10-28 10:51:44 +01:00 
			
		
		
		
	* handle control protocol through websocket The necessary behaviour is already in place, but the wasm build only issued GETs, and the handler was not invoked. * get DERP-over-websocket working for wasm clients * Prepare for testing builtin websocket-over-DERP Still needs some way to assert that clients are connected through websockets, rather than the TCP hijacking version of DERP. * integration tests: properly differentiate between DERP transports * do not touch unrelated code * linter fixes * integration testing: unexport common implementation of derp server scenario * fixup! integration testing: unexport common implementation of derp server scenario * dockertestutil/logs: remove unhelpful comment * update changelog --------- Co-authored-by: Csaba Sarkadi <sarkadicsa@tutanota.de>
		
			
				
	
	
		
			79 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			79 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package dockertestutil
 | 
						|
 | 
						|
import (
 | 
						|
	"bytes"
 | 
						|
	"context"
 | 
						|
	"io"
 | 
						|
	"log"
 | 
						|
	"os"
 | 
						|
	"path"
 | 
						|
 | 
						|
	"github.com/ory/dockertest/v3"
 | 
						|
	"github.com/ory/dockertest/v3/docker"
 | 
						|
)
 | 
						|
 | 
						|
const filePerm = 0o644
 | 
						|
 | 
						|
func WriteLog(
 | 
						|
	pool *dockertest.Pool,
 | 
						|
	resource *dockertest.Resource,
 | 
						|
	stdout io.Writer,
 | 
						|
	stderr io.Writer,
 | 
						|
) error {
 | 
						|
	return pool.Client.Logs(
 | 
						|
		docker.LogsOptions{
 | 
						|
			Context:      context.TODO(),
 | 
						|
			Container:    resource.Container.ID,
 | 
						|
			OutputStream: stdout,
 | 
						|
			ErrorStream:  stderr,
 | 
						|
			Tail:         "all",
 | 
						|
			RawTerminal:  false,
 | 
						|
			Stdout:       true,
 | 
						|
			Stderr:       true,
 | 
						|
			Follow:       false,
 | 
						|
			Timestamps:   false,
 | 
						|
		},
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func SaveLog(
 | 
						|
	pool *dockertest.Pool,
 | 
						|
	resource *dockertest.Resource,
 | 
						|
	basePath string,
 | 
						|
) (string, string, error) {
 | 
						|
	err := os.MkdirAll(basePath, os.ModePerm)
 | 
						|
	if err != nil {
 | 
						|
		return "", "", err
 | 
						|
	}
 | 
						|
 | 
						|
	var stdout, stderr bytes.Buffer
 | 
						|
	err = WriteLog(pool, resource, &stdout, &stderr)
 | 
						|
	if err != nil {
 | 
						|
		return "", "", err
 | 
						|
	}
 | 
						|
 | 
						|
	log.Printf("Saving logs for %s to %s\n", resource.Container.Name, basePath)
 | 
						|
 | 
						|
	stdoutPath := path.Join(basePath, resource.Container.Name+".stdout.log")
 | 
						|
	err = os.WriteFile(
 | 
						|
		stdoutPath,
 | 
						|
		stdout.Bytes(),
 | 
						|
		filePerm,
 | 
						|
	)
 | 
						|
	if err != nil {
 | 
						|
		return "", "", err
 | 
						|
	}
 | 
						|
 | 
						|
	stderrPath := path.Join(basePath, resource.Container.Name+".stderr.log")
 | 
						|
	err = os.WriteFile(
 | 
						|
		stderrPath,
 | 
						|
		stderr.Bytes(),
 | 
						|
		filePerm,
 | 
						|
	)
 | 
						|
	if err != nil {
 | 
						|
		return "", "", err
 | 
						|
	}
 | 
						|
 | 
						|
	return stdoutPath, stderrPath, nil
 | 
						|
}
 |