Function TestReadStartupMessage

server/protocol_test.go:12–113

(t *testing.T)

)

func TestReadStartupMessage(t *testing.T) {
	t.Run("valid startup message", func(t *testing.T) {
		// Build a startup message:
		// - 4 bytes length (including itself)
		// - 4 bytes protocol version (196608 = 3.0)
		// - key-value pairs null-terminated
		// - final null byte
		var buf bytes.Buffer

		params := map[string]string{
			"user":     "testuser",
			"database": "testdb",
		}

		// Calculate length
		bodyLen := 4 // protocol version
		for k, v := range params {
			bodyLen += len(k) + 1 + len(v) + 1
		}
		bodyLen++ // final null

		// Write length (includes itself)
		_ = binary.Write(&buf, binary.BigEndian, int32(bodyLen+4))

		// Write protocol version (3.0 = 196608)
		_ = binary.Write(&buf, binary.BigEndian, uint32(196608))

		// Write params
		for k, v := range params {
			buf.WriteString(k)
			buf.WriteByte(0)
			buf.WriteString(v)
			buf.WriteByte(0)
		}
		buf.WriteByte(0) // final null

		result, err := wire.ReadStartupMessage(&buf)
		if err != nil {
			t.Fatalf("wire.ReadStartupMessage() error = %v", err)
		}

		if result.Params["user"] != "testuser" {
			t.Errorf("user = %q, want %q", result.Params["user"], "testuser")
		}
		if result.Params["database"] != "testdb" {
			t.Errorf("database = %q, want %q", result.Params["database"], "testdb")
		}
	})

	t.Run("SSL request", func(t *testing.T) {
		var buf bytes.Buffer

		// SSL request: length=8, version=80877103
		_ = binary.Write(&buf, binary.BigEndian, int32(8))
		_ = binary.Write(&buf, binary.BigEndian, uint32(80877103))

		result, err := wire.ReadStartupMessage(&buf)
		if err != nil {