fix test

akaladarshi · Jul 30, 2024 · e75c5ff · e75c5ff
1 parent 724099c
commit e75c5ff
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/cli/send_test.go b/cli/send_test.go
@@ -59,16 +59,17 @@ func TestSendCLI(t *testing.T) {
 
 		gomock.InOrder(
 			mockSrvcs.EXPECT().MessageForSend(gomock.Any(), SendParams{
-				To:  mustAddr(address.NewIDAddress(1)),
-				Val: oneFil,
+				From: mustAddr(address.NewIDAddress(1)),
+				To:   mustAddr(address.NewIDAddress(1)),
+				Val:  oneFil,
 			}).Return(arbtProto, nil),
 			mockSrvcs.EXPECT().PublishMessage(gomock.Any(), arbtProto, false).
 				Return(sigMsg, nil, nil),
 			mockSrvcs.EXPECT().Close(),
 		)
-		err := app.Run([]string{"lotus", "send", "t01", "1"})
+		err := app.Run([]string{"lotus", "send", "--from", "t01", "t01", "1"})
 		require.NoError(t, err)
-		require.EqualValues(t, sigMsg.Cid().String()+"\n", buf.String())
+		require.Contains(t, buf.String(), sigMsg.Cid().String()+"\n")
 	})
 }
 
@@ -112,6 +113,6 @@ func TestSendEthereum(t *testing.T) {
 		)
 		err = app.Run([]string{"lotus", "send", "--from-eth-addr", testEthAddr.String(), "--params-hex", "01020304", "f01", "1"})
 		require.NoError(t, err)
-		require.EqualValues(t, sigMsg.Cid().String()+"\n", buf.String())
+		require.Contains(t, buf.String(), sigMsg.Cid().String()+"\n")
 	})
 }