utils_test.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. package utils
  2. import (
  3. "os"
  4. "testing"
  5. )
  6. func TestCheckLocalDns(t *testing.T) {
  7. for resolv, result := range map[string]bool{`# Dynamic
  8. nameserver 10.0.2.3
  9. search docker.com`: false,
  10. `# Dynamic
  11. #nameserver 127.0.0.1
  12. nameserver 10.0.2.3
  13. search docker.com`: false,
  14. `# Dynamic
  15. nameserver 10.0.2.3 #not used 127.0.1.1
  16. search docker.com`: false,
  17. `# Dynamic
  18. #nameserver 10.0.2.3
  19. #search docker.com`: true,
  20. `# Dynamic
  21. nameserver 127.0.0.1
  22. search docker.com`: true,
  23. `# Dynamic
  24. nameserver 127.0.1.1
  25. search docker.com`: true,
  26. `# Dynamic
  27. `: true,
  28. ``: true,
  29. } {
  30. if CheckLocalDns([]byte(resolv)) != result {
  31. t.Fatalf("Wrong local dns detection: {%s} should be %v", resolv, result)
  32. }
  33. }
  34. }
  35. func TestReplaceAndAppendEnvVars(t *testing.T) {
  36. var (
  37. d = []string{"HOME=/"}
  38. o = []string{"HOME=/root", "TERM=xterm"}
  39. )
  40. env := ReplaceOrAppendEnvValues(d, o)
  41. if len(env) != 2 {
  42. t.Fatalf("expected len of 2 got %d", len(env))
  43. }
  44. if env[0] != "HOME=/root" {
  45. t.Fatalf("expected HOME=/root got '%s'", env[0])
  46. }
  47. if env[1] != "TERM=xterm" {
  48. t.Fatalf("expected TERM=xterm got '%s'", env[1])
  49. }
  50. }
  51. // Reading a symlink to a directory must return the directory
  52. func TestReadSymlinkedDirectoryExistingDirectory(t *testing.T) {
  53. var err error
  54. if err = os.Mkdir("/tmp/testReadSymlinkToExistingDirectory", 0777); err != nil {
  55. t.Errorf("failed to create directory: %s", err)
  56. }
  57. if err = os.Symlink("/tmp/testReadSymlinkToExistingDirectory", "/tmp/dirLinkTest"); err != nil {
  58. t.Errorf("failed to create symlink: %s", err)
  59. }
  60. var path string
  61. if path, err = ReadSymlinkedDirectory("/tmp/dirLinkTest"); err != nil {
  62. t.Fatalf("failed to read symlink to directory: %s", err)
  63. }
  64. if path != "/tmp/testReadSymlinkToExistingDirectory" {
  65. t.Fatalf("symlink returned unexpected directory: %s", path)
  66. }
  67. if err = os.Remove("/tmp/testReadSymlinkToExistingDirectory"); err != nil {
  68. t.Errorf("failed to remove temporary directory: %s", err)
  69. }
  70. if err = os.Remove("/tmp/dirLinkTest"); err != nil {
  71. t.Errorf("failed to remove symlink: %s", err)
  72. }
  73. }
  74. // Reading a non-existing symlink must fail
  75. func TestReadSymlinkedDirectoryNonExistingSymlink(t *testing.T) {
  76. var path string
  77. var err error
  78. if path, err = ReadSymlinkedDirectory("/tmp/test/foo/Non/ExistingPath"); err == nil {
  79. t.Fatalf("error expected for non-existing symlink")
  80. }
  81. if path != "" {
  82. t.Fatalf("expected empty path, but '%s' was returned", path)
  83. }
  84. }
  85. // Reading a symlink to a file must fail
  86. func TestReadSymlinkedDirectoryToFile(t *testing.T) {
  87. var err error
  88. var file *os.File
  89. if file, err = os.Create("/tmp/testReadSymlinkToFile"); err != nil {
  90. t.Fatalf("failed to create file: %s", err)
  91. }
  92. file.Close()
  93. if err = os.Symlink("/tmp/testReadSymlinkToFile", "/tmp/fileLinkTest"); err != nil {
  94. t.Errorf("failed to create symlink: %s", err)
  95. }
  96. var path string
  97. if path, err = ReadSymlinkedDirectory("/tmp/fileLinkTest"); err == nil {
  98. t.Fatalf("ReadSymlinkedDirectory on a symlink to a file should've failed")
  99. }
  100. if path != "" {
  101. t.Fatalf("path should've been empty: %s", path)
  102. }
  103. if err = os.Remove("/tmp/testReadSymlinkToFile"); err != nil {
  104. t.Errorf("failed to remove file: %s", err)
  105. }
  106. if err = os.Remove("/tmp/fileLinkTest"); err != nil {
  107. t.Errorf("failed to remove symlink: %s", err)
  108. }
  109. }