package osbuild

import (

	"encoding/json"

	"testing"

	"github.com/google/uuid"

	"github.com/stretchr/testify/assert"

)

func TestAssembler_UnmarshalJSON(t *testing.T) {

	tests := []struct {

		name          string

		assembler     Assembler

		data          []byte

		errorExpected bool

	}{

		{

			// invalid JSON - note the missing brace at the end of the string

			name:          "invalid json",

			data:          []byte(`{"name":"org.osbuild.tar","options":{"filename":""}`),

			errorExpected: true,

		},

		{

			// valid JSON, but with an unknown assembler (org.osbuild.foo)

			name:          "unknown assembler",

			data:          []byte(`{"name":"org.osbuild.foo","options":{"bar":null}}`),

			errorExpected: true,

		},

		{

			name:          "missing options",

			data:          []byte(`{"name":"org.osbuild.rawfs"`),

			errorExpected: true,

		},

		{

			name:          "missing name",

			data:          []byte(`{"options":{"bar":null}}`),

			errorExpected: true,

		},

		{

			name: "qemu assembler empty",

			assembler: Assembler{

				Name:    "org.osbuild.qemu",

				Options: &QEMUAssemblerOptions{},

			},

			data: []byte(`{"name":"org.osbuild.qemu","options":{"format":"","filename":"","size":0,"ptuuid":"","pttype":"","partitions":null}}`),

		},

		{

			name: "qemu assembler full",

			assembler: Assembler{

				Name: "org.osbuild.qemu",

				Options: &QEMUAssemblerOptions{

					Format:   "qcow2",

					Filename: "disk.qcow2",

					Size:     2147483648,

					PTUUID:   "0x14fc63d2",

					PTType:   "mbr",

					Partitions: []QEMUPartition{QEMUPartition{

						Start:    2048,

						Bootable: true,

						Filesystem: &QEMUFilesystem{

							Type:       "ext4",

							UUID:       "76a22bf4-f153-4541-b6c7-0332c0dfaeac",

							Label:      "root",

							Mountpoint: "/",

						},

					}},

				},

			},

			data: []byte(`{"name":"org.osbuild.qemu","options":{"format":"qcow2","filename":"disk.qcow2","size":2147483648,"ptuuid":"0x14fc63d2","pttype":"mbr","partitions":[{"start":2048,"bootable":true,"filesystem":{"type":"ext4","uuid":"76a22bf4-f153-4541-b6c7-0332c0dfaeac","label":"root","mountpoint":"/"}}]}}`),

		},

		{

			name: "tar assembler empty",

			assembler: Assembler{

				Name:    "org.osbuild.tar",

				Options: &TarAssemblerOptions{},

			},

			data: []byte(`{"name":"org.osbuild.tar","options":{"filename":""}}`),

		},

		{

			name: "tar assembler full",

			assembler: Assembler{

				Name: "org.osbuild.tar",

				Options: &TarAssemblerOptions{

					Filename:    "root.tar.xz",

					Compression: "xz",

				},

			},

			data: []byte(`{"name":"org.osbuild.tar","options":{"filename":"root.tar.xz","compression":"xz"}}`),

		},

		{

			name: "rawfs assembler empty",

			assembler: Assembler{

				Name:    "org.osbuild.rawfs",

				Options: &RawFSAssemblerOptions{},

			},

			data: []byte(`{"name":"org.osbuild.rawfs","options":{"filename":"","root_fs_uuid":"00000000-0000-0000-0000-000000000000","size":0}}`),

		},

		{

			name: "rawfs assembler full",

			assembler: Assembler{

				Name: "org.osbuild.rawfs",

				Options: &RawFSAssemblerOptions{

					Filename:           "filesystem.img",

					RootFilesystemUUID: uuid.MustParse("76a22bf4-f153-4541-b6c7-0332c0dfaeac"),

					Size:               2147483648,

				},

			},

			data: []byte(`{"name":"org.osbuild.rawfs","options":{"filename":"filesystem.img","root_fs_uuid":"76a22bf4-f153-4541-b6c7-0332c0dfaeac","size":2147483648}}`),

		},

		{

			name: "ostree commit assembler",

			assembler: Assembler{

				Name: "org.osbuild.ostree.commit",

				Options: &OSTreeCommitAssemblerOptions{

					Ref: "foo",

					Tar: OSTreeCommitAssemblerTarOptions{

						Filename: "foo.tar",

					},

				},

			},

			data: []byte(`{"name":"org.osbuild.ostree.commit","options":{"ref":"foo","tar":{"filename":"foo.tar"}}}`),

		},

	}

	assert := assert.New(t)

	for _, tt := range tests {

		t.Run(tt.name, func(t *testing.T) {

			assembler := &tt.assembler

			var gotAssembler Assembler

			err := gotAssembler.UnmarshalJSON(tt.data)

			if tt.errorExpected {

				assert.NotNil(err)

				return

			} else {

				assert.Nil(err)

			}

			gotBytes, err := json.Marshal(assembler)

			assert.Nilf(err, "Could not marshal assembler: %v", err)

			assert.Equal(tt.data, gotBytes)

			assert.Equal(&gotAssembler, assembler)

		})

	}

}

func TestNewQEMUAssembler(t *testing.T) {

	options := &QEMUAssemblerOptions{}

	expectedAssembler := &Assembler{

		Name:    "org.osbuild.qemu",

		Options: &QEMUAssemblerOptions{},

	}

	assert.Equal(t, expectedAssembler, NewQEMUAssembler(options))

}

func TestNewTarAssembler(t *testing.T) {

	options := &TarAssemblerOptions{}

	expectedAssembler := &Assembler{

		Name:    "org.osbuild.tar",

		Options: &TarAssemblerOptions{},

	}

	assert.Equal(t, expectedAssembler, NewTarAssembler(options))

}

func TestNewRawFSAssembler(t *testing.T) {

	options := &RawFSAssemblerOptions{}

	expectedAssembler := &Assembler{

		Name:    "org.osbuild.rawfs",

		Options: &RawFSAssemblerOptions{},

	}

	assert.Equal(t, expectedAssembler, NewRawFSAssembler(options))

}