pointer_test.go 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. package flags
  2. import (
  3. "testing"
  4. )
  5. func TestPointerBool(t *testing.T) {
  6. var opts = struct {
  7. Value *bool `short:"v"`
  8. }{}
  9. ret := assertParseSuccess(t, &opts, "-v")
  10. assertStringArray(t, ret, []string{})
  11. if !*opts.Value {
  12. t.Errorf("Expected Value to be true")
  13. }
  14. }
  15. func TestPointerString(t *testing.T) {
  16. var opts = struct {
  17. Value *string `short:"v"`
  18. }{}
  19. ret := assertParseSuccess(t, &opts, "-v", "value")
  20. assertStringArray(t, ret, []string{})
  21. assertString(t, *opts.Value, "value")
  22. }
  23. func TestPointerSlice(t *testing.T) {
  24. var opts = struct {
  25. Value *[]string `short:"v"`
  26. }{}
  27. ret := assertParseSuccess(t, &opts, "-v", "value1", "-v", "value2")
  28. assertStringArray(t, ret, []string{})
  29. assertStringArray(t, *opts.Value, []string{"value1", "value2"})
  30. }
  31. func TestPointerMap(t *testing.T) {
  32. var opts = struct {
  33. Value *map[string]int `short:"v"`
  34. }{}
  35. ret := assertParseSuccess(t, &opts, "-v", "k1:2", "-v", "k2:-5")
  36. assertStringArray(t, ret, []string{})
  37. if v, ok := (*opts.Value)["k1"]; !ok {
  38. t.Errorf("Expected key \"k1\" to exist")
  39. } else if v != 2 {
  40. t.Errorf("Expected \"k1\" to be 2, but got %#v", v)
  41. }
  42. if v, ok := (*opts.Value)["k2"]; !ok {
  43. t.Errorf("Expected key \"k2\" to exist")
  44. } else if v != -5 {
  45. t.Errorf("Expected \"k2\" to be -5, but got %#v", v)
  46. }
  47. }
  48. type PointerGroup struct {
  49. Value bool `short:"v"`
  50. }
  51. func TestPointerGroup(t *testing.T) {
  52. var opts = struct {
  53. Group *PointerGroup `group:"Group Options"`
  54. }{}
  55. ret := assertParseSuccess(t, &opts, "-v")
  56. assertStringArray(t, ret, []string{})
  57. if !opts.Group.Value {
  58. t.Errorf("Expected Group.Value to be true")
  59. }
  60. }