package cvss

import (
  "testing"
)

func TestV30Vector(t *testing.T) {
  // test v3 vector
  test := "CVSS:3.0/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:N/A:N"
  expMetrics := []string { "AV:L", "AC:H", "PR:L", "UI:N", "S:U", "C:H", "I:N", "A:N" }

  t.Run("TestV30Vector", func(t *testing.T) {
    // parse vector
    vec, err := NewVector(test)
    if err != nil {
      t.Error(err)
      return
    }

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

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

func TestInvalidV30Vector(t *testing.T) {
  // test invalid vector
  test := "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"

  // parse vector
  got, err := NewVector(test)
  if err != nil && err.Error() != exp {
    t.Errorf("got \"%s\", exp \"%s\"", err.Error(), err)
    return
  } else if err == nil {
    t.Errorf("got \"%s\", exp badMetric", got)
    return
  }
}