65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package htlcswitch
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
)
|
|
|
|
// TestWaitingQueueThreadSafety test the thread safety properties of the
|
|
// waiting queue, by executing methods in separate goroutines which operates
|
|
// with the same data.
|
|
func TestWaitingQueueThreadSafety(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
const numPkts = 1000
|
|
|
|
q := newPacketQueue(numPkts)
|
|
q.Start()
|
|
defer q.Stop()
|
|
|
|
a := make([]uint64, numPkts)
|
|
for i := 0; i < numPkts; i++ {
|
|
a[i] = uint64(i)
|
|
q.AddPkt(&htlcPacket{
|
|
incomingHTLCID: a[i],
|
|
htlc: &lnwire.UpdateAddHTLC{},
|
|
})
|
|
}
|
|
|
|
// The reported length of the queue should be the exact number of
|
|
// packets we added above.
|
|
queueLength := q.Length()
|
|
if queueLength != numPkts {
|
|
t.Fatalf("queue has wrong length: expected %v, got %v", numPkts,
|
|
queueLength)
|
|
}
|
|
|
|
var b []uint64
|
|
for i := 0; i < numPkts; i++ {
|
|
q.SignalFreeSlot()
|
|
|
|
select {
|
|
case packet := <-q.outgoingPkts:
|
|
b = append(b, packet.incomingHTLCID)
|
|
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("timeout")
|
|
}
|
|
}
|
|
|
|
// The length of the queue should be zero at this point.
|
|
time.Sleep(time.Millisecond * 50)
|
|
queueLength = q.Length()
|
|
if queueLength != 0 {
|
|
t.Fatalf("queue has wrong length: expected %v, got %v", 0,
|
|
queueLength)
|
|
}
|
|
|
|
if !reflect.DeepEqual(b, a) {
|
|
t.Fatal("wrong order of the objects")
|
|
}
|
|
}
|