Justin Harms
a3defe5cf6
- Introduced unit tests for the main package to ensure compilation. - Added tests for the manager, including validation of upload sessions and handling of Blender binary paths. - Implemented tests for job token generation and validation, ensuring security and integrity. - Created tests for configuration management and database schema to verify functionality. - Added tests for logger and runner components to enhance overall test coverage and reliability.
45 lines
1008 B
Go
45 lines
1008 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func TestJobConnection_ConnectAndClose(t *testing.T) {
|
|
upgrader := websocket.Upgrader{}
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
var msg map[string]interface{}
|
|
if err := conn.ReadJSON(&msg); err != nil {
|
|
return
|
|
}
|
|
if msg["type"] == "auth" {
|
|
_ = conn.WriteJSON(map[string]string{"type": "auth_ok"})
|
|
}
|
|
// Keep open briefly so client can mark connected.
|
|
time.Sleep(100 * time.Millisecond)
|
|
}))
|
|
defer server.Close()
|
|
|
|
jc := NewJobConnection()
|
|
managerURL := strings.Replace(server.URL, "http://", "http://", 1)
|
|
if err := jc.Connect(managerURL, "/job/1", "token123"); err != nil {
|
|
t.Fatalf("Connect failed: %v", err)
|
|
}
|
|
if !jc.IsConnected() {
|
|
t.Fatal("expected connection to be marked connected")
|
|
}
|
|
jc.Close()
|
|
}
|
|
|