@@ -7,7 +7,6 @@ package cloning
7
7
import (
8
8
"testing"
9
9
10
- "github.com/stretchr/testify/assert"
11
10
"github.com/stretchr/testify/require"
12
11
13
12
"gitlab.com/postgres-ai/database-lab/v2/pkg/models"
@@ -28,17 +27,17 @@ func (s *BaseCloningSuite) TestLatestSnapshot() {
28
27
DataStateAt : "2020-02-20 00:00:00" ,
29
28
}
30
29
31
- assert .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
30
+ require .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
32
31
latestSnapshot , err := s .cloning .getLatestSnapshot ()
33
- assert .Nil (s .T (), latestSnapshot )
34
- assert .EqualError (s .T (), err , "no snapshot found" )
32
+ require .Nil (s .T (), latestSnapshot )
33
+ require .EqualError (s .T (), err , "no snapshot found" )
35
34
36
35
s .cloning .addSnapshot (snapshot1 )
37
36
s .cloning .addSnapshot (snapshot2 )
38
37
39
38
latestSnapshot , err = s .cloning .getLatestSnapshot ()
40
39
require .NoError (s .T (), err )
41
- assert .Equal (s .T (), latestSnapshot , snapshot1 )
40
+ require .Equal (s .T (), latestSnapshot , snapshot2 )
42
41
}
43
42
44
43
func (s * BaseCloningSuite ) TestSnapshotByID () {
@@ -56,28 +55,28 @@ func (s *BaseCloningSuite) TestSnapshotByID() {
56
55
DataStateAt : "2020-02-20 00:00:00" ,
57
56
}
58
57
59
- assert .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
58
+ require .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
60
59
latestSnapshot , err := s .cloning .getLatestSnapshot ()
61
- assert .Nil (s .T (), latestSnapshot )
62
- assert .EqualError (s .T (), err , "no snapshot found" )
60
+ require .Nil (s .T (), latestSnapshot )
61
+ require .EqualError (s .T (), err , "no snapshot found" )
63
62
64
63
s .cloning .addSnapshot (snapshot1 )
65
- assert .Equal (s .T (), 1 , len (s .cloning .snapshotBox .items ))
66
- assert .Equal (s .T (), "TestSnapshotID1" , s .cloning .snapshotBox .items [snapshot1 .ID ].ID )
64
+ require .Equal (s .T (), 1 , len (s .cloning .snapshotBox .items ))
65
+ require .Equal (s .T (), "TestSnapshotID1" , s .cloning .snapshotBox .items [snapshot1 .ID ].ID )
67
66
68
67
s .cloning .addSnapshot (snapshot2 )
69
- assert .Equal (s .T (), 2 , len (s .cloning .snapshotBox .items ))
70
- assert .Equal (s .T (), "TestSnapshotID2" , s .cloning .snapshotBox .items [snapshot2 .ID ].ID )
68
+ require .Equal (s .T (), 2 , len (s .cloning .snapshotBox .items ))
69
+ require .Equal (s .T (), "TestSnapshotID2" , s .cloning .snapshotBox .items [snapshot2 .ID ].ID )
71
70
72
71
latestSnapshot , err = s .cloning .getSnapshotByID ("TestSnapshotID2" )
73
72
require .NoError (s .T (), err )
74
- assert .Equal (s .T (), latestSnapshot , snapshot2 )
73
+ require .Equal (s .T (), latestSnapshot , snapshot2 )
75
74
76
75
s .cloning .resetSnapshots (make (map [string ]* models.Snapshot ))
77
- assert .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
76
+ require .Equal (s .T (), 0 , len (s .cloning .snapshotBox .items ))
78
77
latestSnapshot , err = s .cloning .getLatestSnapshot ()
79
- assert .Nil (s .T (), latestSnapshot )
80
- assert .EqualError (s .T (), err , "no snapshot found" )
78
+ require .Nil (s .T (), latestSnapshot )
79
+ require .EqualError (s .T (), err , "no snapshot found" )
81
80
}
82
81
83
82
func TestCloneCounter (t * testing.T ) {
@@ -91,15 +90,15 @@ func TestCloneCounter(t *testing.T) {
91
90
92
91
snapshot , err := c .getSnapshotByID ("testSnapshotID" )
93
92
require .Nil (t , err )
94
- assert .Equal (t , 0 , snapshot .NumClones )
93
+ require .Equal (t , 0 , snapshot .NumClones )
95
94
96
95
c .incrementCloneNumber ("testSnapshotID" )
97
96
snapshot , err = c .getSnapshotByID ("testSnapshotID" )
98
97
require .Nil (t , err )
99
- assert .Equal (t , 1 , snapshot .NumClones )
98
+ require .Equal (t , 1 , snapshot .NumClones )
100
99
101
100
c .decrementCloneNumber ("testSnapshotID" )
102
101
snapshot , err = c .getSnapshotByID ("testSnapshotID" )
103
102
require .Nil (t , err )
104
- assert .Equal (t , 0 , snapshot .NumClones )
103
+ require .Equal (t , 0 , snapshot .NumClones )
105
104
}
0 commit comments