status-go/jail/internal/loop/loop_test.go

118 lines
2.2 KiB
Go

package loop
import (
"context"
"sync/atomic"
"testing"
"time"
"github.com/status-im/status-go/jail/internal/vm"
"github.com/stretchr/testify/suite"
)
// DummyTask is something that satisfies the loop.Task interface for testing.
type DummyTask struct {
canceled int32
executed int32
}
func (*DummyTask) SetID(int64) {}
func (*DummyTask) GetID() int64 { return 1 }
func (d *DummyTask) Cancel() { atomic.StoreInt32(&d.canceled, 1) }
func (d *DummyTask) Execute(*vm.VM, *Loop) error {
atomic.StoreInt32(&d.executed, 1)
return nil
}
func (d *DummyTask) Canceled() bool {
return atomic.LoadInt32(&d.canceled) == 1
}
func (d *DummyTask) Executed() bool {
return atomic.LoadInt32(&d.executed) == 1
}
func TestLoopSuite(t *testing.T) {
suite.Run(t, new(LoopSuite))
}
type LoopSuite struct {
suite.Suite
loop *Loop
cancel context.CancelFunc
task *DummyTask
}
func (s *LoopSuite) SetupTest() {
s.task = &DummyTask{}
newVM := vm.New()
s.loop = New(newVM)
ctx, cancel := context.WithCancel(context.Background())
s.cancel = cancel
go func() {
a := s.Assertions // Cache assertions reference as otherwise we'd incur in a race condition
err := s.loop.Run(ctx)
a.Equal(context.Canceled, err)
}()
}
func (s *LoopSuite) TestAddAndReady() {
err := s.loop.Add(s.task)
s.NoError(err)
s.False(s.task.Canceled())
err = s.loop.Ready(s.task)
s.NoError(err)
// Wait to process task
time.Sleep(100 * time.Millisecond)
s.True(s.task.Executed())
s.cancel()
}
func (s *LoopSuite) TestLoopErrorWhenClosed() {
s.cancel()
// Wait for the context to cancel and loop to close
time.Sleep(100 * time.Millisecond)
err := s.loop.Add(s.task)
s.Error(err)
err = s.loop.Ready(s.task)
s.Error(err)
s.True(s.task.Canceled())
}
func (s *LoopSuite) TestImmediateExecution() {
err := s.loop.AddAndExecute(s.task)
// Wait for the task to execute
time.Sleep(100 * time.Millisecond)
s.NoError(err)
s.True(s.task.Executed())
s.False(s.task.Canceled())
s.cancel()
}
func (s *LoopSuite) TestImmediateExecutionErrorWhenClosed() {
s.cancel()
// Wait for the context to cancel and loop to close
time.Sleep(100 * time.Millisecond)
err := s.loop.AddAndExecute(s.task)
s.Error(err)
s.False(s.task.Executed())
}