|
5 | 5 | package main
|
6 | 6 |
|
7 | 7 | import (
|
| 8 | + "fmt" |
| 9 | + "os" |
| 10 | + "path/filepath" |
8 | 11 | "strings"
|
9 | 12 | "testing"
|
10 | 13 |
|
@@ -105,3 +108,42 @@ func TestUpdatePlatformProperties(t *testing.T) {
|
105 | 108 | }
|
106 | 109 | })
|
107 | 110 | }
|
| 111 | + |
| 112 | +func Test_safeLink(t *testing.T) { |
| 113 | + type args struct { |
| 114 | + changeSource bool |
| 115 | + } |
| 116 | + t.Log("link folders twice") |
| 117 | + tests := []struct { |
| 118 | + name string |
| 119 | + args args |
| 120 | + wantErr bool |
| 121 | + }{ |
| 122 | + {name: "happy path", args: args{changeSource: true}, wantErr: false}, |
| 123 | + {name: "happy path 2", args: args{changeSource: false}, wantErr: false}, |
| 124 | + } |
| 125 | + for index, tt := range tests { |
| 126 | + t.Run(tt.name, func(t *testing.T) { |
| 127 | + source := createTempDir(t, fmt.Sprintf("source_%d", index)) |
| 128 | + target := createTempDir(t, fmt.Sprintf("target_%d", index)) |
| 129 | + safeLink(source, target) |
| 130 | + os.RemoveAll(source) |
| 131 | + if tt.args.changeSource { |
| 132 | + source = createTempDir(t, fmt.Sprintf("source_new_%d", index)) |
| 133 | + } else { |
| 134 | + source = createTempDir(t, fmt.Sprintf("source_%d", index)) |
| 135 | + } |
| 136 | + if err := safeLink(source, target); (err != nil) != tt.wantErr { |
| 137 | + t.Errorf("safeLink() error = %v, wantErr %v", err, tt.wantErr) |
| 138 | + } |
| 139 | + }) |
| 140 | + } |
| 141 | +} |
| 142 | + |
| 143 | +func createTempDir(t *testing.T, dir string) string { |
| 144 | + path := filepath.Join(t.TempDir(), dir) |
| 145 | + if err := os.Mkdir(path, 0o755); err != nil { |
| 146 | + t.Fatal(err) |
| 147 | + } |
| 148 | + return path |
| 149 | +} |
0 commit comments