-
Notifications
You must be signed in to change notification settings - Fork 3
/
write_worker_test.go
106 lines (95 loc) · 4.19 KB
/
write_worker_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/* This file is part of copyondemand.
*
* Copyright © 2020 Datto, Inc.
* Author: Bryan Ehrlich <[email protected]>
*
* Licensed under the Apache Software License, Version 2.0
* Fedora-License-Identifier: ASL 2.0
* SPDX-2.0-License-Identifier: Apache-2.0
* SPDX-3.0-License-Identifier: Apache-2.0
*
* copyondemand is free software.
* For more information on the license, see LICENSE.
* For more information on free software, see <https://www.gnu.org/philosophy/free-sw.en.html>.
*
* You may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package copyondemand
import (
"github.com/sirupsen/logrus"
"github.com/sirupsen/logrus/hooks/test"
"github.com/stretchr/testify/assert"
"testing"
)
func TestHandleWriteDataActionWrites(t *testing.T) {
logger, _ := test.NewNullLogger()
logger.SetLevel(logrus.DebugLevel)
mockSourceFile := new(MockFile)
mockSourceFile.On("Fd").Return(1)
mockBackingFile := new(MockFile)
fileBackedDevice := CreateMockDevice(t, mockSourceFile, mockBackingFile, logger, BlockSize*2)
readData := fill1s(make([]byte, BlockSize*2))
mockBackingFile.On("WriteAt", readData[0:BlockSize], int64(0)).Return(BlockSize, nil)
mockBackingFile.On("WriteAt", readData[BlockSize:BlockSize*2], int64(BlockSize)).Return(BlockSize, nil)
writeAction := &QueuedWriteAction{}
writeAction.startBlock = 0
writeAction.endBlock = 1
writeAction.data = readData
handleWriteDataAction(fileBackedDevice, writeAction)
assert.True(t, fileBackedDevice.blockMap.isSynced(0))
assert.True(t, fileBackedDevice.blockMap.isSynced(1))
}
func TestHandleWriteDataReconcilesDirtyData(t *testing.T) {
logger, _ := test.NewNullLogger()
logger.SetLevel(logrus.DebugLevel)
mockSourceFile := new(MockFile)
mockSourceFile.On("Fd").Return(1)
mockBackingFile := new(MockFile)
mockBackingFile.output = 2
fileBackedDevice := CreateMockDevice(t, mockSourceFile, mockBackingFile, logger, BlockSize*2)
readData := fill1s(make([]byte, BlockSize*2))
expectedData := fill1s(make([]byte, BlockSize*2))
fill(expectedData[(BlockSize+(BlockSize/4)):(BlockSize*2-(BlockSize/4))], 2)
mockBackingFile.On("WriteAt", readData[0:BlockSize], int64(0)).Return(BlockSize, nil)
mockBackingFile.On("WriteAt", readData[BlockSize:BlockSize*2], int64(BlockSize)).Return(BlockSize, nil)
mockBackingFile.On("ReadAt", make([]byte, BlockSize), int64(BlockSize)).Return(BlockSize, nil)
fileBackedDevice.dirtyBlockMap.recordWrite(1, BlockSize/4, BlockSize/2)
writeAction := &QueuedWriteAction{}
writeAction.startBlock = 0
writeAction.endBlock = 1
writeAction.data = readData
handleWriteDataAction(fileBackedDevice, writeAction)
assert.True(t, fileBackedDevice.blockMap.isSynced(0))
assert.True(t, fileBackedDevice.blockMap.isSynced(1))
assert.Equal(t, expectedData, readData)
}
func TestFixDirtyBlockWorks(t *testing.T) {
logger, _ := test.NewNullLogger()
logger.SetLevel(logrus.DebugLevel)
mockSourceFile := new(MockFile)
mockSourceFile.On("Fd").Return(1)
mockBackingFile := new(MockFile)
mockSourceFile.output = 1
mockBackingFile.output = 2
fileBackedDevice := CreateMockDevice(t, mockSourceFile, mockBackingFile, logger, BlockSize*2)
expectedData := fill1s(make([]byte, BlockSize))
fill(expectedData[(BlockSize/4):(BlockSize-(BlockSize/4))], 2)
mockBackingFile.On("WriteAt", expectedData, int64(BlockSize)).Return(BlockSize, nil)
mockBackingFile.On("ReadAt", make([]byte, BlockSize), int64(BlockSize)).Return(BlockSize, nil)
mockSourceFile.On("ReadAt", make([]byte, BlockSize), int64(BlockSize)).Return(BlockSize, nil)
fileBackedDevice.dirtyBlockMap.recordWrite(1, BlockSize/4, BlockSize/2)
writeAction := &QueuedWriteAction{}
writeAction.startBlock = 1
handleFixDirtyBlockAction(fileBackedDevice, writeAction)
assert.False(t, fileBackedDevice.dirtyBlockMap.isDirty(1))
assert.True(t, fileBackedDevice.blockMap.isSynced(1))
}