From b5ee1f2658970e8f28cb835d1053770f507c805f Mon Sep 17 00:00:00 2001
From: AdoAdoAdo <anaridado@gmail.com>
Date: Wed, 16 Mar 2022 14:57:54 +0200
Subject: [PATCH] data: add unit tests

---
 data/block/blockV2_test.go   | 23 +++++++++++++++++++++++
 data/block/block_test.go     | 17 +++++++++++++++++
 data/block/metaBlock_test.go | 17 +++++++++++++++++
 3 files changed, 57 insertions(+)

diff --git a/data/block/blockV2_test.go b/data/block/blockV2_test.go
index 33664b540..6503f92d1 100644
--- a/data/block/blockV2_test.go
+++ b/data/block/blockV2_test.go
@@ -1194,3 +1194,26 @@ func TestHeaderV2_SetAdditionalDataShouldWork(t *testing.T) {
 	require.Equal(t, gasRefunded, shardBlock.GetScheduledGasRefunded())
 	require.Equal(t, gasProvided, shardBlock.GetScheduledGasProvided())
 }
+
+func TestHeaderV2_HasScheduledMiniBlocks(t *testing.T) {
+	t.Parallel()
+
+	mbh := &block.MiniBlockHeader{}
+	_ = mbh.SetProcessingType(int32(block.Scheduled))
+
+	shardBlock := &block.HeaderV2{
+		Header: &block.Header{
+			MiniBlockHeaders: []block.MiniBlockHeader{*mbh},
+		},
+	}
+	require.True(t, shardBlock.HasScheduledMiniBlocks())
+
+	_ = mbh.SetProcessingType(int32(block.Normal))
+	shardBlock = &block.HeaderV2{
+		Header: &block.Header{
+			MiniBlockHeaders: []block.MiniBlockHeader{*mbh},
+		},
+	}
+
+	require.False(t, shardBlock.HasScheduledMiniBlocks())
+}
diff --git a/data/block/block_test.go b/data/block/block_test.go
index 4958848a1..71c4b82d1 100644
--- a/data/block/block_test.go
+++ b/data/block/block_test.go
@@ -806,3 +806,20 @@ func TestHeader_GetAdditionalData(t *testing.T) {
 	additionalData := h.GetAdditionalData()
 	assert.Nil(t, additionalData)
 }
+
+func TestHeader_HasScheduledMiniBlocks(t *testing.T) {
+	t.Parallel()
+
+	h := &block.Header{}
+	require.False(t, h.HasScheduledMiniBlocks())
+
+	mbHeader := &block.MiniBlockHeader{}
+	_ = mbHeader.SetProcessingType(int32(block.Normal))
+	h.MiniBlockHeaders = []block.MiniBlockHeader{*mbHeader}
+	require.False(t, h.HasScheduledMiniBlocks())
+
+	// not suppoerted for v1 header, so it should return false
+	_ = mbHeader.SetProcessingType(int32(block.Scheduled))
+	h.MiniBlockHeaders = []block.MiniBlockHeader{*mbHeader}
+	require.False(t, h.HasScheduledMiniBlocks())
+}
diff --git a/data/block/metaBlock_test.go b/data/block/metaBlock_test.go
index 56ec35cd3..3ed724b60 100644
--- a/data/block/metaBlock_test.go
+++ b/data/block/metaBlock_test.go
@@ -385,3 +385,20 @@ func TestMetaBlock_SetAdditionalDataShouldDoNothing(t *testing.T) {
 	err := metaBlock.SetAdditionalData(&headerVersionData.AdditionalData{})
 	require.Nil(t, err)
 }
+
+func TestMetaBlock_HasScheduledMiniBlocks(t *testing.T) {
+	t.Parallel()
+
+	metaBlock := &block.MetaBlock{}
+	require.False(t, metaBlock.HasScheduledMiniBlocks())
+
+	mbHeader := &block.MiniBlockHeader{}
+	_ = mbHeader.SetProcessingType(int32(block.Normal))
+	metaBlock.MiniBlockHeaders = []block.MiniBlockHeader{*mbHeader}
+	require.False(t, metaBlock.HasScheduledMiniBlocks())
+
+	_ = mbHeader.SetProcessingType(int32(block.Scheduled))
+	metaBlock.MiniBlockHeaders = []block.MiniBlockHeader{*mbHeader}
+
+	require.True(t, metaBlock.HasScheduledMiniBlocks())
+}