@@ -23,6 +23,7 @@ import (
2323 "time"
2424
2525 "github.com/algorand/go-algorand/daemon/algod/api/spec/v1"
26+ "github.com/algorand/go-algorand/test/partitiontest"
2627 "github.com/stretchr/testify/require"
2728)
2829
@@ -38,6 +39,7 @@ func bw(client Client) *blockWatcher {
3839// When the status continues to report block 300
3940// Then blockIfStalled will block until the next block is reported
4041func TestBlockIfStalled (t * testing.T ) {
42+ partitiontest .PartitionTest (t )
4143 client := mockClient {
4244 error : []error {nil , nil , nil },
4345 status : makeNodeStatuses (300 , 300 , 300 , 301 ),
@@ -59,6 +61,7 @@ func TestBlockIfStalled(t *testing.T) {
5961// When the status continues to increase quickly
6062// Then blockIfCatchup will block until a block is reported twice
6163func TestBlockIfCatchup (t * testing.T ) {
64+ partitiontest .PartitionTest (t )
6265 client := mockClient {
6366 error : []error {nil , nil , nil },
6467 status : makeNodeStatuses (301 , 302 , 303 , 304 , 305 , 306 , 307 , 308 , 309 , 310 , 310 ),
@@ -80,6 +83,7 @@ func TestBlockIfCatchup(t *testing.T) {
8083// When the status is not changing quickly
8184// Then blockIfCatchup will return after the first status call.
8285func TestBlockIfCaughtUp (t * testing.T ) {
86+ partitiontest .PartitionTest (t )
8387 client := mockClient {
8488 error : []error {nil , nil , nil },
8589 status : makeNodeStatuses (300 ),
@@ -111,6 +115,7 @@ func (l *testlistener) onBlock(block v1.Block) {
111115}
112116
113117func TestE2E (t * testing.T ) {
118+ partitiontest .PartitionTest (t )
114119 client := makeMockClient (
115120 []error {nil , nil , nil },
116121 makeNodeStatuses (300 , 301 , 302 , 302 , 302 , 302 , 302 , 302 , 310 , 320 , 321 , 321 , 321 , 322 ),
@@ -159,6 +164,7 @@ func TestE2E(t *testing.T) {
159164}
160165
161166func TestAbortDuringStall (t * testing.T ) {
167+ partitiontest .PartitionTest (t )
162168 client := makeMockClient (
163169 []error {},
164170 makeNodeStatuses (300 ),
0 commit comments