From 9bb71be1431e20e50dcf77cc8822f805d11cd7a2 Mon Sep 17 00:00:00 2001
From: clemlak <clementlakhal@gmail.com>
Date: Thu, 16 Nov 2023 14:44:26 +0400
Subject: [PATCH] test: add G3MStrategy to Setup

---
 test/Setup.sol | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/test/Setup.sol b/test/Setup.sol
index 32d98ead..94d7202b 100644
--- a/test/Setup.sol
+++ b/test/Setup.sol
@@ -26,6 +26,7 @@ import "contracts/Portfolio.sol";
 
 // Strategy to test
 import "contracts/strategies/NormalStrategy.sol";
+import "contracts/strategies/G3MStrategy.sol";
 
 // Contracts in the test environment
 struct SubjectsType {
@@ -75,6 +76,9 @@ interface ISetup {
 
     /// @dev Returns the normal strategy address.
     function normalStrategy() external view returns (address);
+
+    /// @dev Returns the G3M strategy address.
+    function g3mStrategy() external view returns (address);
 }
 
 contract Setup is ISetup, Test, ERC1155TokenReceiver {
@@ -124,6 +128,9 @@ contract Setup is ISetup, Test, ERC1155TokenReceiver {
             address(new NormalStrategy(_subjects.portfolio));
         vm.label(_subjects.normalStrategy, "normal-strategy");
 
+        _subjects.g3mStrategy = address(new G3MStrategy(_subjects.portfolio));
+        vm.label(_subjects.g3mStrategy, "g3m-strategy");
+
         _ghost_state = GhostType({
             actor: address(this),
             subject: _subjects.portfolio,
@@ -516,5 +523,10 @@ contract Setup is ISetup, Test, ERC1155TokenReceiver {
         return _subjects.normalStrategy;
     }
 
+    /// @inheritdoc ISetup
+    function g3mStrategy() public view override returns (address) {
+        return _subjects.g3mStrategy;
+    }
+
     receive() external payable { }
 }