Function TestConnectionSetup

agentprotocol/server_test.go:14–100

region Tests

(t *testing.T)

// region Tests
func TestConnectionSetup(t *testing.T) {
	log := log.NewTestLogger(t)
	fromClientReader, fromClientWriter := io.Pipe()
	toClientReader, toClientWriter := io.Pipe()

	clientCtx := proto.NewForwardCtx(toClientReader, fromClientWriter, log)

	serverCtx := proto.NewForwardCtx(fromClientReader, toClientWriter, log)

	closeChan := make(chan struct{})

	go func() {
		connChan, err := serverCtx.StartReverseForwardClient(
			"127.0.0.1",
			8080,
			false,
		)
		if err != nil {
			panic(err)
		}

		testConServer := <-connChan
		err = testConServer.Accept()
		if err != nil {
			log.Error("Error accept connection", err)
		}
		buf := make([]byte, 512)
		nBytes, err := testConServer.Read(buf)
		if err != nil {
			log.Error("Failed to read from server")
		}
		_, err = testConServer.Write(buf[:nBytes])
		if err != nil {
			log.Error("Failed to write to server")
		}
		<-closeChan
		serverCtx.Kill()
	}()

	conType, setup, connectionChan, err := clientCtx.StartClient()
	if err != nil {
		t.Fatal("Test failed with error", err)
	}
	if conType != proto.CONNECTION_TYPE_PORT_FORWARD {
		panic(fmt.Errorf("Invalid connection type %d", conType))
	}

	go func() {
		for {
			conn, ok := <-connectionChan
			if !ok {
				break
			}
			_ = conn.Reject()
			_ = conn.Close()
		}
	}()