Function Test_stopBackgroundManagers

db/database_test.go:4371–4432

(t *testing.T)

}

func Test_stopBackgroundManagers(t *testing.T) {
	db, ctx := setupTestDB(t)
	defer db.Close(ctx)

	testCases := []struct {
		resyncManager               *BackgroundManager
		tombstoneCompactionManager  *BackgroundManager
		attachmentCompactionManager *BackgroundManager
		expected                    int
	}{
		{
			expected: 0,
		},
		{
			resyncManager: &BackgroundManager{
				name:    "test_resync",
				Process: &testBackgroundProcess{isStoppable: true},
			},
			expected: 1,
		},
		{
			resyncManager: &BackgroundManager{
				name:    "test_resync",
				Process: &testBackgroundProcess{isStoppable: true},
			},
			tombstoneCompactionManager: &BackgroundManager{
				name:    "test_tombstone",
				Process: &testBackgroundProcess{isStoppable: true},
			},
			attachmentCompactionManager: &BackgroundManager{
				name:    "test_attachment",
				Process: &testBackgroundProcess{isStoppable: true},
			},
			expected: 3,
		},
	}

	emptyOptions := map[string]interface{}{}

	for i, testCase := range testCases {
		t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
			db.ResyncManager = testCase.resyncManager
			db.AttachmentCompactionManager = testCase.attachmentCompactionManager
			db.TombstoneCompactionManager = testCase.tombstoneCompactionManager
			if db.ResyncManager != nil {
				err := db.ResyncManager.Start(ctx, emptyOptions)
				assert.NoError(t, err)
			}
			if db.AttachmentCompactionManager != nil {
				err := db.AttachmentCompactionManager.Start(ctx, emptyOptions)
				assert.NoError(t, err)
			}
			if db.TombstoneCompactionManager != nil {
				err := db.TombstoneCompactionManager.Start(ctx, emptyOptions)
				assert.NoError(t, err)
			}

			bgManagers := db.stopBackgroundManagers()