diff options
Diffstat (limited to 'repo_test.go')
| -rw-r--r-- | repo_test.go | 55 |
1 files changed, 47 insertions, 8 deletions
diff --git a/repo_test.go b/repo_test.go index 00c94df..ed2cd28 100644 --- a/repo_test.go +++ b/repo_test.go @@ -6,7 +6,6 @@ import ( "io" "io/ioutil" "os" - "path" "path/filepath" "reflect" "testing" @@ -130,7 +129,7 @@ func TestReadFiles3(t *testing.T) { func TestNoSuchFile(t *testing.T) { tmpDir := t.TempDir() - os.Symlink("./notexisting", path.Join(tmpDir, "linknotexisting")) + os.Symlink("./notexisting", filepath.Join(tmpDir, "linknotexisting")) var buff bytes.Buffer files := listFiles(tmpDir) assertLen(t, 1, files, "Files") @@ -301,6 +300,44 @@ func TestRestoreZlib(t *testing.T) { assertSameTree(t, assertSameFile, expected, dest, "Restore") } +func TestHashes(t *testing.T) { + dest := t.TempDir() + source := filepath.Join("testdata", "repo_8k") + + chunks := make(chan IdentifiedChunk, 16) + storeQueue := make(chan chunkData, 16) + storeEnd := make(chan bool) + + repo1 := NewRepo(source) + repo1.chunkReadWrapper = utils.NopReadWrapper + repo1.chunkWriteWrapper = utils.NopWriteWrapper + go repo1.loadChunks([]string{filepath.Join(source, "00000")}, chunks) + for c := range chunks { + fp, sk := repo1.hashChunk(c.GetId(), c.Reader()) + content, err := io.ReadAll(c.Reader()) + if err != nil { + t.Error(err) + } + storeQueue <- chunkData{ + hashes: chunkHashes{fp, sk}, + content: content, + id: c.GetId(), + } + } + repo2 := NewRepo(dest) + repo2.chunkReadWrapper = utils.NopReadWrapper + repo2.chunkWriteWrapper = utils.NopWriteWrapper + os.MkdirAll(filepath.Join(dest, "00000", chunksName), 0775) + go repo2.storageWorker(0, storeQueue, storeEnd) + close(storeQueue) + <-storeEnd + assertLen(t, 0, repo2.fingerprints, "Fingerprints") + assertLen(t, 0, repo2.sketches, "Sketches") + repo2.loadHashes([]string{filepath.Join(dest, "00000")}) + assertSame(t, repo1.fingerprints, repo2.fingerprints, "Fingerprint maps") + assertSame(t, repo1.sketches, repo2.sketches, "Sketches maps") +} + func assertSameTree(t *testing.T, apply func(t *testing.T, expected string, actual string, prefix string), expected string, actual string, prefix string) { actualFiles := listFiles(actual) expectedFiles := listFiles(expected) @@ -341,14 +378,10 @@ func assertCompatibleRepoFile(t *testing.T, expected string, actual string, pref aRecipe := loadRecipe(actual) assertLen(t, len(eRecipe), aRecipe, prefix) for i, eChunk := range eRecipe { - if reflect.DeepEqual(eChunk, aRecipe[i]) { - continue - } else { - t.Fatal(prefix, "chunk do not match:", aRecipe[i], ", expected", eChunk) - } + assertSame(t, eChunk, aRecipe[i], prefix+" chunks") } } else if filepath.Base(expected) == hashesName { - // TODO: check Hashes file + // Hashes file is checked in TestHashes } else { // Chunk content file assertSameFile(t, expected, actual, prefix) @@ -390,3 +423,9 @@ func assertChunkContent(t *testing.T, expected []byte, c Chunk, prefix string) { } assertSameSlice(t, expected, buf, prefix+" Chunk content") } + +func assertSame(t *testing.T, expected interface{}, actual interface{}, prefix string) { + if !reflect.DeepEqual(expected, actual) { + t.Error(prefix, "do not match, expected:", expected, ", actual:", actual) + } +} |