fix logger test and locker test

go-co-op · Nov 14, 2023 · 66caa1a · 66caa1a
1 parent 17ea8ca
commit 66caa1a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/logger_test.go b/logger_test.go
@@ -38,6 +38,10 @@ func TestNewLogger(t *testing.T) {
 			"error",
 			LogLevelError,
 		},
+		{
+			"Less than error",
+			-1,
+		},
 	}
 
 	for _, tt := range tests {

diff --git a/scheduler_test.go b/scheduler_test.go
@@ -1024,6 +1024,7 @@ func (t *testLocker) Lock(_ context.Context, _ string) (Lock, error) {
 	if t.jobLocked {
 		return nil, fmt.Errorf("job already locked")
 	}
+	t.jobLocked = true
 	return &testLock{}, nil
 }
 
@@ -1056,15 +1057,14 @@ func TestScheduler_WithDistributed(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			elector := &testElector{}
 			jobsRan := make(chan struct{}, 20)
 			ctx, cancel := context.WithCancel(context.Background())
 			schedulersDone := make(chan struct{}, tt.count)
 
 			for i := tt.count; i > 0; i-- {
 				go func() {
 					s, err := newTestScheduler(
-						WithDistributedElector(elector),
+						tt.opt,
 					)
 					require.NoError(t, err)