Function TestWriteReadyForQuery

server/protocol_test.go:332–368

(t *testing.T)

}

func TestWriteReadyForQuery(t *testing.T) {
	tests := []struct {
		name     string
		txStatus byte
	}{
		{"idle", 'I'},
		{"in transaction", 'T'},
		{"failed transaction", 'E'},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var buf bytes.Buffer

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

			// Message type should be 'Z'
			if buf.Bytes()[0] != 'Z' {
				t.Errorf("message type = %c, want Z", buf.Bytes()[0])
			}

			// Length should be 5 (4 for length + 1 for status)
			length := binary.BigEndian.Uint32(buf.Bytes()[1:5])
			if length != 5 {
				t.Errorf("length = %d, want 5", length)
			}

			// Check status
			if buf.Bytes()[5] != tt.txStatus {
				t.Errorf("txStatus = %c, want %c", buf.Bytes()[5], tt.txStatus)
			}
		})
	}
}

func TestWriteErrorResponse(t *testing.T) {
	var buf bytes.Buffer