1package server
 2
 3import (
 4	"errors"
 5	"fmt"
 6	"log"
 7	"os"
 8	"testing"
 9	"time"
10
11	"github.com/charmbracelet/soft-serve/server/backend/sqlite"
12	"github.com/charmbracelet/soft-serve/server/config"
13	"github.com/charmbracelet/ssh"
14	bm "github.com/charmbracelet/wish/bubbletea"
15	"github.com/charmbracelet/wish/testsession"
16	"github.com/matryer/is"
17	"github.com/muesli/termenv"
18	gossh "golang.org/x/crypto/ssh"
19)
20
21func TestSession(t *testing.T) {
22	is := is.New(t)
23	t.Run("authorized repo access", func(t *testing.T) {
24		s := setup(t)
25		s.Stderr = os.Stderr
26		defer s.Close()
27		err := s.RequestPty("xterm", 80, 40, nil)
28		is.NoErr(err)
29		go func() {
30			time.Sleep(1 * time.Second)
31			s.Signal(gossh.SIGTERM)
32			// FIXME: exit with code 0 instead of forcibly closing the session
33			s.Close()
34		}()
35		err = s.Run("test")
36		var ee *gossh.ExitMissingError
37		is.True(errors.As(err, &ee))
38	})
39}
40
41func setup(tb testing.TB) *gossh.Session {
42	tb.Helper()
43	is := is.New(tb)
44	dp := tb.TempDir()
45	is.NoErr(os.Setenv("SOFT_SERVE_DATA_PATH", dp))
46	is.NoErr(os.Setenv("SOFT_SERVE_GIT_LISTEN_ADDR", ":9418"))
47	is.NoErr(os.Setenv("SOFT_SERVE_SSH_LISTEN_ADDR", fmt.Sprintf(":%d", randomPort())))
48	tb.Cleanup(func() {
49		is.NoErr(os.Unsetenv("SOFT_SERVE_DATA_PATH"))
50		is.NoErr(os.Unsetenv("SOFT_SERVE_GIT_LISTEN_ADDR"))
51		is.NoErr(os.Unsetenv("SOFT_SERVE_SSH_LISTEN_ADDR"))
52		is.NoErr(os.RemoveAll(dp))
53	})
54	cfg := config.DefaultConfig()
55	fb, err := sqlite.NewSqliteBackend(cfg)
56	if err != nil {
57		log.Fatal(err)
58	}
59	cfg = cfg.WithBackend(fb)
60	return testsession.New(tb, &ssh.Server{
61		Handler: bm.MiddlewareWithProgramHandler(SessionHandler(cfg), termenv.ANSI256)(func(s ssh.Session) {
62			_, _, active := s.Pty()
63			tb.Logf("PTY active %v", active)
64			tb.Log(s.Command())
65			s.Exit(0)
66		}),
67	}, nil)
68}