package osbuild

import (
	"bytes"
	"encoding/json"
	"reflect"
	"testing"
)

func TestSource_UnmarshalJSON(t *testing.T) {
	type fields struct {
		Name   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.files"}`),
			},
			wantErr: true,
		},
		{
			name: "missing name",
			args: args{
				data: []byte(`{"foo":null,"options":{"bar":null}}`),
			},
			wantErr: true,
		},
		{
			name: "files-empty",
			fields: fields{
				Name:   "org.osbuild.files",
				Source: &FilesSource{URLs: map[string]FileSource{}},
			},
			args: args{
				data: []byte(`{"org.osbuild.files":{"urls":{}}}`),
			},
		},
		{
			name: "files",
			fields: fields{
				Name: "org.osbuild.files",
				Source: &FilesSource{URLs: map[string]FileSource{
					"checksum1": FileSource{URL: "url1"},
					"checksum2": FileSource{URL: "url2"},
				}},
			},
			args: args{
				data: []byte(`{"org.osbuild.files":{"urls":{"checksum1":{"url":"url1"},"checksum2":{"url":"url2"}}}}`),
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			sources := &Sources{
				tt.fields.Name: 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", err, tt.wantErr)
			}
			if tt.wantErr {
				return
			}
			gotBytes, err := json.Marshal(sources)
			if err != nil {
				t.Errorf("Could not marshal source: %v", err)
			}
			if !bytes.Equal(gotBytes, tt.args.data) {
				t.Errorf("Expected '%v', got '%v'", string(tt.args.data), string(gotBytes))
			}
			if !reflect.DeepEqual(&gotSources, sources) {
				t.Errorf("got '%v', expected '%v'", &gotSources, sources)
			}
		})
	}
}