refactor(netx): move construction logic outside package (#798)
For testability, replace most if-based construction logic with calls to well-tested factories living in other packages. While there, acknowledge that a bunch of types could now be private and make them private, modifying the code to call the public factories allowing to construct said types instead. Part of https://github.com/ooni/probe/issues/2121
This commit is contained in:
parent
2d3d5d9cdc
commit
6b85dfce88
20 changed files with 475 additions and 184 deletions
|
|
@ -15,6 +15,32 @@ import (
|
|||
"github.com/ooni/probe-cli/v3/internal/netxlite/filtering"
|
||||
)
|
||||
|
||||
func TestMaybeWrapHTTPTransport(t *testing.T) {
|
||||
const snapshotSize = 1024
|
||||
|
||||
t.Run("with non-nil saver", func(t *testing.T) {
|
||||
saver := &Saver{}
|
||||
underlying := &mocks.HTTPTransport{}
|
||||
txp := saver.MaybeWrapHTTPTransport(underlying, snapshotSize)
|
||||
realTxp := txp.(*HTTPTransportSaver)
|
||||
if realTxp.HTTPTransport != underlying {
|
||||
t.Fatal("unexpected result")
|
||||
}
|
||||
if realTxp.SnapshotSize != snapshotSize {
|
||||
t.Fatal("did not set snapshotSize correctly")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("with nil saver", func(t *testing.T) {
|
||||
var saver *Saver
|
||||
underlying := &mocks.HTTPTransport{}
|
||||
txp := saver.MaybeWrapHTTPTransport(underlying, snapshotSize)
|
||||
if txp != underlying {
|
||||
t.Fatal("unexpected result")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestHTTPTransportSaver(t *testing.T) {
|
||||
|
||||
t.Run("CloseIdleConnections", func(t *testing.T) {
|
||||
|
|
|
|||
Loading…
Reference in a new issue