parsers_test.go 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. package parsers
  2. import (
  3. "reflect"
  4. "testing"
  5. )
  6. func TestParseKeyValueOpt(t *testing.T) {
  7. invalids := map[string]string{
  8. "": "Unable to parse key/value option: ",
  9. "key": "Unable to parse key/value option: key",
  10. }
  11. for invalid, expectedError := range invalids {
  12. if _, _, err := ParseKeyValueOpt(invalid); err == nil || err.Error() != expectedError {
  13. t.Fatalf("Expected error %v for %v, got %v", expectedError, invalid, err)
  14. }
  15. }
  16. valids := map[string][]string{
  17. "key=value": {"key", "value"},
  18. " key = value ": {"key", "value"},
  19. "key=value1=value2": {"key", "value1=value2"},
  20. " key = value1 = value2 ": {"key", "value1 = value2"},
  21. }
  22. for valid, expectedKeyValue := range valids {
  23. key, value, err := ParseKeyValueOpt(valid)
  24. if err != nil {
  25. t.Fatal(err)
  26. }
  27. if key != expectedKeyValue[0] || value != expectedKeyValue[1] {
  28. t.Fatalf("Expected {%v: %v} got {%v: %v}", expectedKeyValue[0], expectedKeyValue[1], key, value)
  29. }
  30. }
  31. }
  32. func TestParseUintList(t *testing.T) {
  33. valids := map[string]map[int]bool{
  34. "": {},
  35. "7": {7: true},
  36. "1-6": {1: true, 2: true, 3: true, 4: true, 5: true, 6: true},
  37. "0-7": {0: true, 1: true, 2: true, 3: true, 4: true, 5: true, 6: true, 7: true},
  38. "0,3-4,7,8-10": {0: true, 3: true, 4: true, 7: true, 8: true, 9: true, 10: true},
  39. "0-0,0,1-4": {0: true, 1: true, 2: true, 3: true, 4: true},
  40. "03,1-3": {1: true, 2: true, 3: true},
  41. "3,2,1": {1: true, 2: true, 3: true},
  42. "0-2,3,1": {0: true, 1: true, 2: true, 3: true},
  43. }
  44. for k, v := range valids {
  45. out, err := ParseUintList(k)
  46. if err != nil {
  47. t.Fatalf("Expected not to fail, got %v", err)
  48. }
  49. if !reflect.DeepEqual(out, v) {
  50. t.Fatalf("Expected %v, got %v", v, out)
  51. }
  52. }
  53. invalids := []string{
  54. "this",
  55. "1--",
  56. "1-10,,10",
  57. "10-1",
  58. "-1",
  59. "-1,0",
  60. }
  61. for _, v := range invalids {
  62. if out, err := ParseUintList(v); err == nil {
  63. t.Fatalf("Expected failure with %s but got %v", v, out)
  64. }
  65. }
  66. }