(t *testing.T)
| 14 | } |
| 15 | |
| 16 | func TestInflowTake(t *testing.T) { |
| 17 | var f inflow |
| 18 | f.init(1000) |
| 19 | if !f.take(500) { |
| 20 | t.Fatal("take(500) should succeed with avail=1000") |
| 21 | } |
| 22 | if f.avail != 500 { |
| 23 | t.Fatalf("avail = %d, want 500 after take(500)", f.avail) |
| 24 | } |
| 25 | if f.take(501) { |
| 26 | t.Fatal("take(501) should fail with avail=500") |
| 27 | } |
| 28 | } |
| 29 | |
| 30 | func TestInflowAdd(t *testing.T) { |
| 31 | var f inflow |