github.com/timandy/routine@v1.1.4-0.20240507073150-e4a3e1fe2ba5/routine_test.go (about)

     1  package routine
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  )
     8  
     9  func TestInheritedTask(t *testing.T) {
    10  	tls := NewInheritableThreadLocal[string]()
    11  	it := inheritedTask{context: nil, function: func() {
    12  		assert.Equal(t, "", tls.Get())
    13  	}}
    14  	task := NewFutureTask(it.run)
    15  	go task.Run()
    16  	assert.Nil(t, task.Get())
    17  	assert.True(t, task.IsDone())
    18  	//
    19  	it2 := inheritedTask{context: nil, function: func() {
    20  		assert.Equal(t, "", tls.Get())
    21  	}}
    22  	task2 := NewFutureTask(it2.run)
    23  	go func() {
    24  		tls.Set("hello")
    25  		task2.Run()
    26  	}()
    27  	assert.Nil(t, task2.Get())
    28  	assert.True(t, task2.IsDone())
    29  	//
    30  	tls.Set("world")
    31  	it3 := inheritedTask{context: createInheritedMap(), function: func() {
    32  		assert.Equal(t, "world", tls.Get())
    33  	}}
    34  	task3 := NewFutureTask(it3.run)
    35  	go task3.Run()
    36  	assert.Nil(t, task3.Get())
    37  	assert.True(t, task3.IsDone())
    38  	//
    39  	it4 := inheritedTask{context: createInheritedMap(), function: func() {
    40  		assert.Equal(t, "world", tls.Get())
    41  	}}
    42  	task4 := NewFutureTask(it4.run)
    43  	go func() {
    44  		tls.Set("hello")
    45  		task4.Run()
    46  	}()
    47  	assert.Nil(t, task4.Get())
    48  	assert.True(t, task4.IsDone())
    49  }
    50  
    51  func TestInheritedWaitTask(t *testing.T) {
    52  	tls := NewInheritableThreadLocal[string]()
    53  	it := inheritedWaitTask{context: nil, function: func(token CancelToken) {
    54  		assert.Equal(t, "", tls.Get())
    55  	}}
    56  	task := NewFutureTask(it.run)
    57  	go task.Run()
    58  	assert.Nil(t, task.Get())
    59  	assert.True(t, task.IsDone())
    60  	//
    61  	it2 := inheritedWaitTask{context: nil, function: func(token CancelToken) {
    62  		assert.Equal(t, "", tls.Get())
    63  	}}
    64  	task2 := NewFutureTask(it2.run)
    65  	go func() {
    66  		tls.Set("hello")
    67  		task2.Run()
    68  	}()
    69  	assert.Nil(t, task2.Get())
    70  	assert.True(t, task2.IsDone())
    71  	//
    72  	tls.Set("world")
    73  	it3 := inheritedWaitTask{context: createInheritedMap(), function: func(token CancelToken) {
    74  		assert.Equal(t, "world", tls.Get())
    75  	}}
    76  	task3 := NewFutureTask(it3.run)
    77  	go task3.Run()
    78  	assert.Nil(t, task3.Get())
    79  	assert.True(t, task3.IsDone())
    80  	//
    81  	it4 := inheritedWaitTask{context: createInheritedMap(), function: func(token CancelToken) {
    82  		assert.Equal(t, "world", tls.Get())
    83  	}}
    84  	task4 := NewFutureTask(it4.run)
    85  	go func() {
    86  		tls.Set("hello")
    87  		task4.Run()
    88  	}()
    89  	assert.Nil(t, task4.Get())
    90  	assert.True(t, task4.IsDone())
    91  }
    92  
    93  func TestInheritedWaitResultTask(t *testing.T) {
    94  	tls := NewInheritableThreadLocal[string]()
    95  	it := inheritedWaitResultTask[string]{context: nil, function: func(token CancelToken) string {
    96  		assert.Equal(t, "", tls.Get())
    97  		return tls.Get()
    98  	}}
    99  	task := NewFutureTask(it.run)
   100  	go task.Run()
   101  	assert.Equal(t, "", task.Get())
   102  	assert.True(t, task.IsDone())
   103  	//
   104  	it2 := inheritedWaitResultTask[string]{context: nil, function: func(token CancelToken) string {
   105  		assert.Equal(t, "", tls.Get())
   106  		return tls.Get()
   107  	}}
   108  	task2 := NewFutureTask(it2.run)
   109  	go func() {
   110  		tls.Set("hello")
   111  		task2.Run()
   112  	}()
   113  	assert.Equal(t, "", task2.Get())
   114  	assert.True(t, task2.IsDone())
   115  	//
   116  	tls.Set("world")
   117  	it3 := inheritedWaitResultTask[string]{context: createInheritedMap(), function: func(token CancelToken) string {
   118  		assert.Equal(t, "world", tls.Get())
   119  		return tls.Get()
   120  	}}
   121  	task3 := NewFutureTask(it3.run)
   122  	go task3.Run()
   123  	assert.Equal(t, "world", task3.Get())
   124  	assert.True(t, task3.IsDone())
   125  	//
   126  	it4 := inheritedWaitResultTask[string]{context: createInheritedMap(), function: func(token CancelToken) string {
   127  		assert.Equal(t, "world", tls.Get())
   128  		return tls.Get()
   129  	}}
   130  	task4 := NewFutureTask(it4.run)
   131  	go func() {
   132  		tls.Set("hello")
   133  		task4.Run()
   134  	}()
   135  	assert.Equal(t, "world", task4.Get())
   136  	assert.True(t, task4.IsDone())
   137  }