package cvss

import (
  "testing"
)

// Test cvss v2
func TestCvssV2Parser(t *testing.T) {
  // test v2 vector
  test := "AV:L/AC:L/Au:N/C:N/I:N/A:P"
  expMetrics := []string { "AV:L", "AC:L", "Au:N", "C:N", "I:N", "A:P" }

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

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

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

// Test cvss v2
func TestInvalidV2Vector(t *testing.T) {
  // test invalid vector
  test := "AV:L/junk/Au:N/C:N/I:N/A:P"
  exp := "invalid CVSS 2.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
  }
}