github.com/pawelgaczynski/giouring@v0.0.0-20230826085535-69588b89acb9/register_test.go (about)

     1  // MIT License
     2  //
     3  // Copyright (c) 2023 Paweł Gaczyński
     4  //
     5  // Permission is hereby granted, free of charge, to any person obtaining a
     6  // copy of this software and associated documentation files (the
     7  // "Software"), to deal in the Software without restriction, including
     8  // without limitation the rights to use, copy, modify, merge, publish,
     9  // distribute, sublicense, and/or sell copies of the Software, and to
    10  // permit persons to whom the Software is furnished to do so, subject to
    11  // the following conditions:
    12  //
    13  // The above copyright notice and this permission notice shall be included
    14  // in all copies or substantial portions of the Software.
    15  //
    16  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
    17  // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
    18  // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
    19  // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
    20  // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
    21  // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
    22  // SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
    23  
    24  package giouring
    25  
    26  import (
    27  	"runtime"
    28  	"syscall"
    29  	"testing"
    30  	"time"
    31  
    32  	. "github.com/stretchr/testify/require"
    33  )
    34  
    35  func TestRegisterRingFd(t *testing.T) {
    36  	var maxSQEs uint32 = 1024
    37  	ring, err := CreateRing(maxSQEs)
    38  	Nil(t, err)
    39  
    40  	_, err = ring.RegisterRingFd()
    41  	Nil(t, err)
    42  
    43  	defer ring.QueueExit()
    44  
    45  	var (
    46  		entry *SubmissionQueueEntry
    47  		cqe   *CompletionQueueEvent
    48  	)
    49  
    50  	for i := 0; i < 1000; i++ {
    51  		entry = ring.GetSQE()
    52  		NotNil(t, entry)
    53  
    54  		entry.PrepareNop()
    55  		entry.UserData = uint64(i)
    56  
    57  		timespec := syscall.NsecToTimespec((time.Millisecond).Nanoseconds())
    58  		cqe, err = ring.SubmitAndWaitTimeout(1, &timespec, nil)
    59  		Nil(t, err)
    60  		NotNil(t, cqe)
    61  
    62  		runtime.KeepAlive(timespec)
    63  
    64  		ring.CQESeen(cqe)
    65  	}
    66  }