feat: context-based tracing to record delayed DNS responses (#870)
See https://github.com/ooni/probe/issues/2221 Co-authored-by: decfox <decfox@github.com> Co-authored-by: Simone Basso <bassosimone@gmail.com>
This commit is contained in:
parent
fe6d378a1f
commit
2301a30630
10 changed files with 498 additions and 317 deletions
|
|
@ -71,6 +71,30 @@ func TestTrace(t *testing.T) {
|
|||
}
|
||||
})
|
||||
|
||||
t.Run("OnDelayedDNSResponse", func(t *testing.T) {
|
||||
var called bool
|
||||
tx := &Trace{
|
||||
MockOnDelayedDNSResponse: func(started time.Time, txp model.DNSTransport,
|
||||
query model.DNSQuery, response model.DNSResponse,
|
||||
addrs []string, err error, finished time.Time) error {
|
||||
called = true
|
||||
return nil
|
||||
},
|
||||
}
|
||||
tx.OnDelayedDNSResponse(
|
||||
time.Now(),
|
||||
&DNSTransport{},
|
||||
&DNSQuery{},
|
||||
&DNSResponse{},
|
||||
[]string{},
|
||||
nil,
|
||||
time.Now(),
|
||||
)
|
||||
if !called {
|
||||
t.Fatal("not called")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("OnConnectDone", func(t *testing.T) {
|
||||
var called bool
|
||||
tx := &Trace{
|
||||
|
|
|
|||
Loading…
Reference in a new issue