|
package main_test
|
Mock storage implementation that is used in unit tests.
Documentation in literate-programming-style is available at:
https://redhatinsights.github.io/ccx-notification-writer/packages/mockstoragetest.html
|
import (
"time"
types "github.com/RedHatInsights/insights-results-types"
)
|
MockStorage structure represents a mocked implementation of Storage
interface to be used by tested consumer
|
type MockStorage struct {
databaseInitMigrationCalled int
databaseInitializationCalled int
databaseCleanupCalled int
databaseDropTablesCalled int
databaseDropIndexesCalled int
getLatestKafkaOffsetCalled int
printNewReportsForCleanupCalled int
cleanupNewReportsCalled int
printOldReportsForCleanupCalled int
cleanupOldReportsCalled int
printReadErrorsForCleanupCalled int
cleanupReadReportsCalled int
closeCalled int
writeReportCalled int
}
|
NewMockStorage constructs new mock storage instance
|
func NewMockStorage ( ) MockStorage {
return MockStorage {
databaseInitMigrationCalled : 0 ,
databaseInitializationCalled : 0 ,
databaseCleanupCalled : 0 ,
databaseDropTablesCalled : 0 ,
databaseDropIndexesCalled : 0 ,
getLatestKafkaOffsetCalled : 0 ,
printNewReportsForCleanupCalled : 0 ,
cleanupNewReportsCalled : 0 ,
printOldReportsForCleanupCalled : 0 ,
cleanupOldReportsCalled : 0 ,
printReadErrorsForCleanupCalled : 0 ,
cleanupReadReportsCalled : 0 ,
closeCalled : 0 ,
writeReportCalled : 0 ,
}
}
|
Close is a mocked reimplementation of the real Close method.
|
func ( storage * MockStorage ) Close ( ) error {
storage . closeCalled ++
|
return no error
|
return nil
}
|
WriteReportForCluster is a mocked reimplementation of the real
WriteReportForCluster method.
|
func ( storage * MockStorage ) WriteReportForCluster ( _ types . OrgID , _ types . AccountNumber , _ types . ClusterName , _ types . ClusterReport , _ time . Time , _ types . KafkaOffset ) error {
storage . writeReportCalled ++
|
return no error
|
return nil
}
|
DatabaseInitialization is a mocked reimplementation of the real
DatabaseInitialization method.
|
func ( storage * MockStorage ) DatabaseInitialization ( ) error {
storage . databaseInitializationCalled ++
|
return no error
|
return nil
}
|
DatabaseInitMigration is a mocked reimplementation of the real
DatabaseInitMigration method.
|
func ( storage * MockStorage ) DatabaseInitMigration ( ) error {
storage . databaseInitMigrationCalled ++
|
return no error
|
return nil
}
|
DatabaseCleanup is a mocked reimplementation of the real DatabaseCleanup
method.
|
func ( storage * MockStorage ) DatabaseCleanup ( ) error {
storage . databaseCleanupCalled ++
|
return no error
|
return nil
}
|
DatabaseDropTables is a mocked reimplementation of the real
DatabaseDropTables method.
|
func ( storage * MockStorage ) DatabaseDropTables ( ) error {
storage . databaseDropTablesCalled ++
|
return no error
|
return nil
}
|
DatabaseDropIndexes is a mocked reimplementation of the real
DatabaseDropIndexes method.
|
func ( storage * MockStorage ) DatabaseDropIndexes ( ) error {
storage . databaseDropIndexesCalled ++
|
return no error
|
return nil
}
|
GetLatestKafkaOffset is a mocked reimplementation of the real
GetLatestKafkaOffset method.
|
func ( storage * MockStorage ) GetLatestKafkaOffset ( ) ( types . KafkaOffset , error ) {
storage . getLatestKafkaOffsetCalled ++
|
return some offset + no error
|
return 1 , nil
}
|
PrintNewReportsForCleanup is a mocked reimplementation of the real
PrintNewReportsForCleanup method.
|
func ( storage * MockStorage ) PrintNewReportsForCleanup ( _ string ) error {
storage . printNewReportsForCleanupCalled ++
|
return no error
|
return nil
}
|
CleanupNewReports is a mocked reimplementation of the real CleanupNewReports
method.
|
func ( storage * MockStorage ) CleanupNewReports ( _ string ) ( int , error ) {
storage . cleanupNewReportsCalled ++
|
return number of cleaned records + no error
|
return 1 , nil
}
|
PrintOldReportsForCleanup is a mocked reimplementation of the real
PrintOldReportsForCleanup method.
|
func ( storage * MockStorage ) PrintOldReportsForCleanup ( _ string ) error {
storage . printOldReportsForCleanupCalled ++
|
return no error
|
return nil
}
|
CleanupOldReports is a mocked reimplementation of the real CleanupOldReports
method.
|
func ( storage * MockStorage ) CleanupOldReports ( _ string ) ( int , error ) {
storage . cleanupOldReportsCalled ++
|
return number of cleaned records + no error
|
return 1 , nil
}
|
PrintReadErrorsForCleanup is a mocked reimplementation of the real
PrintReadErrorsForCleanup method.
|
func ( storage * MockStorage ) PrintReadErrorsForCleanup ( _ string ) error {
storage . printReadErrorsForCleanupCalled ++
|
return no error
|
return nil
}
|
CleanupReadErrors is a mocked reimplementation of the real CleanupReadErrors
method.
|
func ( storage * MockStorage ) CleanupReadErrors ( _ string ) ( int , error ) {
storage . cleanupOldReportsCalled ++
|
return number of cleaned records + no error
|
return 1 , nil
}
|