support withdrawAndHarvest

richard-xtek · Apr 16, 2021 · 85c012a · 85c012a
1 parent 4e6f786
commit 85c012a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/spec/MasterChefV2.spec b/spec/MasterChefV2.spec
@@ -243,6 +243,8 @@ rule preserveTotalAssetOfUser(method f, uint256 pid, address user,
 		withdraw(e, pid, amount, to);
 	} else if (f.selector == emergencyWithdraw(uint256, address).selector) {
 		emergencyWithdraw(e, pid, to);
+	} else if (f.selector == withdrawAndHarvest(uint256,uint256,address).selector) {
+		withdrawAndHarvest(e, pid, amount, to);
 	} else {
 		calldataarg args;
 		f(e, args);
@@ -385,7 +387,10 @@ function callFunctionWithParams(method f, uint256 pid, address sender, address t
 		harvest(e, pid, to);
 	} else if (f.selector == emergencyWithdraw(uint256, address).selector) {
 		emergencyWithdraw(e, pid, to);
-	} else {
+	} else if (f.selector == withdrawAndHarvest(uint256,uint256,address).selector) {
+		withdrawAndHarvest(e, pid, amount, to);
+	}
+	else {
 		calldataarg args;
 		f(e,args);
 	}