Testing a maker contract

Work in progress

This page is currently being updated - thank you for your understanding.

After following the tutorial Post a Smart Offer or some of the guides in this section, you have created a maker contract and are ready to test it. Mangrove offers a helper contract, MangroveTest, that helps setup everything needed for writing a test using the Mangrove core protocol.

We are going to use Foundry as the test runner - and we are going to use some of the features that Foundry provides for testing. Please refer to Set Up Your Local Environment for instructions on how to get and setup Foundry. For this guide, we assume basic knowledge of writing and running tests with Foundry - as can be gained by reading the relevant sections on Foundry -> Tests in the Foundry book.

For this example, we are going to write a simple test for the Amplifier contract that we implemented in the advanced part of Creating a Direct contract.

Creating the test file

When creating the test file, remember to use the naming convention <name>.t.sol, this way Foundry knows what files are tests.

Imports

The first thing to do is to import the relevant contracts. As mentioned, we are going to use the helper contract provided by the Mangrove test library, MangroveTest. This allows us to avoid having to fork an existing chain with Mangrove deployed. We may simply deploy a Mangrove protocol for this specific purpose before running our tests.

The Mangrove test library provides other helpers. We import Polygon - a helper to fork the polygon chain. We do this because we want to use the real addresses for WETH, USDC and DAI. (This is not strictly necessary - we could just create some test tokens and use those instead.)

We also import the Amplifier contract - the contract that we wish to test. The last thing we import is MgvStructs, which contain struct definitions for the offer views, which we shall need in the test.

The console import is not strictly needed - however, it can be very useful, if we want to log something to the console while we are developing the test. Let us import it for now.

Amplifier.t.sol - Imports

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.10;

import {StratTest} from "@mgv-strats/test/lib/StratTest.sol";
import "@mgv/test/lib/forks/Polygon.sol";
import "@mgv-strats/src/toy_strategies/offer_maker/Amplifier.sol";
import {Local} from "@mgv/src/core/MgvLib.sol";
import {MgvReader} from "@mgv/src/periphery/MgvReader.sol";

import {console} from "@mgv/forge-std/console.sol";

See entire file on GitHub

Test Contract and Setup

We call the test contract AmplifierTest and let it inherit from MangroveTest to be able to use all helper functions. We override the basic setUp() function provided by MangroveTest to setup the actors we shall need in the test. We need 3 tokens, the fork the test should run on, a taker address and the Amplifier contract.

For this test, we are going to use the test contract as the offer maker. Consequentially, it should be able to receive funds (from the taker), so we make sure to define the receive() function.

We do the following in setUp() - refer to the full implementation below:

  • use the pinned fork of Polygon provided the Mangrove test library

  • get three tokens from the fork

  • setup the two relevant markets - DAI/WETH and USDC/WETH (using a helper from MangroveTest)

  • create an address for the taker, and provide it with native tokens, as well as DAI and USDC for taking the offer(s)

  • as the taker, we also need to approve Mangrove for taking the funds

