/*
 * @Author: EnderByEndera
 * @Date: 2020-12-04 15:03:09
 * @LastEditTime: 2021-01-06 09:59:08
 * @LastEditors: Please set LastEditors
 * @Description: Test commrules.go
 * @FilePath: /commdetection/rules/commrules_test.go
 */
package rules

import (
	"commdetection/comm"
	"commdetection/model"
	"fmt"
	"testing"
)

func TestRule(t *testing.T) {
	t.Run("Test AddRule", func(t *testing.T) {
		r := model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
			{
				Name:     "RuleB",
				RuleFunc: "EvaluatePathRule",
			},
		}
		r = AddRule(r, model.Rule{
			Name:     "RuleC",
			RuleFunc: "EvaluateCommandRule",
		})
		predict := model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
			{
				Name:     "RuleB",
				RuleFunc: "EvaluatePathRule",
			},
			{
				Name:     "RuleC",
				RuleFunc: "EvaluateCommandRule",
			},
		}
		if len(r) != len(predict) {
			fmt.Println(r)
			t.Errorf("results are not as predicted")
			return
		}
		for i := range r {
			if r[i].Name != predict[i].Name {
				fmt.Println(r)
				t.Errorf("results are not as predicted")
				return
			}
		}
		cs := model.CommScore{
			Command: model.Command{CommName: "wget"},
			Score:   100,
		}
		for _, rule := range r {
			cs = RuleFuncMap[rule.RuleFunc](cs)
		}
		if cs.Score != 100*0.8*0.8 {
			t.Errorf("results are not as predicted")
		}
	})

	t.Run("Test AddRule with the same name", func(t *testing.T) {
		r := model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
			{
				Name:     "RuleB",
				RuleFunc: "EvaluatePathRule",
			},
		}
		r = AddRule(r, model.Rule{
			Name:     "RuleA",
			RuleFunc: "EvaluateCommandRule",
		})
		if len(r) == 3 {
			t.Errorf("results are not as predicted")
		}
	})

	t.Run("Test DeleteRuleByName", func(t *testing.T) {
		r := model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
			{
				Name:     "RuleB",
				RuleFunc: "EvaluatePathRule",
			},
		}
		r = DeleteRuleByName(r, "RuleB")
		if len(r) != 1 {
			fmt.Println(r)
			t.Errorf("results are not as predicted")
		}
		r = DeleteRuleByName(r, "RuleB")
		if len(r) != 1 {
			fmt.Println(r)
			t.Errorf("results are not as predicted")
		}
	})

	t.Run("Test MarshallEvaluationRules", func(t *testing.T) {
		rs := model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
		}
		err := model.MarshalEvaluationRules(rs)
		if err != nil {
			t.Error(err)
		}
	})
}

func TestInitCommScores(t *testing.T) {
	comms := comm.GetCommands()
	comms = comm.FlushCommands(comms, []comm.Filter{comm.WhichCommandFilter})
	css := InitCommScores(comms)
	fmt.Println(css)
}

func TestEvaluateCommScore(t *testing.T) {
	css := EvaluateCommScore([]model.CommScore{
		{
			Command: model.Command{
				CommName: "wget",
				Args:     []string{""},
			},
			Score: 100,
		},
	},
		model.Rules{
			{
				Name:     "RuleA",
				RuleFunc: "EvaluateCommandRule",
			},
		})
	if css[0].Score != 80.0 {
		t.Errorf("Wrong Score")
	}
}