From 9c3763e66dab892d7692327aa1cdfdafaecc149b Mon Sep 17 00:00:00 2001
From: Neeraj Kashyap <nkashy1@gmail.com>
Date: Fri, 17 Dec 2021 17:03:45 -0800
Subject: [PATCH] Added test for removal of ERC20Facet

---
 dao/test_moonstream.py | 48 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 47 insertions(+), 1 deletion(-)

diff --git a/dao/test_moonstream.py b/dao/test_moonstream.py
index 83d2655..351b8df 100644
--- a/dao/test_moonstream.py
+++ b/dao/test_moonstream.py
@@ -1,7 +1,7 @@
 from brownie import accounts
 
 from . import ERC20Facet, ERC20Initializer
-from .core import facet_cut
+from .core import ZERO_ADDRESS, facet_cut
 from .test_core import MoonstreamDAOTestCase
 
 
@@ -67,3 +67,49 @@ class TestDeployment(MoonstreamDAOTestCase):
         symbol = diamond_erc20.symbol()
         expected_symbol = "MNSTR"
         self.assertEqual(symbol, expected_symbol)
+
+
+class TestRemoveFacet(MoonstreamDAOTestCase):
+    def test_remove_facet(self):
+        initializer = ERC20Initializer.ERC20Initializer(None)
+        initializer.deploy({"from": accounts[0]})
+
+        erc20_facet = ERC20Facet.ERC20Facet(None)
+        erc20_facet.deploy({"from": accounts[0]})
+
+        diamond_address = self.contracts["Diamond"]
+        facet_cut(
+            diamond_address,
+            "ERC20Facet",
+            erc20_facet.address,
+            "add",
+            {"from": accounts[0]},
+            initializer.address,
+        )
+
+        diamond_erc20 = ERC20Facet.ERC20Facet(diamond_address)
+        name = diamond_erc20.name()
+        expected_name = "Moonstream DAO"
+        self.assertEqual(name, expected_name)
+
+        symbol = diamond_erc20.symbol()
+        expected_symbol = "MNSTR"
+        self.assertEqual(symbol, expected_symbol)
+
+        decimals = diamond_erc20.decimals()
+        expected_decimals = 18
+        self.assertEqual(decimals, expected_decimals)
+
+        facet_cut(
+            diamond_address,
+            "ERC20Facet",
+            ZERO_ADDRESS,
+            "remove",
+            {"from": accounts[0]},
+        )
+
+        with self.assertRaises(Exception):
+            name = diamond_erc20.name()
+
+        with self.assertRaises(Exception):
+            symbol = diamond_erc20.symbol()