In the implementation, we use standard cheatcodes, startPrank(<address>), stopPrank(<address> provided by Foundry for setting up the test (see Foundry -> Cheatcodes).

We also use a helper function $() provided by the Mangrove testing library offering a shorthand for writing address() and casting the contract to its address.

Amplifier.t.sol - Contract and Setup

contract AmplifierTest is StratTest {
  IERC20 weth;
  IERC20 dai;
  IERC20 usdc;

  PolygonFork fork;

  address payable taker;
  Amplifier strat;
  OLKey olKeyWethDai;
  uint constant GASREQ = 200_000;

  receive() external payable virtual {}

  function setUp() public override {
    // use the pinned Polygon fork
    fork = new PinnedPolygonFork(39880000); // use polygon fork to use dai, usdc and weth addresses
    fork.setUp();

    // use convenience helpers to setup Mangrove
    mgv = setupMangrove();
    reader = new MgvReader($(mgv));

    // setup tokens, markets and approve them
    dai = IERC20(fork.get("DAI.e"));
    weth = IERC20(fork.get("WETH.e"));
    usdc = IERC20(fork.get("USDC.e"));
    olKeyWethDai = OLKey($(weth), $(dai), options.defaultTickSpacing);
    olKey = OLKey($(usdc), $(weth), options.defaultTickSpacing);
    lo = olKey.flipped();

    setupMarket(olKeyWethDai);
    setupMarket(olKey);

    // setup separate taker and give some native token (for gas) + USDC and DAI
    taker = freshAddress("taker");
    deal(taker, 10_000_000);

    deal($(usdc), taker, cash(usdc, 10_000));
    deal($(dai), taker, cash(dai, 10_000));

    // approve DAI and USDC on Mangrove for taker
    vm.startPrank(taker);
    dai.approve($(mgv), type(uint).max);
    usdc.approve($(mgv), type(uint).max);
    vm.stopPrank();
  }

See entire file on GitHub

Testing a Successful Fill

With setUp done, let us write the first test.

We start by testing that we can take the offer fully without any rejections, and that after trade execution both offers are retracted (recall that was how we decided to implement Amplifier). Let us call the test test_success_fill - in line with the naming guidelines of the Foundry test framework.

Breaking the problem down, for this test, we need to

  • deploy Amplifier on our local fork,

  • take one of the offers, and verify our conditions

We implement these two steps as separate functions (we need to deploy the contract for other tests, also).

Amplifier.t.sol - Testing a successful fill

function test_success_fill() public {
  deployStrat();

  execTraderStratWithFillSuccess();
}

See entire file on GitHub

Now, of course, we need to implement deployStrat() and execTraderStratWithFillSuccess().

Deploying Amplifier on the fork

Deploying a contract on the fork is just calling its constructor of the contract. We give the address of the testing contract($(this)) as the administrator of the Amplifier contract to be able to use that directly as the maker.

After deployment, we use the activate() helper provided by MangroveOffer, which helps setup the correct token approvals for Mangrove for our maker contract.

Before calling activate() we also take the opportunity to demonstrate the related helper, checkList(), which checks whether necessary approvals have been setup, and, if not, reverts (recall that expectRevert(<message>) is a Foundry cheatcode).

Amplifier.t.sol - Deploying on a fork

function deployStrat() public {
  strat = new Amplifier({
    mgv: IMangrove($(mgv)),
    base: weth,
    stable1: usdc,
    stable2: dai,
    tickSpacing1: olKey.tickSpacing,
    tickSpacing2: olKeyWethDai.tickSpacing,
    admin: $(this) // for ease, set this contract (will be Test runner) as admin for the strat
  });

  // NOTE:
  // For this test, we're locking base, ie WETH, in the vault of the contract
  // - so Amplifier is not really used for amplified liquidity, in this example.
  // However, to employ actual amplified liquidity it is simply a matter of
  // setting up a more refined router.
  // check that we actually need to activate for the two 'wants' tokens
  IERC20[] memory tokens = new IERC20[](3);
  tokens[0] = dai;
  tokens[1] = usdc;
  tokens[2] = weth;

  vm.expectRevert("mgvOffer/LogicMustApproveMangrove");
  strat.checkList(tokens);

  // and now activate them
  strat.activate(tokens);
}

See entire file on GitHub

With this, we have a function that can deploy a new Amplifier contract on our local chain. Let us turn to writing the actual test.

Breaking down the test-specification further

Let us unfold the specification for the test itself.

For this first test, we want to

  • post and fund a pair of offers using Amplifier.newAmplifiedOffers(...),

  • snipe one of the offers, and,

  • verify these properties

    • both maker and taker received tokens as expected

    • after trade execution both offers are retracted

Let us continue our divide-and-conquer strategy and write helper methods for the two first steps, before writing the body of the test. (As we shall see, we can reuse these helper methods, later.)

Post and Fund Offers

Posting offers with Amplifier is simply a call to newAmplifiedOffers, sending along funds for the provision. (And we make a note to ensure in the test body that the testing contract has the funds for the provision.)

Amplifier.t.sol - Post and fund offers

function postAndFundOffers(uint makerGivesAmount, uint makerWantsAmountDAI, uint makerWantsAmountUSDC, uint gasreq)
  public
  returns (uint offerId1, uint offerId2)
{
  (offerId1, offerId2) = strat.newAmplifiedOffers{value: 2 ether}({
    gives: makerGivesAmount, // WETH
    wants1: makerWantsAmountUSDC, // USDC
    wants2: makerWantsAmountDAI, // DAI
    gasreq: gasreq
  });
}

See entire file on GitHub

Taking a Single Offer

Mangrove provides snipe exactly for taking a specific offer.

We make sure to impersonate the taker that we setup an address for above. (Recall that Foundry provides vm.prank() for this exact purpose.)

Amplifier.t.sol - Take a single offer

function takeOffer(uint makerWantsAmount, IERC20 makerWantsToken, uint offerId)
  public
  returns (uint takerGot, uint takerGave, uint bounty)
{
  OLKey memory _olKey = OLKey($(weth), $(makerWantsToken), olKey.tickSpacing);
  Tick tick = mgv.offers(_olKey, offerId).tick();
  // try to take one of the offers (using the separate taker account)
  vm.prank(taker);
  (takerGot, takerGave, bounty,) =
    mgv.marketOrderByTick({olKey: _olKey, maxTick: tick, fillVolume: makerWantsAmount, fillWants: false});
}

See entire file on GitHub

And Finally, The Test

With these helper methods, the test body amounts to the following steps:

  • Approve the router of Amplifier to access the WETH of the tester contract, which was given as reserveId when deploying the Amplifier (read more about Approvals).

  • Provide the tester contract with some WETH to be able to successfully give what the offer promises. (Using another cheatcode from Foundry, deal().)

  • Post and fund offers with postAndFundOffers().

  • Snipe one offer with takeOffer().

  • Assert and verify the properties we listed above.

Amplifier.t.sol - Putting it all together

function execTraderStratWithFillSuccess() public {
  uint makerGivesAmount = 0.15 ether;
  uint makerWantsAmountDAI = cash(dai, 300);
  uint makerWantsAmountUSDC = cash(usdc, 300);

  weth.approve($(strat.router()), type(uint).max);

  deal($(weth), $(this), cash(weth, 10));

  (uint offerId1, uint offerId2) =
    postAndFundOffers(makerGivesAmount, makerWantsAmountDAI, makerWantsAmountUSDC, GASREQ);

  (uint takerGot, uint takerGave,) = takeOffer(makerWantsAmountDAI, dai, offerId1);

  // assert that
  assertEq(takerGot, reader.minusFee(olKeyWethDai.flipped(), makerGivesAmount), "taker got wrong amount");
  assertTrue((makerWantsAmountDAI - takerGave) * 100000 / makerWantsAmountDAI < 10, "taker gave wrong amount");

  // assert that neither offer posted by Amplifier are live (= have been retracted)
  Offer offer_on_dai = mgv.offers(olKeyWethDai, offerId1);
  Offer offer_on_usdc = mgv.offers(olKey, offerId2);
  assertTrue(!offer_on_dai.isLive(), "weth->dai offer should have been retracted");
  assertTrue(!offer_on_usdc.isLive(), "weth->usdc offer should have been retracted");
}

See entire file on GitHub

In verifying, we use a few helper functions provided by MangroveTest

  • cash(token, amount) - that helps us ensure that we use the correct decimals for a specific token,

  • minusFee(address_outbound, address_inbound, price) - that calculates the fee for a given market and price,

and a few view functions provided by Mangrove, offers() and isLive() (see Views on Offers).

Running Tests with Foundry

In Foundry tests are run with the Forge tool, typically invoking something like forge test. Please refer to the Foundry Book -> Tests for updated documentation.

Adding More Tests - Testing Partial Fills

With our helper functions for posting and sniping offers, we we can easily add more tests. For instance, testing that an offer was correctly handled for a partial fill looks like this:

Amplifier.t.sol - Testing partial fill

function test_success_partialFill() public {
  deployStrat();

  execTraderStratWithPartialFillSuccess();
}

See entire file on GitHubAmplifier.t.sol - Helper function

function execTraderStratWithPartialFillSuccess() public {
  uint makerGivesAmount = 0.15 ether;
  uint makerWantsAmountDAI = cash(dai, 300);
  uint makerWantsAmountUSDC = cash(usdc, 300);

  weth.approve($(strat.router()), type(uint).max);

  deal($(weth), $(this), cash(weth, 5));

  // post offers with Amplifier liquidity
  (uint offerId1, uint offerId2) =
    postAndFundOffers(makerGivesAmount, makerWantsAmountDAI, makerWantsAmountUSDC, GASREQ);

  //only take half of the offer
  (uint takerGot, uint takerGave,) = takeOffer(makerWantsAmountDAI / 2, dai, offerId1);

  // assert that
  uint offerGaveMinusFee = reader.minusFee(olKeyWethDai.flipped(), makerGivesAmount / 2);
  assertTrue(((takerGot - offerGaveMinusFee) * 10_000) / (makerGivesAmount / 2) < 10, "taker got wrong amount");
  assertEq(takerGave, makerWantsAmountDAI / 2, "taker gave wrong amount");

  // assert that neither offer posted by Amplifier are live (= have been retracted)
  Offer offer_on_dai = mgv.offers(olKeyWethDai, offerId1);
  Offer offer_on_usdc = mgv.offers(lo, offerId2);
  assertTrue(offer_on_dai.isLive(), "weth->dai offer should not have been retracted");
  assertTrue(offer_on_usdc.isLive(), "weth->usdc offer should not have been retracted");
}

See entire file on GitHub

A full test file for for the Amplifier contract can be found here.

When you have sufficiently tested your maker contract, you may want to deploy your contract to a real chain.rovedao/mangrove-strats/blob/a265abeb96a053e386d346c7c9e431878382749c/test/toy_strategies/Amplifier.t.sol#L72-L76

https://github.com/mangrovedao/mangrove-strats/blob/a265abeb96a053e386d346c7c9e431878382749c/test/toy_strategies/Amplifier.t.sol#L143-L169

A full test file for for the Amplifier contract can be found here.

When you have sufficiently tested your %%maker contract|maker-contract%%, you may want to deploy your contract to a real chain.

Last updated