1818
1919import com .tinyengine .it .common .base .Result ;
2020import com .tinyengine .it .mapper .BlockGroupMapper ;
21- import com .tinyengine .it .model .dto .BlockGroupDto ;
2221import com .tinyengine .it .model .entity .BlockGroup ;
2322import com .tinyengine .it .service .material .BlockGroupService ;
2423
3029import org .mockito .MockitoAnnotations ;
3130
3231import java .util .Arrays ;
32+ import java .util .Collections ;
3333import java .util .List ;
3434
3535/**
@@ -64,21 +64,21 @@ void testGetAllBlockGroups() {
6464
6565 @ Test
6666 void testCreateBlockGroups () {
67- when (blockGroupService .createBlockGroup (any (BlockGroup .class ))).thenReturn (new Result <List <BlockGroupDto >>());
67+ when (blockGroupService .createBlockGroup (any (BlockGroup .class ))).thenReturn (new Result <List <BlockGroup >>());
6868
69- Result <List <BlockGroupDto >> result = blockGroupController .createBlockGroups (new BlockGroup ());
70- Assertions .assertEquals (new Result <List <BlockGroupDto >>(), result );
69+ Result <List <BlockGroup >> result = blockGroupController .createBlockGroups (new BlockGroup ());
70+ Assertions .assertEquals (new Result <List <BlockGroup >>(), result );
7171 }
7272
7373 @ Test
7474 void testUpdateBlockGroups () {
75- when (blockGroupService .updateBlockGroupById (any (BlockGroup .class ))).thenReturn (Integer . valueOf ( 0 ) );
76- BlockGroupDto blockGroupDto = new BlockGroupDto ();
77- when (blockGroupMapper . getBlockGroupsById ( anyInt ())) .thenReturn (Arrays .< BlockGroupDto > asList ( blockGroupDto ) );
75+ when (blockGroupService .updateBlockGroupById (any (BlockGroup .class ))).thenReturn (1 );
76+ BlockGroup blockGroup = new BlockGroup ();
77+ when (blockGroupService . findBlockGroupById ( 1 )) .thenReturn (blockGroup );
7878
79- Result <List <BlockGroupDto >> result =
80- blockGroupController .updateBlockGroups (Integer . valueOf ( 0 ) , new BlockGroup ());
81- Assertions .assertEquals (blockGroupDto , result .getData (). get ( 0 ));
79+ Result <List <BlockGroup >> result =
80+ blockGroupController .updateBlockGroups (1 , new BlockGroup ());
81+ Assertions .assertEquals ("200" , result .getCode ( ));
8282 }
8383
8484 @ Test
@@ -87,10 +87,8 @@ void testDeleteBlockGroups() {
8787 mockData .setId (1 );
8888 when (blockGroupService .findBlockGroupById (anyInt ())).thenReturn (mockData );
8989 when (blockGroupService .deleteBlockGroupById (anyInt ())).thenReturn (Integer .valueOf (0 ));
90- BlockGroupDto resultData = new BlockGroupDto ();
91- when (blockGroupMapper .getBlockGroupsById (anyInt ())).thenReturn (Arrays .<BlockGroupDto >asList (resultData ));
9290
93- Result <List <BlockGroupDto >> result = blockGroupController .deleteBlockGroups (1 );
94- Assertions .assertEquals (resultData , result .getData (). get ( 0 ));
91+ Result <List <BlockGroup >> result = blockGroupController .deleteBlockGroups (1 );
92+ Assertions .assertEquals ("200" , result .getCode ( ));
9593 }
9694}
0 commit comments