gered/ollama
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Harden sched TestLoad

Browse source 
Give the go routine a moment to deliver the expired event
This commit is contained in:
Daniel Hiltgen 2024-04-23 13:07:16 -07:00
parent 058f6cd2cc
commit d8851cb7a0
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
server/sched_test.go
View file

@ -92,6 +92,7 @@ func TestLoad(t *testing.T) {
runner := s.loaded["dummy_model_path"] runner := s.loaded["dummy_model_path"]
require.NotNil(t, runner) require.NotNil(t, runner)
require.Equal(t, uint(0), runner.refCount) require.Equal(t, uint(0), runner.refCount)
time.Sleep(1 * time.Millisecond)
require.Len(t, s.expiredCh, 1) require.Len(t, s.expiredCh, 1)
} }