💾 Archived View for source.community › ckaznocha › gemini › blob › main › geminitest › server.go captured on 2024-05-26 at 15:02:08. Gemini links have been rewritten to link to archived content
⬅️ Previous capture (2024-02-05)
-=-=-=-=-=-=-
.
,-. ,-. . . ,-. ,-. ,-. ,-. ,-. ,-,-. ,-,-. . . ,-. . |- . .
`-. | | | | | | |-' | | | | | | | | | | | | | | | | |
`-' `-' `-^ ' `-' `-' :: `-' `-' ' ' ' ' ' ' `-^ ' ' ' `' `-|
/|
`-'
git clone https://source.community/ckaznocha/gemini.git
View raw contents of /geminitest/server.go (main)
────╮
1│ package geminitest
2│
3│ import (
4│ "crypto/tls"
5│ "crypto/x509"
6│ "fmt"
7│ "net"
8│ "sync"
9│ "testing"
10│
11│ "source.community/ckaznocha/gemini"
12│ )
13│
14│ // Server is gemini server listing on a randomly chosen open port.
15│ type Server struct {
16│ Listener net.Listener
17│ TLS *tls.Config
18│ Config *gemini.Server
19│ certificate *x509.Certificate
20│ URL string
21│ Addr string
22│ wg sync.WaitGroup
23│ mu sync.Mutex
24│ closed bool
25│ }
26│
27│ func newLocalListener() net.Listener {
28│ l, err := net.Listen("tcp", "127.0.0.1:0")
29│ if err != nil {
30│ if l, err = net.Listen("tcp6", "[::1]:0"); err != nil {
31│ panic(fmt.Sprintf("httptest: failed to listen on a port: %v", err))
32│ }
33│ }
34│
35│ return l
36│ }
37│
38│ // NewUnstartedServer returns a server which has not been started.
39│ func NewUnstartedServer(t *testing.T, handler gemini.Handler) *Server {
40│ t.Helper()
41│
42│ return &Server{
43│ Listener: newLocalListener(),
44│ Config: &gemini.Server{
45│ Handler: handler,
46│ LogHandler: func(message string, isError bool) { t.Log(message) },
47│ },
48│ }
49│ }
50│
51│ // Start starts an unstarted server.
52│ func (s *Server) Start() {
53│ if s.URL != "" {
54│ panic("Server already started")
55│ }
56│
57│ cert, err := tls.LoadX509KeyPair("testdata/cert.pem", "testdata/key.pem")
58│ if err != nil {
59│ panic(fmt.Sprintf("httptest: NewTLSServer: %v", err))
60│ }
61│
62│ if existingConfig := s.TLS; existingConfig != nil {
63│ s.TLS = existingConfig.Clone()
64│ } else {
65│ s.TLS = new(tls.Config)
66│ }
67│
68│ if len(s.TLS.Certificates) == 0 {
69│ s.TLS.Certificates = []tls.Certificate{cert}
70│ }
71│
72│ s.certificate, err = x509.ParseCertificate(s.TLS.Certificates[0].Certificate[0])
73│ if err != nil {
74│ panic(fmt.Sprintf("httptest: NewTLSServer: %v", err))
75│ }
76│
77│ certpool := x509.NewCertPool()
78│ certpool.AddCert(s.certificate)
79│
80│ s.Listener = tls.NewListener(s.Listener, s.TLS)
81│ s.URL = "gemini://" + s.Listener.Addr().String()
82│ s.Addr = s.Listener.Addr().String()
83│ s.goServe()
84│ }
85│
86│ func (s *Server) goServe() {
87│ s.wg.Add(1)
88│
89│ go func() {
90│ defer s.wg.Done()
91│
92│ if err := s.Config.Serve(s.Listener); err != nil && s.Config.LogHandler != nil {
93│ s.Config.LogHandler(err.Error(), true)
94│ }
95│ }()
96│ }
97│
98│ // NewServer returns a test server which has been started and is ready to use.
99│ func NewServer(t *testing.T, handler gemini.Handler) *Server {
100│ t.Helper()
101│
102│ ts := NewUnstartedServer(t, handler)
103│ ts.Start()
104│
105│ return ts
106│ }
107│
108│ // Close closes the server.
109│ func (s *Server) Close() {
110│ s.mu.Lock()
111│ if !s.closed {
112│ s.closed = true
113│
114│ if err := s.Listener.Close(); err != nil && s.Config.LogHandler != nil {
115│ s.Config.LogHandler(err.Error(), true)
116│ }
117│ }
118│
119│ s.mu.Unlock()
120│
121│ s.wg.Wait()
122│ }
────╯
· · ·
© 2024 source.community