25 lines
401 B
Go
25 lines
401 B
Go
|
package dialid
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGood(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
id := ContextDialID(ctx)
|
||
|
|
if id != 0 {
|
||
|
|
t.Fatal("unexpected ID for empty context")
|
||
|
|
}
|
||
|
|
ctx = WithDialID(ctx)
|
||
|
|
id = ContextDialID(ctx)
|
||
|
|
if id != 1 {
|
||
|
|
t.Fatal("expected ID equal to 1")
|
||
|
|
}
|
||
|
|
ctx = WithDialID(ctx)
|
||
|
|
id = ContextDialID(ctx)
|
||
|
|
if id != 2 {
|
||
|
|
t.Fatal("expected ID equal to 2")
|
||
|
|
}
|
||
|
|
}
|