47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
|
package probeservices_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/engine/probeservices"
|
||
|
|
"github.com/ooni/probe-cli/v3/internal/engine/probeservices/testorchestra"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFetchPsiphonConfig(t *testing.T) {
|
||
|
|
clnt := newclient()
|
||
|
|
if err := clnt.MaybeRegister(context.Background(), testorchestra.MetadataFixture()); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if err := clnt.MaybeLogin(context.Background()); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
data, err := clnt.FetchPsiphonConfig(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
var config interface{}
|
||
|
|
if err := json.Unmarshal(data, &config); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFetchPsiphonConfigNotRegistered(t *testing.T) {
|
||
|
|
clnt := newclient()
|
||
|
|
state := probeservices.State{
|
||
|
|
// Explicitly empty so the test is more clear
|
||
|
|
}
|
||
|
|
if err := clnt.StateFile.Set(state); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
data, err := clnt.FetchPsiphonConfig(context.Background())
|
||
|
|
if !errors.Is(err, probeservices.ErrNotRegistered) {
|
||
|
|
t.Fatal("expected an error here")
|
||
|
|
}
|
||
|
|
if data != nil {
|
||
|
|
t.Fatal("expected nil data here")
|
||
|
|
}
|
||
|
|
}
|