From 0e16989b4da8ae1d3d35b3d07f3f42bd4687987c Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Sun, 5 Nov 2023 14:13:39 +0800
Subject: [PATCH] [fix] sui tests

---
 sui/api/process_test.go          | 8 ++++----
 sui/storages/local/block_test.go | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/sui/api/process_test.go b/sui/api/process_test.go
index 39692af6..d4557582 100644
--- a/sui/api/process_test.go
+++ b/sui/api/process_test.go
@@ -131,11 +131,11 @@ func TestBlockGet(t *testing.T) {
 	}
 
 	assert.IsType(t, []core.IBlock{}, res)
-	assert.Equal(t, 4, len(res.([]core.IBlock)))
+	assert.Equal(t, 6, len(res.([]core.IBlock)))
 	assert.Equal(t, "ColumnsTwo", res.([]core.IBlock)[0].(*local.Block).ID)
 	assert.Equal(t, "Hero", res.([]core.IBlock)[1].(*local.Block).ID)
-	assert.Equal(t, "Section", res.([]core.IBlock)[2].(*local.Block).ID)
-	assert.Equal(t, "Table", res.([]core.IBlock)[3].(*local.Block).ID)
+	assert.Equal(t, "Image", res.([]core.IBlock)[2].(*local.Block).ID)
+	assert.Equal(t, "Section", res.([]core.IBlock)[3].(*local.Block).ID)
 }
 
 func TestBlockFind(t *testing.T) {
@@ -210,7 +210,7 @@ func TestTemplateComponentGet(t *testing.T) {
 	}
 
 	assert.IsType(t, []core.IComponent{}, res)
-	assert.Equal(t, 3, len(res.([]core.IComponent)))
+	assert.Equal(t, 5, len(res.([]core.IComponent)))
 	assert.Equal(t, "Box", res.([]core.IComponent)[0].(*local.Component).ID)
 	assert.Equal(t, "Card", res.([]core.IComponent)[1].(*local.Component).ID)
 	assert.Equal(t, "Nav", res.([]core.IComponent)[2].(*local.Component).ID)
diff --git a/sui/storages/local/block_test.go b/sui/storages/local/block_test.go
index 9eaf6460..aa41e253 100644
--- a/sui/storages/local/block_test.go
+++ b/sui/storages/local/block_test.go
@@ -34,10 +34,10 @@ func TestTemplateBlocks(t *testing.T) {
 	assert.Equal(t, "/Hero/Hero.js", blocks[1].(*Block).Codes.JS.File)
 	assert.Equal(t, "/Hero/Hero.ts", blocks[1].(*Block).Codes.TS.File)
 
-	assert.Equal(t, "Section", blocks[2].(*Block).ID)
-	assert.Equal(t, "/Section/Section.html", blocks[2].(*Block).Codes.HTML.File)
-	assert.Equal(t, "/Section/Section.js", blocks[2].(*Block).Codes.JS.File)
-	assert.Equal(t, "/Section/Section.ts", blocks[2].(*Block).Codes.TS.File)
+	assert.Equal(t, "Image", blocks[2].(*Block).ID)
+	assert.Equal(t, "/Image/Image.html", blocks[2].(*Block).Codes.HTML.File)
+	assert.Equal(t, "/Image/Image.js", blocks[2].(*Block).Codes.JS.File)
+	assert.Equal(t, "/Image/Image.ts", blocks[2].(*Block).Codes.TS.File)
 }
 
 func TestTemplateBlockJS(t *testing.T) {