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

Fix stale test logic

Browse source 
The model processing was recently changed to be deferred but
this test scenario hadn't been adjusted for that change in behavior.
This commit is contained in:
Daniel Hiltgen 2024-05-06 14:15:37 -07:00
parent aa93423fbf
commit 0a954e5066
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
server/sched_test.go
View file

@ -352,11 +352,9 @@ func TestGetRunner(t *testing.T) {
scenario1c.req.model.ModelPath = "bad path" scenario1c.req.model.ModelPath = "bad path"
slog.Info("scenario1c") slog.Info("scenario1c")
successCh1c, errCh1c := s.GetRunner(scenario1c.ctx, scenario1c.req.model, scenario1c.req.opts, scenario1c.req.sessionDuration) successCh1c, errCh1c := s.GetRunner(scenario1c.ctx, scenario1c.req.model, scenario1c.req.opts, scenario1c.req.sessionDuration)
require.Len(t, s.pendingReqCh, 0) // Starts in pending channel, then should be quickly processsed to return an error
require.Len(t, successCh1c, 0)
require.Len(t, errCh1c, 0)
time.Sleep(5 * time.Millisecond) time.Sleep(5 * time.Millisecond)
require.Len(t, successCh1c, 0)
s.loadedMu.Lock() s.loadedMu.Lock()
require.Len(t, s.loaded, 0) require.Len(t, s.loaded, 0)
s.loadedMu.Unlock() s.loadedMu.Unlock()