42 lines
785 B
Go
42 lines
785 B
Go
|
// Copyright 2012, Google Inc. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package sync2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSemaNoTimeout(t *testing.T) {
|
||
|
|
s := NewSemaphore(1, 0)
|
||
|
|
s.Acquire()
|
||
|
|
released := false
|
||
|
|
go func() {
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
released = true
|
||
|
|
s.Release()
|
||
|
|
}()
|
||
|
|
s.Acquire()
|
||
|
|
if !released {
|
||
|
|
t.Errorf("want true, got false")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSemaTimeout(t *testing.T) {
|
||
|
|
s := NewSemaphore(1, 5*time.Millisecond)
|
||
|
|
s.Acquire()
|
||
|
|
go func() {
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
s.Release()
|
||
|
|
}()
|
||
|
|
if ok := s.Acquire(); ok {
|
||
|
|
t.Errorf("want false, got true")
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
if ok := s.Acquire(); !ok {
|
||
|
|
t.Errorf("want true, got false")
|
||
|
|
}
|
||
|
|
}
|