package cvss

import (
  "testing"
)

func TestNewV30Vector(t *testing.T) {
  passTests := []struct {
    val string
    exp []string
  } {
    {
      val: "CVSS:3.0/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:N/A:N",
      exp: []string { "AV:L", "AC:H", "PR:L", "UI:N", "S:U", "C:H", "I:N", "A:N" },
    },
  }

  for _, test := range(passTests) {
    t.Run(test.val, func(t *testing.T) {
      // parse vector
      vec, err := newV30Vector(test.val)
      if err != nil {
        t.Error(err)
        return
      }

      // check version
      if vec.Version() != V30 {
        t.Errorf("got %s, exp %s", vec.Version(), V30)
        return
      }

      // check metric length
      if len(vec.Metrics()) != len(test.exp) {
        t.Errorf("got %d, exp %d", len(vec.Metrics()), len(test.exp))
        return
      }

      // check metrics
      for i, m := range(vec.Metrics()) {
        got := m.String()
        if got != test.exp[i] {
          t.Errorf("got %s, exp %s", got, test.exp[i])
        }
      }
    })
  }

  failTests := []struct {
    val string
    exp string
  } {
    {
      val: "CVSS:3.0/AV:L/junk/PR:L/UI:N/S:U/C:H/I:N/A:N",
      exp: "invalid CVSS 3.0 metric: junk",
    },
  }

  for _, test := range(failTests) {
    t.Run(test.val, func(t *testing.T) {
      got, err := newV30Vector(test.val)
      if err != nil && err.Error() != test.exp {
        t.Errorf("got \"%s\", exp \"%s\"", err.Error(), test.exp)
      } else if err == nil {
        t.Errorf("got \"%s\", exp badMetric", got)
      }
    })
  }
}