(t *testing.T)
| 150 | } |
| 151 | |
| 152 | func TestEnumerateDataFileIDs(t *testing.T) { |
| 153 | dir := t.TempDir() |
| 154 | |
| 155 | // Create some test files |
| 156 | createFile := func(name string) { |
| 157 | f, err := os.Create(filepath.Join(dir, name)) |
| 158 | if err != nil { |
| 159 | t.Fatal(err) |
| 160 | } |
| 161 | if err := f.Close(); err != nil { |
| 162 | t.Errorf("Failed to close file: %v", err) |
| 163 | } |
| 164 | } |
| 165 | |
| 166 | createFile("0.dat") |
| 167 | createFile("1.dat") |
| 168 | createFile("100.dat") |
| 169 | createFile("merge_0.dat") |
| 170 | createFile("merge_1.dat") |
| 171 | createFile("other.txt") |
| 172 | createFile("0.hint") |
| 173 | |
| 174 | userIDs, mergeIDs, err := enumerateDataFileIDs(dir) |
| 175 | if err != nil { |
| 176 | t.Fatalf("enumerateDataFileIDs failed: %v", err) |
| 177 | } |
| 178 | |
| 179 | if len(userIDs) != 3 { |
| 180 | t.Errorf("len(userIDs) = %d, want 3", len(userIDs)) |
| 181 | } |
| 182 | if userIDs[0] != 0 || userIDs[1] != 1 || userIDs[2] != 100 { |
| 183 | t.Errorf("userIDs = %v, want [0 1 100]", userIDs) |
| 184 | } |
| 185 | |
| 186 | if len(mergeIDs) != 2 { |
| 187 | t.Errorf("len(mergeIDs) = %d, want 2", len(mergeIDs)) |
| 188 | } |
| 189 | expectedMerge0 := GetMergeFileID(0) |
| 190 | expectedMerge1 := GetMergeFileID(1) |
| 191 | if mergeIDs[0] != expectedMerge0 || mergeIDs[1] != expectedMerge1 { |
| 192 | t.Errorf("mergeIDs = %v, want [%d %d]", mergeIDs, expectedMerge0, expectedMerge1) |
| 193 | } |
| 194 | } |
| 195 | |
| 196 | func TestPurgeMergeFiles(t *testing.T) { |
| 197 | dir := t.TempDir() |
nothing calls this directly
no test coverage detected