package scanner

import (
	"testing"

	"github.com/future-architect/vuls/constant"
	"github.com/future-architect/vuls/models"
)

func Test_isRunningKernel(t *testing.T) {
	type args struct {
		pack    models.Package
		family  string
		release string
		kernel  models.Kernel
	}
	tests := []struct {
		name         string
		args         args
		wantIsKernel bool
		wantRunning  bool
	}{
		{
			name: "Amazon not kernel",
			args: args{
				pack: models.Package{
					Name:    "kernel-livepatch-4.14.165-131.185",
					Version: "1.0",
					Release: "0.amzn1",
					Arch:    "x86_64",
				},
				family:  constant.Amazon,
				release: "1",
				kernel: models.Kernel{
					Release: "4.9.43-17.38.amzn1.x86_64",
				},
			},
			wantIsKernel: false,
			wantRunning:  false,
		},
		{
			name: "Amazon kernel and running",
			args: args{
				pack: models.Package{
					Name:    "kernel",
					Version: "4.9.43",
					Release: "17.38.amzn1",
					Arch:    "x86_64",
				},
				family:  constant.Amazon,
				release: "1",
				kernel: models.Kernel{
					Release: "4.9.43-17.38.amzn1.x86_64",
				},
			},
			wantIsKernel: true,
			wantRunning:  true,
		},
		{
			name: "Amazon kernel but not running",
			args: args{
				pack: models.Package{
					Name:    "kernel",
					Version: "4.9.38",
					Release: "16.35.amzn1",
					Arch:    "x86_64",
				},
				family:  constant.Amazon,
				release: "1",
				kernel: models.Kernel{
					Release: "4.9.43-17.38.amzn1.x86_64",
				},
			},
			wantIsKernel: true,
			wantRunning:  false,
		},
		{
			name: "SUES not kernel",
			args: args{
				pack: models.Package{
					Name:    "bash",
					Version: "4.4",
					Release: "19.6.1",
					Arch:    "x86_64",
				},
				family:  constant.SUSEEnterpriseServer,
				release: "12.2",
				kernel: models.Kernel{
					Release: "4.4.74-92.35-default",
				},
			},
			wantIsKernel: false,
			wantRunning:  false,
		},
		{
			name: "SUSE kernel and running",
			args: args{
				pack: models.Package{
					Name:    "kernel-default",
					Version: "4.4.74",
					Release: "92.35.1",
					Arch:    "x86_64",
				},
				family:  constant.SUSEEnterpriseServer,
				release: "12.2",
				kernel: models.Kernel{
					Release: "4.4.74-92.35-default",
				},
			},
			wantIsKernel: true,
			wantRunning:  true,
		},
		{
			name: "SUES kernel but not running",
			args: args{
				pack: models.Package{
					Name:    "kernel-default",
					Version: "4.4.59",
					Release: "92.20.2",
					Arch:    "x86_64",
				},
				family:  constant.SUSEEnterpriseServer,
				release: "12.2",
				kernel: models.Kernel{
					Release: "4.4.74-92.35-default",
				},
			},
			wantIsKernel: true,
			wantRunning:  false,
		},
		{
			name: "kernel is kernel-debug, but pack is kernel",
			args: args{
				pack: models.Package{
					Name:    "kernel",
					Version: "5.14.0",
					Release: "70.13.1.el9_0",
					Arch:    "x86_64",
				},
				family:  constant.RedHat,
				release: "9.0",
				kernel: models.Kernel{
					Release: "5.14.0-427.13.1.el9_4.x86_64+debug",
				},
			},
			wantIsKernel: true,
			wantRunning:  false,
		},
		{
			name: "old redhat kernel release style",
			args: args{
				pack: models.Package{
					Name:    "kernel-debug",
					Version: "2.6.18",
					Release: "419.el5",
					Arch:    "x86_64",
				},
				family:  constant.RedHat,
				release: "5.11",
				kernel: models.Kernel{
					Release: "2.6.18-419.el5debug",
				},
			},
			wantIsKernel: true,
			wantRunning:  true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			gotIsKernel, gotRunning := isRunningKernel(tt.args.pack, tt.args.family, tt.args.release, tt.args.kernel)
			if gotIsKernel != tt.wantIsKernel {
				t.Errorf("isRunningKernel() gotIsKernel = %v, want %v", gotIsKernel, tt.wantIsKernel)
			}
			if gotRunning != tt.wantRunning {
				t.Errorf("isRunningKernel() gotRunning = %v, want %v", gotRunning, tt.wantRunning)
			}
		})
	}
}