diff options
author | Paul Duncan <pabs@pablotron.org> | 2022-02-02 21:45:23 -0500 |
---|---|---|
committer | Paul Duncan <pabs@pablotron.org> | 2022-02-02 21:45:23 -0500 |
commit | e3a6bc8502a0366aee9040e938789a6016e2fa4a (patch) | |
tree | 1d0098dec135d77dd1c8d4a8b0c54bf2aa1141fa /internal/cpe/part_test.go | |
parent | b6496c2d20904e665116133a9e9bf9ae6e3b45b8 (diff) | |
download | cvez-e3a6bc8502a0366aee9040e938789a6016e2fa4a.tar.bz2 cvez-e3a6bc8502a0366aee9040e938789a6016e2fa4a.zip |
internal/cpe: add token.go, part.go, and tests
Diffstat (limited to 'internal/cpe/part_test.go')
-rw-r--r-- | internal/cpe/part_test.go | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/internal/cpe/part_test.go b/internal/cpe/part_test.go new file mode 100644 index 0000000..269d16e --- /dev/null +++ b/internal/cpe/part_test.go @@ -0,0 +1,108 @@ +package cpe + +import ( + "testing" +) + +func TestNewPart(t *testing.T) { + passTests := []struct { + name string + val token + exp Part + } {{ + name: "any", + val: token { Type: anyToken }, + exp: AnyPart, + }, { + name: "na", + val: token { Type: naToken }, + exp: NAPart, + }, { + name: "a", + val: token { Type: valToken, Val: "a" }, + exp: ApplicationPart, + }, { + name: "h", + val: token { Type: valToken, Val: "h" }, + exp: HardwarePart, + }, { + name: "o", + val: token { Type: valToken, Val: "o" }, + exp: OperatingSystemPart, + }} + + for _, test := range(passTests) { + t.Run(test.name, func(t *testing.T) { + if got, err := newPart(test.val); err != nil { + t.Error(err) + } else if got != test.exp { + t.Errorf("got %s, exp %s", got, test.exp) + } + }) + } + + failTests := []struct { + name string + val token + exp string + } {{ + name: "invalid token", + val: token { Type: valToken, Val: "foo" }, + exp: "unknown part: \"foo\"", + }, { + name: "empty token", + val: token { Type: valToken }, + exp: "unknown part: \"\"", + }, { + name: "unknown token type", + val: token { Type: tokenType(255) }, + exp: "unknown token type: 0xff", + }} + + for _, test := range(failTests) { + t.Run(test.name, func(t *testing.T) { + // tokenize, check for error + got, err := newPart(test.val) + if err == nil { + t.Errorf("got %v, exp error", got) + } else if err.Error() != test.exp { + t.Errorf("got \"%s\", exp \"%s\"", err.Error(), test.exp) + } + }) + } +} + +func TestPartString(t *testing.T) { + tests := []struct { + val Part + exp string + } {{ + val: AnyPart, + exp: "*", + }, { + val: NAPart, + exp: "-", + }, { + val: ApplicationPart, + exp: "a", + }, { + val: HardwarePart, + exp: "h", + }, { + val: OperatingSystemPart, + exp: "o", + }, { + val: Part(255), + exp: "Part(255)", + }} + + for _, test := range(tests) { + t.Run(test.exp, func(t *testing.T) { + got := test.val.String() + + if got != test.exp { + t.Errorf("got \"%s\", exp \"%s\"", got, test.exp) + } + }) + } +} |