Skip to content

Commit

Permalink
add test for query backend
Browse files Browse the repository at this point in the history
  • Loading branch information
ainghazal committed Jun 23, 2024
1 parent b8ac182 commit dda649a
Showing 1 changed file with 43 additions and 0 deletions.
43 changes: 43 additions & 0 deletions internal/experiment/openvpn/richerinput_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,9 @@ package openvpn
import (
"context"
"errors"
"fmt"
"testing"
"time"

"github.com/google/go-cmp/cmp"
"github.com/ooni/probe-cli/v3/internal/mocks"
Expand Down Expand Up @@ -173,3 +175,44 @@ func TestTargetLoaderLoad(t *testing.T) {
})
}
}

func TestTargetLoaderLoadFromBackend(t *testing.T) {
loader := &targetloading.Loader{
ExperimentName: "openvpn",
InputPolicy: model.InputOrQueryBackend,
Logger: model.DiscardLogger,
Session: &mocks.Session{},
}
sess := &mocks.Session{}
sess.MockFetchOpenVPNConfig = func(context.Context, string, string) (*model.OOAPIVPNProviderConfig, error) {
return &model.OOAPIVPNProviderConfig{
Provider: "riseupvpn",
Config: &model.OOAPIVPNConfig{},
Inputs: []string{
"openvpn://target0",
"openvpn://target1",
},
DateUpdated: time.Now(),
}, nil
}
tl := &targetLoader{
loader: loader,
options: &Config{},
session: sess,
}
targets, err := tl.Load(context.Background())
if err != nil {
t.Fatal("expected no error")
}
fmt.Println("targets", targets)
if len(targets) != 2 {
t.Fatal("expected 2 targets")
}
if targets[0].String() != "openvpn://target0" {
t.Fatal("expected openvpn://target0")
}
if targets[1].String() != "openvpn://target1" {
t.Fatal("expected openvpn://target1")
}

}

0 comments on commit dda649a

Please sign in to comment.