adjust unit tests with the provider class changes

dansan566 · Oct 26, 2019 · 47a041a · 47a041a
1 parent c96321e
commit 47a041a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/calc/test_calculatePhase0.py b/src/calc/test_calculatePhase0.py
@@ -2,7 +2,7 @@
 
 from calc.calculate_phase0 import CalculatePhase0
 from model import reward_log
-from rpc.prpc_reward_api import PRpcRewardApiImpl
+from api.provider_factory import ProviderFactory
 
 BAKING_ADDRESS = "tz1Z1tMai15JWUWeN2PKL9faXXVPMuWamzJj"
 
@@ -14,7 +14,7 @@ def test_calculate(self):
         nw = {'NAME': 'MAINNET', 'NB_FREEZE_CYCLE': 5, 'BLOCK_TIME_IN_SEC': 60, 'BLOCKS_PER_CYCLE': 4096,
                 'BLOCKS_PER_ROLL_SNAPSHOT': 256}
 
-        api = PRpcRewardApiImpl(nw,  BAKING_ADDRESS, "mainnet.tezrpc.me")
+        api = ProviderFactory(provider='prpc').newRewardApi(nw,  BAKING_ADDRESS, '')
         model = api.get_rewards_for_cycle_map(153)
 
         phase0 = CalculatePhase0(model)