-
Notifications
You must be signed in to change notification settings - Fork 203
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
smoke: add smoking test for chunk dedup
Add smoking test case for chunk dedup. Signed-off-by: Yadong Ding <[email protected]>
- Loading branch information
1 parent
a938849
commit a2918de
Showing
7 changed files
with
274 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,95 @@ | ||
| // Copyright 2024 Nydus Developers. All rights reserved. | ||
| // | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| package tests | ||
|
|
||
| import ( | ||
| "database/sql" | ||
| "fmt" | ||
| "path/filepath" | ||
| "testing" | ||
|
|
||
| _ "github.com/mattn/go-sqlite3" | ||
|
|
||
| "github.com/containerd/nydus-snapshotter/pkg/converter" | ||
| "github.com/dragonflyoss/nydus/smoke/tests/texture" | ||
| "github.com/dragonflyoss/nydus/smoke/tests/tool" | ||
| "github.com/dragonflyoss/nydus/smoke/tests/tool/test" | ||
| "github.com/opencontainers/go-digest" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| type CasTestSuite struct{} | ||
|
|
||
| func (c *CasTestSuite) TestCas() test.Generator { | ||
| scenarios := tool.DescartesIterator{} | ||
| scenarios.Dimension(paramEnablePrefetch, []interface{}{false, true}) | ||
|
|
||
| return func() (name string, testCase test.Case) { | ||
| if !scenarios.HasNext() { | ||
| return | ||
| } | ||
| scenario := scenarios.Next() | ||
|
|
||
| return scenario.Str(), func(t *testing.T) { | ||
| c.testCasTables(t, scenario.GetBool(paramEnablePrefetch)) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func (c *CasTestSuite) testCasTables(t *testing.T, enablePrefetch bool) { | ||
| ctx := tool.DefaultContext(t) | ||
|
|
||
| // Prepare work directory | ||
| ctx.PrepareWorkDir(t) | ||
| defer ctx.Destroy(t) | ||
|
|
||
| lowerLayer := texture.MakeLowerLayer(t, filepath.Join(ctx.Env.WorkDir, "source")) | ||
| lowerOCIBlobDigest, lowerRafsBlobDigest := lowerLayer.PackRef(t, *ctx, ctx.Env.BlobDir, ctx.Build.OCIRefGzip) | ||
| mergeOption := converter.MergeOption{ | ||
| BuilderPath: ctx.Binary.Builder, | ||
| ChunkDictPath: "", | ||
| OCIRef: true, | ||
| } | ||
| actualDigests, lowerBootstrap := tool.MergeLayers(t, *ctx, mergeOption, []converter.Layer{ | ||
| { | ||
| Digest: lowerRafsBlobDigest, | ||
| OriginalDigest: &lowerOCIBlobDigest, | ||
| }, | ||
| }) | ||
| require.Equal(t, []digest.Digest{lowerOCIBlobDigest}, actualDigests) | ||
|
|
||
| ctx.Env.BootstrapPath = lowerBootstrap | ||
| ctx.Runtime.EnablePrefetch = enablePrefetch | ||
| ctx.Runtime.ChunkDedupDb = filepath.Join(ctx.Env.WorkDir, "cas.db") | ||
|
|
||
| nydusd, err := tool.NewNydusdWithContext(*ctx) | ||
| require.NoError(t, err) | ||
| err = nydusd.Mount() | ||
| require.NoError(t, err) | ||
| defer nydusd.Umount() | ||
| nydusd.Verify(t, lowerLayer.FileTree) | ||
|
|
||
| db, err := sql.Open("sqlite3", ctx.Runtime.ChunkDedupDb) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| defer db.Close() | ||
|
|
||
| for _, expectedTable := range []string{"Blobs", "Chunks"} { | ||
| var count int | ||
| query := fmt.Sprintf("SELECT COUNT(*) FROM %s;", expectedTable) | ||
| err := db.QueryRow(query).Scan(&count) | ||
| require.NoError(t, err) | ||
| if expectedTable == "Blobs" { | ||
| require.Equal(t, count, 1) | ||
| } else { | ||
| require.Equal(t, count, 8) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func TestCas(t *testing.T) { | ||
| test.Run(t, &CasTestSuite{}) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,150 @@ | ||
| // Copyright 2024 Nydus Developers. All rights reserved. | ||
| // | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| package tests | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "io" | ||
| "net" | ||
| "net/http" | ||
| "os" | ||
| "path/filepath" | ||
| "testing" | ||
| "time" | ||
|
|
||
| "github.com/containerd/nydus-snapshotter/pkg/converter" | ||
| "github.com/opencontainers/go-digest" | ||
| "github.com/stretchr/testify/require" | ||
|
|
||
| "github.com/dragonflyoss/nydus/smoke/tests/texture" | ||
| "github.com/dragonflyoss/nydus/smoke/tests/tool" | ||
| "github.com/dragonflyoss/nydus/smoke/tests/tool/test" | ||
| ) | ||
|
|
||
| const ( | ||
| paramIteration = "iteration" | ||
| ) | ||
|
|
||
| type ChunkDedupTestSuite struct{} | ||
|
|
||
| type BackendMetrics struct { | ||
| ReadCount uint64 `json:"read_count"` | ||
| ReadAmountTotal uint64 `json:"read_amount_total"` | ||
| ReadErrors uint64 `json:"read_errors"` | ||
| } | ||
|
|
||
| func (c *ChunkDedupTestSuite) TestChunkDedup() test.Generator { | ||
| scenarios := tool.DescartesIterator{} | ||
| scenarios.Dimension(paramIteration, []interface{}{1}) | ||
|
|
||
| file, _ := os.CreateTemp("", "cas-*.db") | ||
| defer os.Remove(file.Name()) | ||
|
|
||
| return func() (name string, testCase test.Case) { | ||
| if !scenarios.HasNext() { | ||
| return | ||
| } | ||
| scenario := scenarios.Next() | ||
|
|
||
| return scenario.Str(), func(t *testing.T) { | ||
| c.testRemoteWithDedup(t, file.Name()) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func (c *ChunkDedupTestSuite) testRemoteWithDedup(t *testing.T, dbPath string) { | ||
| ctx, layer := c.prepareContext(t, dbPath) | ||
| defer ctx.Destroy(t) | ||
|
|
||
| nydusd, err := tool.NewNydusdWithContext(*ctx) | ||
| require.NoError(t, err) | ||
| err = nydusd.Mount() | ||
| require.NoError(t, err) | ||
| defer nydusd.Umount() | ||
| nydusd.Verify(t, layer.FileTree) | ||
| metrics := c.getBackendReadMount(t, filepath.Join(ctx.Env.WorkDir, "nydusd-api.sock")) | ||
| require.Zero(t, metrics.ReadErrors) | ||
|
|
||
| ctx2, layer2 := c.prepareContext(t, dbPath) | ||
| defer ctx2.Destroy(t) | ||
|
|
||
| nydusd2, err := tool.NewNydusdWithContext(*ctx2) | ||
| require.NoError(t, err) | ||
| err = nydusd2.Mount() | ||
| require.NoError(t, err) | ||
| defer nydusd2.Umount() | ||
| nydusd2.Verify(t, layer2.FileTree) | ||
| metrics2 := c.getBackendReadMount(t, filepath.Join(ctx2.Env.WorkDir, "nydusd-api.sock")) | ||
| require.Zero(t, metrics2.ReadErrors) | ||
|
|
||
| require.Greater(t, metrics.ReadCount, metrics2.ReadCount) | ||
| require.Greater(t, metrics.ReadAmountTotal, metrics2.ReadAmountTotal) | ||
| } | ||
|
|
||
| func (c *ChunkDedupTestSuite) prepareContext(t *testing.T, dbPath string) (*tool.Context, *tool.Layer) { | ||
| ctx := tool.DefaultContext(t) | ||
|
|
||
| // Prepare work directory | ||
| ctx.PrepareWorkDir(t) | ||
|
|
||
| lowerLayer := texture.MakeLowerLayer(t, filepath.Join(ctx.Env.WorkDir, "source")) | ||
| lowerOCIBlobDigest, lowerRafsBlobDigest := lowerLayer.PackRef(t, *ctx, ctx.Env.BlobDir, ctx.Build.OCIRefGzip) | ||
| mergeOption := converter.MergeOption{ | ||
| BuilderPath: ctx.Binary.Builder, | ||
| ChunkDictPath: "", | ||
| OCIRef: true, | ||
| } | ||
| actualDigests, lowerBootstrap := tool.MergeLayers(t, *ctx, mergeOption, []converter.Layer{ | ||
| { | ||
| Digest: lowerRafsBlobDigest, | ||
| OriginalDigest: &lowerOCIBlobDigest, | ||
| }, | ||
| }) | ||
| require.Equal(t, []digest.Digest{lowerOCIBlobDigest}, actualDigests) | ||
|
|
||
| ctx.Env.BootstrapPath = lowerBootstrap | ||
| ctx.Runtime.EnablePrefetch = false | ||
| ctx.Runtime.ChunkDedupDb = dbPath | ||
| return ctx, lowerLayer | ||
| } | ||
|
|
||
| func (c *ChunkDedupTestSuite) getBackendReadMount(t *testing.T, sockPath string) *BackendMetrics { | ||
| transport := &http.Transport{ | ||
| MaxIdleConns: 10, | ||
| IdleConnTimeout: 10 * time.Second, | ||
| ExpectContinueTimeout: 1 * time.Second, | ||
| DialContext: func(ctx context.Context, _, _ string) (net.Conn, error) { | ||
| dialer := &net.Dialer{ | ||
| Timeout: 5 * time.Second, | ||
| KeepAlive: 5 * time.Second, | ||
| } | ||
| return dialer.DialContext(ctx, "unix", sockPath) | ||
| }, | ||
| } | ||
|
|
||
| client := &http.Client{ | ||
| Timeout: 30 * time.Second, | ||
| Transport: transport, | ||
| } | ||
|
|
||
| resp, err := client.Get("http://unix/api/v1/metrics/backend") | ||
| require.NoError(t, err) | ||
| defer resp.Body.Close() | ||
|
|
||
| body, err := io.ReadAll(resp.Body) | ||
| require.NoError(t, err) | ||
|
|
||
| var metrics BackendMetrics | ||
| if err = json.Unmarshal(body, &metrics); err != nil { | ||
| require.NoError(t, err) | ||
| } | ||
|
|
||
| return &metrics | ||
| } | ||
|
|
||
| func TestChunkDedup(t *testing.T) { | ||
| test.Run(t, &ChunkDedupTestSuite{}) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters