Function TestCloseSyncWhileFillEmpty

ethstorage/p2p/protocol/sync_test.go:1081–1125

TestCloseSyncWhileFillEmpty test the sync can be cancel while the fill empty is running.

(t *testing.T)

// TestCloseSyncWhileFillEmpty test the sync can be cancel while the fill empty is running.
func TestCloseSyncWhileFillEmpty(t *testing.T) {
	var (
		kvSize      = defaultChunkSize
		kvEntries   = uint64(512)
		lastKvIndex = uint64(0)
		db          = rawdb.NewMemoryDatabase()
		mux         = new(event.Feed)
		shards      = []uint64{0}
		shardMap    = make(map[common.Address][]uint64)
		m           = metrics.NewMetrics("sync_test")
		chainID     = new(big.Int).SetUint64(3333)
	)

	metafile, err := CreateMetaFile(metafileName, int64(kvEntries))
	if err != nil {
		t.Error("Create metafileName fail", err.Error())
	}
	defer metafile.Close()

	shardMap[contract] = shards
	shardManager, files := createEthStorage(contract, shards, defaultChunkSize, kvSize, kvEntries, common.Address{}, defaultEncodeType)
	if shardManager == nil {
		t.Fatalf("createEthStorage failed")
	}
	defer func(files []string) {
		for _, file := range files {
			os.Remove(file)
		}
	}(files)

	makeKVStorage(contract, shards, defaultChunkSize, kvSize, kvEntries, lastKvIndex, common.Address{}, defaultEncodeType, metafile)

	l1 := NewMockL1Source(lastKvIndex, metafileName)
	sm := ethstorage.NewStorageManager(shardManager, l1, lg)
	sm.Reset(0)
	_, syncCl := createLocalHostAndSyncClient(t, chainID, db, sm, m, mux)
	syncCl.Start()
	time.Sleep(10 * time.Millisecond)
	syncCl.Close()

	t.Log("Fill empty status", "filled", syncCl.tasks[0].state.EmptyFilled, "toFill", syncCl.tasks[0].state.EmptyToFill)
	if syncCl.syncDone {
		t.Fatalf("fill empty should be cancel")
	}
}

// TestAddPeerAfterSyncDone test add peer after sync done, the peer should add successfully (the connection is kept),
// as the remote peer may need to sync data from this local peer, we also need to use the sync client to control