package cvss

import "testing"

func TestCategoryString(t *testing.T) {
  tests := []struct {
    cat Category
    exp string
  } {
    { Base, "Base" },
    { Temporal, "Temporal" },
    { Environmental, "Environmental" },
  }

  for _, test := range(tests) {
    t.Run(test.exp, func(t *testing.T) {
      got := test.cat.String()
      if got != test.exp {
        t.Errorf("got: %s, exp: %s", got, test.exp)
      }
    })
  }
}

func TestInvalidCategory(t *testing.T) {
  tests := []struct {
    cat Category
    exp string
  } {
    { Category(byte(255)), "Category(255)" },
  }

  for _, test := range(tests) {
    t.Run(test.exp, func(t *testing.T) {
      got := test.cat.String()
      if got != test.exp {
        t.Errorf("got: %s, exp: %s", got, test.exp)
      }
    })
  }
}