115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package osbuild
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestSource_UnmarshalJSON(t *testing.T) {
|
|
type fields struct {
|
|
Type string
|
|
Source Source
|
|
}
|
|
type args struct {
|
|
data []byte
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "invalid json",
|
|
args: args{
|
|
data: []byte(`{"name":"org.osbuild.foo","options":{"bar":null}`),
|
|
},
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "unknown source",
|
|
args: args{
|
|
data: []byte(`{"name":"org.osbuild.foo","options":{"bar":null}}`),
|
|
},
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "missing options",
|
|
args: args{
|
|
data: []byte(`{"name":"org.osbuild.curl"}`),
|
|
},
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "missing name",
|
|
args: args{
|
|
data: []byte(`{"foo":null,"options":{"bar":null}}`),
|
|
},
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "curl-empty",
|
|
fields: fields{
|
|
Type: "org.osbuild.curl",
|
|
Source: &CurlSource{Items: map[string]CurlSourceItem{}},
|
|
},
|
|
args: args{
|
|
data: []byte(`{"org.osbuild.curl":{"items":{}}}`),
|
|
},
|
|
},
|
|
{
|
|
name: "curl-with-secrets",
|
|
fields: fields{
|
|
Type: "org.osbuild.curl",
|
|
Source: &CurlSource{
|
|
Items: map[string]CurlSourceItem{
|
|
"checksum1": CurlSourceOptions{URL: "url1", Secrets: &URLSecrets{Name: "org.osbuild.rhsm"}},
|
|
"checksum2": CurlSourceOptions{URL: "url2", Secrets: &URLSecrets{Name: "whatever"}},
|
|
}},
|
|
},
|
|
args: args{
|
|
data: []byte(`{"org.osbuild.curl":{"items":{"checksum1":{"url":"url1","secrets":{"name":"org.osbuild.rhsm"}},"checksum2":{"url":"url2","secrets":{"name":"whatever"}}}}}`),
|
|
},
|
|
},
|
|
{
|
|
name: "curl-url-only",
|
|
fields: fields{
|
|
Type: "org.osbuild.curl",
|
|
Source: &CurlSource{
|
|
Items: map[string]CurlSourceItem{
|
|
"checksum1": URL("url1"),
|
|
"checksum2": URL("url2"),
|
|
}},
|
|
},
|
|
args: args{
|
|
data: []byte(`{"org.osbuild.curl":{"items":{"checksum1":"url1","checksum2":"url2"}}}`),
|
|
},
|
|
},
|
|
}
|
|
for idx, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
sources := &Sources{
|
|
tt.fields.Type: tt.fields.Source,
|
|
}
|
|
var gotSources Sources
|
|
if err := gotSources.UnmarshalJSON(tt.args.data); (err != nil) != tt.wantErr {
|
|
t.Errorf("Sources.UnmarshalJSON() error = %v, wantErr %v [idx: %d]", err, tt.wantErr, idx)
|
|
}
|
|
if tt.wantErr {
|
|
return
|
|
}
|
|
gotBytes, err := json.Marshal(sources)
|
|
if err != nil {
|
|
t.Errorf("Could not marshal source: %v [idx: %d]", err, idx)
|
|
}
|
|
if !bytes.Equal(gotBytes, tt.args.data) {
|
|
t.Errorf("Expected '%v', got '%v' [idx: %d]", string(tt.args.data), string(gotBytes), idx)
|
|
}
|
|
if !reflect.DeepEqual(&gotSources, sources) {
|
|
t.Errorf("got '%v', expected '%v' [idx:%d]", &gotSources, sources, idx)
|
|
}
|
|
})
|
|
}
|
|
}
|