mirror of
https://github.com/logos-storage/logos-storage-go.git
synced 2026-01-02 13:23:11 +00:00
refactor the archive downloader tests to use testify in the test suite mode
This commit is contained in:
Marcin Czenko
parent
59c4aecaf0
commit
e703d06366
@ -7,17 +7,30 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"go-codex-client/communities"
|
||||
"go-codex-client/protobuf"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"go-codex-client/communities"
|
||||
mock_communities "go-codex-client/communities/mock"
|
||||
"go-codex-client/protobuf"
|
||||
|
||||
"go.uber.org/mock/gomock"
|
||||
)
|
||||
|
||||
// Helper function to create a test index with a single archive
|
||||
func createTestIndex() *protobuf.CodexWakuMessageArchiveIndex {
|
||||
return &protobuf.CodexWakuMessageArchiveIndex{
|
||||
// CodexArchiveDownloaderTestifySuite demonstrates testify's suite functionality
|
||||
type CodexArchiveDownloaderTestifySuite struct {
|
||||
suite.Suite
|
||||
ctrl *gomock.Controller
|
||||
mockClient *mock_communities.MockCodexClientInterface
|
||||
index *protobuf.CodexWakuMessageArchiveIndex
|
||||
}
|
||||
|
||||
// SetupTest runs before each test method
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) SetupTest() {
|
||||
suite.ctrl = gomock.NewController(suite.T())
|
||||
suite.mockClient = mock_communities.NewMockCodexClientInterface(suite.ctrl)
|
||||
suite.index = &protobuf.CodexWakuMessageArchiveIndex{
|
||||
Archives: map[string]*protobuf.CodexWakuMessageArchiveIndexMetadata{
|
||||
"test-archive-hash-1": {
|
||||
Cid: "test-cid-1",
|
||||
@ -30,35 +43,32 @@ func createTestIndex() *protobuf.CodexWakuMessageArchiveIndex {
|
||||
}
|
||||
}
|
||||
|
||||
func TestCodexArchiveDownloader_BasicSingleArchive(t *testing.T) {
|
||||
// Create gomock controller
|
||||
ctrl := gomock.NewController(t)
|
||||
defer ctrl.Finish()
|
||||
// TearDownTest runs after each test method
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) TearDownTest() {
|
||||
suite.ctrl.Finish()
|
||||
}
|
||||
|
||||
// Create test data
|
||||
index := createTestIndex()
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) TestBasicSingleArchive() {
|
||||
// Test data
|
||||
communityID := "test-community"
|
||||
existingArchiveIDs := []string{} // No existing archives
|
||||
cancelChan := make(chan struct{})
|
||||
|
||||
// Create mock client using gomock
|
||||
mockClient := mock_communities.NewMockCodexClientInterface(ctrl)
|
||||
|
||||
// Set up expectations
|
||||
mockClient.EXPECT().
|
||||
// Set up mock expectations - same as before
|
||||
suite.mockClient.EXPECT().
|
||||
TriggerDownloadWithContext(gomock.Any(), "test-cid-1").
|
||||
Return(&communities.CodexManifest{CID: "test-cid-1"}, nil).
|
||||
Times(1)
|
||||
|
||||
// First HasCid call returns false, second returns true (simulating polling)
|
||||
gomock.InOrder(
|
||||
mockClient.EXPECT().HasCid("test-cid-1").Return(false, nil),
|
||||
mockClient.EXPECT().HasCid("test-cid-1").Return(true, nil),
|
||||
suite.mockClient.EXPECT().HasCid("test-cid-1").Return(false, nil),
|
||||
suite.mockClient.EXPECT().HasCid("test-cid-1").Return(true, nil),
|
||||
)
|
||||
|
||||
// Create downloader with mock client
|
||||
downloader := communities.NewCodexArchiveDownloader(mockClient, index, communityID, existingArchiveIDs, cancelChan)
|
||||
|
||||
downloader := communities.NewCodexArchiveDownloader(suite.mockClient, suite.index, communityID, existingArchiveIDs, cancelChan)
|
||||
|
||||
// Set fast polling interval for tests (10ms instead of default 1s)
|
||||
downloader.SetPollingInterval(10 * time.Millisecond)
|
||||
|
||||
@ -74,65 +84,112 @@ func TestCodexArchiveDownloader_BasicSingleArchive(t *testing.T) {
|
||||
callbackTo = to
|
||||
})
|
||||
|
||||
// Verify initial state
|
||||
if downloader.GetTotalArchivesCount() != 1 {
|
||||
t.Errorf("Expected 1 total archive, got %d", downloader.GetTotalArchivesCount())
|
||||
}
|
||||
|
||||
if downloader.GetTotalDownloadedArchivesCount() != 0 {
|
||||
t.Errorf("Expected 0 downloaded archives initially, got %d", downloader.GetTotalDownloadedArchivesCount())
|
||||
}
|
||||
|
||||
if downloader.IsDownloadComplete() {
|
||||
t.Error("Expected download to not be complete initially")
|
||||
}
|
||||
// Verify initial state - compare testify vs standard assertions
|
||||
// Testify version is more readable and provides better error messages
|
||||
assert.Equal(suite.T(), 1, downloader.GetTotalArchivesCount(), "Total archives count should be 1")
|
||||
assert.Equal(suite.T(), 0, downloader.GetTotalDownloadedArchivesCount(), "Downloaded archives should be 0 initially")
|
||||
assert.False(suite.T(), downloader.IsDownloadComplete(), "Download should not be complete initially")
|
||||
|
||||
// Start the download
|
||||
downloader.StartDownload()
|
||||
|
||||
// Wait for download to complete (with timeout)
|
||||
timeout := time.After(5 * time.Second)
|
||||
ticker := time.NewTicker(100 * time.Millisecond)
|
||||
defer ticker.Stop()
|
||||
require.Eventually(suite.T(), func() bool {
|
||||
return downloader.IsDownloadComplete()
|
||||
}, 5*time.Second, 100*time.Millisecond, "Download should complete within 5 seconds")
|
||||
|
||||
waitLoop:
|
||||
for {
|
||||
select {
|
||||
case <-timeout:
|
||||
t.Fatal("Timeout waiting for download to complete")
|
||||
case <-ticker.C:
|
||||
if downloader.IsDownloadComplete() {
|
||||
break waitLoop
|
||||
}
|
||||
}
|
||||
}
|
||||
// Verify final state
|
||||
if !downloader.IsDownloadComplete() {
|
||||
t.Error("Expected download to be complete")
|
||||
}
|
||||
// Verify final state - testify makes these assertions more expressive
|
||||
assert.True(suite.T(), downloader.IsDownloadComplete(), "Download should be complete")
|
||||
assert.Equal(suite.T(), 1, downloader.GetTotalDownloadedArchivesCount(), "Should have 1 downloaded archive")
|
||||
|
||||
if downloader.GetTotalDownloadedArchivesCount() != 1 {
|
||||
t.Errorf("Expected 1 downloaded archive, got %d", downloader.GetTotalDownloadedArchivesCount())
|
||||
}
|
||||
// Verify callback was invoked - multiple related assertions grouped logically
|
||||
assert.True(suite.T(), callbackInvoked, "Callback should be invoked")
|
||||
assert.Equal(suite.T(), "test-archive-hash-1", callbackHash, "Callback hash should match expected")
|
||||
assert.Equal(suite.T(), uint64(1000), callbackFrom, "Callback from should be 1000")
|
||||
assert.Equal(suite.T(), uint64(2000), callbackTo, "Callback to should be 2000")
|
||||
|
||||
// Verify callback was invoked
|
||||
if !callbackInvoked {
|
||||
t.Error("Expected callback to be invoked")
|
||||
}
|
||||
|
||||
if callbackHash != "test-archive-hash-1" {
|
||||
t.Errorf("Expected callback hash 'test-archive-hash-1', got '%s'", callbackHash)
|
||||
}
|
||||
|
||||
if callbackFrom != 1000 {
|
||||
t.Errorf("Expected callback from 1000, got %d", callbackFrom)
|
||||
}
|
||||
|
||||
if callbackTo != 2000 {
|
||||
t.Errorf("Expected callback to 2000, got %d", callbackTo)
|
||||
}
|
||||
|
||||
t.Logf("✅ Basic single archive download test passed")
|
||||
t.Logf(" - All mock expectations satisfied")
|
||||
t.Logf(" - Callback invoked: %v", callbackInvoked)
|
||||
suite.T().Log("✅ Basic single archive download test passed")
|
||||
suite.T().Log(" - All mock expectations satisfied")
|
||||
suite.T().Logf(" - Callback invoked: %v", callbackInvoked)
|
||||
}
|
||||
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) TestMultipleArchives() {
|
||||
// Create test data with multiple archives
|
||||
index := &protobuf.CodexWakuMessageArchiveIndex{
|
||||
Archives: map[string]*protobuf.CodexWakuMessageArchiveIndexMetadata{
|
||||
"archive-1": {
|
||||
Cid: "cid-1",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 1000, To: 2000},
|
||||
},
|
||||
"archive-2": {
|
||||
Cid: "cid-2",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 2000, To: 3000},
|
||||
},
|
||||
"archive-3": {
|
||||
Cid: "cid-3",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 3000, To: 4000},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
communityID := "test-community"
|
||||
existingArchiveIDs := []string{} // No existing archives
|
||||
cancelChan := make(chan struct{})
|
||||
|
||||
// Set up expectations for all 3 archives - testify makes verification cleaner
|
||||
expectedCids := []string{"cid-1", "cid-2", "cid-3"}
|
||||
|
||||
for _, cid := range expectedCids {
|
||||
suite.mockClient.EXPECT().
|
||||
TriggerDownloadWithContext(gomock.Any(), cid).
|
||||
Return(&communities.CodexManifest{CID: cid}, nil).
|
||||
Times(1)
|
||||
|
||||
// Each archive becomes available after one poll
|
||||
gomock.InOrder(
|
||||
suite.mockClient.EXPECT().HasCid(cid).Return(false, nil),
|
||||
suite.mockClient.EXPECT().HasCid(cid).Return(true, nil),
|
||||
)
|
||||
}
|
||||
|
||||
// Create downloader
|
||||
downloader := communities.NewCodexArchiveDownloader(suite.mockClient, index, communityID, existingArchiveIDs, cancelChan)
|
||||
downloader.SetPollingInterval(10 * time.Millisecond)
|
||||
|
||||
// Track completed archives
|
||||
completedArchives := make(map[string]bool)
|
||||
downloader.SetOnArchiveDownloaded(func(hash string, from, to uint64) {
|
||||
completedArchives[hash] = true
|
||||
})
|
||||
|
||||
// Initial state verification
|
||||
assert.Equal(suite.T(), 3, downloader.GetTotalArchivesCount(), "Should have 3 total archives")
|
||||
assert.Equal(suite.T(), 0, downloader.GetTotalDownloadedArchivesCount(), "Should start with 0 downloaded")
|
||||
assert.False(suite.T(), downloader.IsDownloadComplete(), "Should not be complete initially")
|
||||
|
||||
// Start download
|
||||
downloader.StartDownload()
|
||||
|
||||
// Wait for all downloads to complete
|
||||
require.Eventually(suite.T(), func() bool {
|
||||
return downloader.IsDownloadComplete()
|
||||
}, 10*time.Second, 100*time.Millisecond, "All downloads should complete within 10 seconds")
|
||||
|
||||
// Final state verification - testify makes these checks very readable
|
||||
assert.True(suite.T(), downloader.IsDownloadComplete(), "Download should be complete")
|
||||
assert.Equal(suite.T(), 3, downloader.GetTotalDownloadedArchivesCount(), "Should have downloaded all 3 archives")
|
||||
|
||||
// Verify all archives were processed
|
||||
assert.Len(suite.T(), completedArchives, 3, "Should have completed exactly 3 archives")
|
||||
assert.Contains(suite.T(), completedArchives, "archive-1", "Should have completed archive-1")
|
||||
assert.Contains(suite.T(), completedArchives, "archive-2", "Should have completed archive-2")
|
||||
assert.Contains(suite.T(), completedArchives, "archive-3", "Should have completed archive-3")
|
||||
|
||||
suite.T().Log("✅ Multiple archives test passed")
|
||||
suite.T().Logf(" - Completed %d out of %d archives", len(completedArchives), 3)
|
||||
}
|
||||
|
||||
// Run the test suite
|
||||
func TestCodexArchiveDownloaderSuite(t *testing.T) {
|
||||
suite.Run(t, new(CodexArchiveDownloaderTestifySuite))
|
||||
}
|
||||
|
||||
@ -1,296 +0,0 @@
|
||||
//go:build !disable_torrent
|
||||
// +build !disable_torrent
|
||||
|
||||
package communities_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/stretchr/testify/suite"
|
||||
|
||||
"go-codex-client/communities"
|
||||
"go-codex-client/protobuf"
|
||||
mock_communities "go-codex-client/communities/mock"
|
||||
|
||||
"go.uber.org/mock/gomock"
|
||||
)
|
||||
|
||||
// CodexArchiveDownloaderTestifySuite demonstrates testify's suite functionality
|
||||
type CodexArchiveDownloaderTestifySuite struct {
|
||||
suite.Suite
|
||||
ctrl *gomock.Controller
|
||||
mockClient *mock_communities.MockCodexClientInterface
|
||||
index *protobuf.CodexWakuMessageArchiveIndex
|
||||
}
|
||||
|
||||
// SetupTest runs before each test method
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) SetupTest() {
|
||||
suite.ctrl = gomock.NewController(suite.T())
|
||||
suite.mockClient = mock_communities.NewMockCodexClientInterface(suite.ctrl)
|
||||
suite.index = &protobuf.CodexWakuMessageArchiveIndex{
|
||||
Archives: map[string]*protobuf.CodexWakuMessageArchiveIndexMetadata{
|
||||
"test-archive-hash-1": {
|
||||
Cid: "test-cid-1",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{
|
||||
From: 1000,
|
||||
To: 2000,
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// TearDownTest runs after each test method
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) TearDownTest() {
|
||||
suite.ctrl.Finish()
|
||||
}
|
||||
|
||||
// TestBasicSingleArchive demonstrates testify assertions vs standard library
|
||||
func (suite *CodexArchiveDownloaderTestifySuite) TestBasicSingleArchive() {
|
||||
// Test data
|
||||
communityID := "test-community"
|
||||
existingArchiveIDs := []string{} // No existing archives
|
||||
cancelChan := make(chan struct{})
|
||||
|
||||
// Set up mock expectations - same as before
|
||||
suite.mockClient.EXPECT().
|
||||
TriggerDownloadWithContext(gomock.Any(), "test-cid-1").
|
||||
Return(&communities.CodexManifest{CID: "test-cid-1"}, nil).
|
||||
Times(1)
|
||||
|
||||
// First HasCid call returns false, second returns true (simulating polling)
|
||||
gomock.InOrder(
|
||||
suite.mockClient.EXPECT().HasCid("test-cid-1").Return(false, nil),
|
||||
suite.mockClient.EXPECT().HasCid("test-cid-1").Return(true, nil),
|
||||
)
|
||||
|
||||
// Create downloader with mock client
|
||||
downloader := communities.NewCodexArchiveDownloader(suite.mockClient, suite.index, communityID, existingArchiveIDs, cancelChan)
|
||||
|
||||
// Set fast polling interval for tests (10ms instead of default 1s)
|
||||
downloader.SetPollingInterval(10 * time.Millisecond)
|
||||
|
||||
// Set up callback to track completion
|
||||
var callbackInvoked bool
|
||||
var callbackHash string
|
||||
var callbackFrom, callbackTo uint64
|
||||
|
||||
downloader.SetOnArchiveDownloaded(func(hash string, from, to uint64) {
|
||||
callbackInvoked = true
|
||||
callbackHash = hash
|
||||
callbackFrom = from
|
||||
callbackTo = to
|
||||
})
|
||||
|
||||
// Verify initial state - compare testify vs standard assertions
|
||||
// Testify version is more readable and provides better error messages
|
||||
assert.Equal(suite.T(), 1, downloader.GetTotalArchivesCount(), "Total archives count should be 1")
|
||||
assert.Equal(suite.T(), 0, downloader.GetTotalDownloadedArchivesCount(), "Downloaded archives should be 0 initially")
|
||||
assert.False(suite.T(), downloader.IsDownloadComplete(), "Download should not be complete initially")
|
||||
|
||||
// Start the download
|
||||
downloader.StartDownload()
|
||||
|
||||
// Wait for download to complete (with timeout)
|
||||
require.Eventually(suite.T(), func() bool {
|
||||
return downloader.IsDownloadComplete()
|
||||
}, 5*time.Second, 100*time.Millisecond, "Download should complete within 5 seconds")
|
||||
|
||||
// Verify final state - testify makes these assertions more expressive
|
||||
assert.True(suite.T(), downloader.IsDownloadComplete(), "Download should be complete")
|
||||
assert.Equal(suite.T(), 1, downloader.GetTotalDownloadedArchivesCount(), "Should have 1 downloaded archive")
|
||||
|
||||
// Verify callback was invoked - multiple related assertions grouped logically
|
||||
assert.True(suite.T(), callbackInvoked, "Callback should be invoked")
|
||||
assert.Equal(suite.T(), "test-archive-hash-1", callbackHash, "Callback hash should match expected")
|
||||
assert.Equal(suite.T(), uint64(1000), callbackFrom, "Callback from should be 1000")
|
||||
assert.Equal(suite.T(), uint64(2000), callbackTo, "Callback to should be 2000")
|
||||
|
||||
suite.T().Log("✅ Basic single archive download test passed (testify version)")
|
||||
suite.T().Log(" - All mock expectations satisfied")
|
||||
suite.T().Logf(" - Callback invoked: %v", callbackInvoked)
|
||||
}
|
||||
|
||||
// TestBasicSingleArchiveFunction demonstrates testify without test suite (function-based)
|
||||
func TestCodexArchiveDownloader_BasicSingleArchive_Testify(t *testing.T) {
|
||||
// Create gomock controller
|
||||
ctrl := gomock.NewController(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
// Create test data using testify helper
|
||||
index := createTestIndexTestify()
|
||||
communityID := "test-community"
|
||||
existingArchiveIDs := []string{} // No existing archives
|
||||
cancelChan := make(chan struct{})
|
||||
|
||||
// Create mock client
|
||||
mockClient := mock_communities.NewMockCodexClientInterface(ctrl)
|
||||
|
||||
// Set up expectations
|
||||
mockClient.EXPECT().
|
||||
TriggerDownloadWithContext(gomock.Any(), "test-cid-1").
|
||||
Return(&communities.CodexManifest{CID: "test-cid-1"}, nil).
|
||||
Times(1)
|
||||
|
||||
// First HasCid call returns false, second returns true (simulating polling)
|
||||
gomock.InOrder(
|
||||
mockClient.EXPECT().HasCid("test-cid-1").Return(false, nil),
|
||||
mockClient.EXPECT().HasCid("test-cid-1").Return(true, nil),
|
||||
)
|
||||
|
||||
// Create downloader with mock client
|
||||
downloader := communities.NewCodexArchiveDownloader(mockClient, index, communityID, existingArchiveIDs, cancelChan)
|
||||
|
||||
// Set fast polling interval for tests (10ms instead of default 1s)
|
||||
downloader.SetPollingInterval(10 * time.Millisecond)
|
||||
|
||||
// Set up callback to track completion
|
||||
var callbackData struct {
|
||||
invoked bool
|
||||
hash string
|
||||
from uint64
|
||||
to uint64
|
||||
}
|
||||
|
||||
downloader.SetOnArchiveDownloaded(func(hash string, from, to uint64) {
|
||||
callbackData.invoked = true
|
||||
callbackData.hash = hash
|
||||
callbackData.from = from
|
||||
callbackData.to = to
|
||||
})
|
||||
|
||||
// Verify initial state using testify assertions
|
||||
// Notice how readable and self-documenting these are compared to if/t.Error patterns
|
||||
require.Equal(t, 1, downloader.GetTotalArchivesCount())
|
||||
require.Equal(t, 0, downloader.GetTotalDownloadedArchivesCount())
|
||||
require.False(t, downloader.IsDownloadComplete())
|
||||
|
||||
// Start the download
|
||||
downloader.StartDownload()
|
||||
|
||||
// Wait for download to complete - testify's Eventually is much cleaner than manual loops
|
||||
require.Eventually(t, func() bool {
|
||||
return downloader.IsDownloadComplete()
|
||||
}, 5*time.Second, 100*time.Millisecond)
|
||||
|
||||
// Verify final state
|
||||
assert.True(t, downloader.IsDownloadComplete())
|
||||
assert.Equal(t, 1, downloader.GetTotalDownloadedArchivesCount())
|
||||
|
||||
// Verify callback - testify allows for compound assertions
|
||||
require.True(t, callbackData.invoked, "Callback should be invoked")
|
||||
|
||||
// Group related assertions for better test organization
|
||||
assert.Equal(t, "test-archive-hash-1", callbackData.hash)
|
||||
assert.Equal(t, uint64(1000), callbackData.from)
|
||||
assert.Equal(t, uint64(2000), callbackData.to)
|
||||
|
||||
t.Log("✅ Function-based testify test passed")
|
||||
}
|
||||
|
||||
// TestMultipleArchivesWithTestify demonstrates testify's advantages for complex scenarios
|
||||
func TestCodexArchiveDownloader_MultipleArchives_Testify(t *testing.T) {
|
||||
// Create gomock controller
|
||||
ctrl := gomock.NewController(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
// Create test data with multiple archives
|
||||
index := &protobuf.CodexWakuMessageArchiveIndex{
|
||||
Archives: map[string]*protobuf.CodexWakuMessageArchiveIndexMetadata{
|
||||
"archive-1": {
|
||||
Cid: "cid-1",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 1000, To: 2000},
|
||||
},
|
||||
"archive-2": {
|
||||
Cid: "cid-2",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 2000, To: 3000},
|
||||
},
|
||||
"archive-3": {
|
||||
Cid: "cid-3",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{From: 3000, To: 4000},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
communityID := "test-community"
|
||||
existingArchiveIDs := []string{} // No existing archives
|
||||
cancelChan := make(chan struct{})
|
||||
|
||||
// Create mock client
|
||||
mockClient := mock_communities.NewMockCodexClientInterface(ctrl)
|
||||
|
||||
// Set up expectations for all 3 archives - testify makes verification cleaner
|
||||
expectedCids := []string{"cid-1", "cid-2", "cid-3"}
|
||||
|
||||
for _, cid := range expectedCids {
|
||||
mockClient.EXPECT().
|
||||
TriggerDownloadWithContext(gomock.Any(), cid).
|
||||
Return(&communities.CodexManifest{CID: cid}, nil).
|
||||
Times(1)
|
||||
|
||||
// Each archive becomes available after one poll
|
||||
gomock.InOrder(
|
||||
mockClient.EXPECT().HasCid(cid).Return(false, nil),
|
||||
mockClient.EXPECT().HasCid(cid).Return(true, nil),
|
||||
)
|
||||
}
|
||||
|
||||
// Create downloader
|
||||
downloader := communities.NewCodexArchiveDownloader(mockClient, index, communityID, existingArchiveIDs, cancelChan)
|
||||
downloader.SetPollingInterval(10 * time.Millisecond)
|
||||
|
||||
// Track completed archives
|
||||
completedArchives := make(map[string]bool)
|
||||
downloader.SetOnArchiveDownloaded(func(hash string, from, to uint64) {
|
||||
completedArchives[hash] = true
|
||||
})
|
||||
|
||||
// Initial state verification
|
||||
assert.Equal(t, 3, downloader.GetTotalArchivesCount(), "Should have 3 total archives")
|
||||
assert.Equal(t, 0, downloader.GetTotalDownloadedArchivesCount(), "Should start with 0 downloaded")
|
||||
assert.False(t, downloader.IsDownloadComplete(), "Should not be complete initially")
|
||||
|
||||
// Start download
|
||||
downloader.StartDownload()
|
||||
|
||||
// Wait for all downloads to complete
|
||||
require.Eventually(t, func() bool {
|
||||
return downloader.IsDownloadComplete()
|
||||
}, 10*time.Second, 100*time.Millisecond, "All downloads should complete within 10 seconds")
|
||||
|
||||
// Final state verification - testify makes these checks very readable
|
||||
assert.True(t, downloader.IsDownloadComplete(), "Download should be complete")
|
||||
assert.Equal(t, 3, downloader.GetTotalDownloadedArchivesCount(), "Should have downloaded all 3 archives")
|
||||
|
||||
// Verify all archives were processed
|
||||
assert.Len(t, completedArchives, 3, "Should have completed exactly 3 archives")
|
||||
assert.Contains(t, completedArchives, "archive-1", "Should have completed archive-1")
|
||||
assert.Contains(t, completedArchives, "archive-2", "Should have completed archive-2")
|
||||
assert.Contains(t, completedArchives, "archive-3", "Should have completed archive-3")
|
||||
|
||||
t.Log("✅ Multiple archives testify test passed")
|
||||
t.Logf(" - Completed %d out of %d archives", len(completedArchives), 3)
|
||||
}
|
||||
|
||||
// Helper function for testify tests
|
||||
func createTestIndexTestify() *protobuf.CodexWakuMessageArchiveIndex {
|
||||
return &protobuf.CodexWakuMessageArchiveIndex{
|
||||
Archives: map[string]*protobuf.CodexWakuMessageArchiveIndexMetadata{
|
||||
"test-archive-hash-1": {
|
||||
Cid: "test-cid-1",
|
||||
Metadata: &protobuf.WakuMessageArchiveMetadata{
|
||||
From: 1000,
|
||||
To: 2000,
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// Run the test suite
|
||||
func TestCodexArchiveDownloaderSuite(t *testing.T) {
|
||||
suite.Run(t, new(CodexArchiveDownloaderTestifySuite))
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user