oonimkall: add wrappers for /test-list/urls (#221)

* First wrapper implementation

* Adding CountryCode parameter

* requested changes

* Update pkg/oonimkall/session.go

Co-authored-by: Simone Basso <bassosimone@gmail.com>

* Requested changes

* Update pkg/oonimkall/session_integration_test.go

Co-authored-by: Simone Basso <bassosimone@gmail.com>

* Update pkg/oonimkall/session_integration_test.go

Co-authored-by: Simone Basso <bassosimone@gmail.com>

* Remove duplicated code

* Apply suggestions from code review

Co-authored-by: Simone Basso <bassosimone@gmail.com>
This commit is contained in:
Lorenzo Primiterra 2021-02-26 09:00:25 +00:00 committed by GitHub
commit 772de83a06
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 121 additions and 0 deletions

View file

@ -419,6 +419,60 @@ func TestCheckInNoParams(t *testing.T) {
}
}
func TestFetchURLListSuccess(t *testing.T) {
sess, err := NewSession()
if err != nil {
t.Fatal(err)
}
ctx := sess.NewContext()
config := oonimkall.URLListConfig{
Limit: 10,
}
config.AddCategory("NEWS")
config.AddCategory("CULTR")
result, err := sess.FetchURLList(ctx, &config)
if err != nil {
t.Fatalf("unexpected error: %+v", err)
}
if result == nil || result.Results == nil {
t.Fatal("got nil result")
}
for _, entry := range result.Results {
if entry.CategoryCode != "NEWS" && entry.CategoryCode != "CULTR" {
t.Fatalf("unexpected category code: %+v", entry)
}
}
}
func TestFetchURLListWithCC(t *testing.T) {
sess, err := NewSession()
if err != nil {
t.Fatal(err)
}
ctx := sess.NewContext()
config := oonimkall.URLListConfig{
CountryCode: "IT",
}
config.AddCategory("NEWS")
config.AddCategory("CULTR")
result, err := sess.FetchURLList(ctx, &config)
if err != nil {
t.Fatalf("unexpected error: %+v", err)
}
if result == nil || result.Results == nil {
t.Fatal("got nil result")
}
found := false
for _, entry := range result.Results {
if entry.CountryCode == "IT" {
found = true
}
}
if !found {
t.Fatalf("not found url for country code: IT")
}
}
func TestMain(m *testing.M) {
// Here we're basically testing whether eventually the finalizers
// will run and the number of active sessions and cancels will become