From 9d21d65bc71b5386c7f1373d7df2416557017e68 Mon Sep 17 00:00:00 2001 From: Vaibhav Kamra Date: Thu, 7 Jul 2022 17:23:34 -0700 Subject: [PATCH] Use MockExchangeData in kopia backup tests (#303) This implements the DataStreamInfo interface needed by BackupCollections --- .../connector/mockconnector/mock_data_collection.go | 8 ++++---- src/internal/kopia/wrapper_test.go | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/internal/connector/mockconnector/mock_data_collection.go b/src/internal/connector/mockconnector/mock_data_collection.go index 01724606a..88dd76383 100644 --- a/src/internal/connector/mockconnector/mock_data_collection.go +++ b/src/internal/connector/mockconnector/mock_data_collection.go @@ -54,14 +54,14 @@ func (medc *MockExchangeDataCollection) Items() <-chan connector.DataStream { // ExchangeData represents a single item retrieved from exchange type MockExchangeData struct { - id string - reader io.ReadCloser + ID string + Reader io.ReadCloser } func (med *MockExchangeData) UUID() string { - return med.id + return med.ID } func (med *MockExchangeData) ToReader() io.ReadCloser { - return med.reader + return med.Reader } diff --git a/src/internal/kopia/wrapper_test.go b/src/internal/kopia/wrapper_test.go index 57d66aa05..49c9640c9 100644 --- a/src/internal/kopia/wrapper_test.go +++ b/src/internal/kopia/wrapper_test.go @@ -304,9 +304,9 @@ func (suite *KopiaSimpleRepoIntegrationSuite) SetupTest() { &kopiaDataCollection{ path: testPath, streams: []connector.DataStream{ - &kopiaDataStream{ - uuid: testFileUUID, - reader: io.NopCloser(bytes.NewReader(testFileData)), + &mockconnector.MockExchangeData{ + ID: testFileUUID, + Reader: io.NopCloser(bytes.NewReader(testFileData)), }, }, }, @@ -314,10 +314,10 @@ func (suite *KopiaSimpleRepoIntegrationSuite) SetupTest() { stats, err := suite.w.BackupCollections(suite.ctx, collections) require.NoError(t, err) + require.Equal(t, stats.ErrorCount, 0) require.Equal(t, stats.TotalFileCount, 1) require.Equal(t, stats.TotalDirectoryCount, 3) require.Equal(t, stats.IgnoredErrorCount, 0) - require.Equal(t, stats.ErrorCount, 0) require.False(t, stats.Incomplete) suite.snapshotID = manifest.ID(stats.SnapshotID)