test stagepolicy

planetarium · Jul 25, 2023 · 8ba2414 · 8ba2414
1 parent ae04edd
commit 8ba2414
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Lib9c.Policy/NCStagePolicy.cs b/Lib9c.Policy/NCStagePolicy.cs
@@ -18,7 +18,7 @@ public class NCStagePolicy : IStagePolicy
         private readonly ConcurrentDictionary<Address, SortedList<Transaction, TxId>> _txs;
         private readonly int _quotaPerSigner;
 
-        private ImmutableHashSet<Address> _bannedAccounts = new[]
+        public ImmutableHashSet<Address> BannedAccounts = new[]
         {
             new Address("de96aa7702a7a1fd18ee0f84a5a0c7a2c28ec840"),
             new Address("153281c93274bEB9726A03C33d3F19a8D78ad805"),
@@ -87,7 +87,7 @@ public IEnumerable<Transaction> Iterate(BlockChain blockChain, bool filtered = t
                         {
                             Log.Debug("Adding {signer} to banned accounts for 30 minutes", tx.Signer);
                             _bannedAccountTracker.Add(tx.Signer, DateTimeOffset.Now);
-                            _bannedAccounts = _bannedAccounts.Add(tx.Signer);
+                            BannedAccounts = BannedAccounts.Add(tx.Signer);
                         }
                     }
                 }
@@ -104,15 +104,15 @@ public IEnumerable<Transaction> Iterate(BlockChain blockChain, bool filtered = t
 
         public bool Stage(BlockChain blockChain, Transaction transaction)
         {
-            if (_bannedAccounts.Contains(transaction.Signer))
+            if (BannedAccounts.Contains(transaction.Signer))
             {
                 if (_bannedAccountTracker.ContainsKey(transaction.Signer))
                 {
                     if ((DateTimeOffset.Now - _bannedAccountTracker[transaction.Signer]).Minutes >= 30)
                     {
                         Log.Debug("Removing {signer} from banned accounts after 30 minutes", transaction.Signer);
                         _bannedAccountTracker.Remove(transaction.Signer);
-                        _bannedAccounts = _bannedAccounts.Remove(transaction.Signer);
+                        BannedAccounts = BannedAccounts.Remove(transaction.Signer);
                     }
                 }