wip. test_predictoor_agent1.py is passing on a basic test

pdr_backend/predictoor/approach1/predictoor_agent1.py

@@ -1,9 +1,11 @@
 import time
-from typing import Dict
+from typing import Dict, Tuple
 
 from enforce_typing import enforce_types
 
-from pdr_backend.predictoor.predictoor_config import PredictoorConfig
+from pdr_backend.models.feed import Feed
+from pdr_backend.predictoor.approach1.predictoor_config1 import \
+    PredictoorConfig1
 
 
 @enforce_types
@@ -15,10 +17,13 @@ class PredictoorAgent1:
     - When a value can be predicted, call predict.py::predict_function()
     """
 
-    def __init__(self, config: PredictoorConfig):
-        self.config = config 
-        self.feeds = self.config.get_feeds() # [addr] : feed
-        self.contracts = self.config.get_contracts() # [addr] : contract
+    def __init__(self, config: PredictoorConfig1):
+        self.config = config
+
+        self.feeds: Dict[str,Feed] = self.config.get_feeds() # [addr] : Feed
+
+        feed_addrs = list(self.feeds.keys())
+        self.contracts = self.config.get_contracts(feed_addrs) # [addr] : contract
 
         self.prev_block_time: int = 0
         self.prev_block_number: int = 0

pdr_backend/predictoor/approach1/test/test_predictoor_agent1.py

@@ -1,6 +1,70 @@
-#from pdr_backend.approach1.predictoor_agent1 import PredictoorAgent1
+from unittest.mock import patch, Mock
 
-def test1():
-    pass
+from enforce_typing import enforce_types
+
+from pdr_backend.predictoor.approach1.predictoor_config1 import \
+    PredictoorConfig1
+from pdr_backend.predictoor.approach1.predictoor_agent1 import \
+    PredictoorAgent1
+
+PRIV_KEY = "0xef4b441145c1d0f3b4bc6d61d29f5c6e502359481152f869247c7a4244d45209"
+
+ADDR = "0xe8933f2950aec1080efad1ca160a6bb641ad245d"  # predictoor contract addr
+
+FEED_DICT = {  # info inside a predictoor contract
+    "name": "Contract Name",
+    "address": ADDR,
+    "symbol": "test",
+    "seconds_per_epoch": 300,
+    "seconds_per_subscription": 60,
+    "trueval_submit_timeout": 15,
+    "owner": "0xowner",
+    "pair": "BTC-ETH",
+    "timeframe": "1h",
+    "source": "binance",
+}
+
+
+def test_predictoor_agent1(monkeypatch):
+    _setenvs(monkeypatch)
+
+    # mock query_feed_contracts()
+    def _mock_query_feed_contracts(*args, **kwargs):  # pylint: disable=unused-argument
+        feed_dicts = {ADDR: FEED_DICT}
+        return feed_dicts
+    monkeypatch.setattr(
+        "pdr_backend.models.base_config.query_feed_contracts",
+        _mock_query_feed_contracts,
+    )
+
+    # mock PredictoorContract
+    def _mock_contract(*args, **kwarg):  # pylint: disable=unused-argument
+        m = Mock()
+        m.contract_address = ADDR
+        return m
+    monkeypatch.setattr(
+        "pdr_backend.models.base_config.PredictoorContract",
+        _mock_contract
+    )
+
+    # now do work
+    c = PredictoorConfig1()
+    agent = PredictoorAgent1(c)
+
+
+def _setenvs(monkeypatch):
+    #envvars handled by PredictoorConfig1
+    monkeypatch.setenv("SECONDS_TILL_EPOCH_END", "60")
+    monkeypatch.setenv("STAKE_AMOUNT", "30000")
+
+    #envvars handled by BaseConfig
+    monkeypatch.setenv("RPC_URL", "http://foo")
+    monkeypatch.setenv("SUBGRAPH_URL", "http://bar")
+    monkeypatch.setenv("PRIVATE_KEY", PRIV_KEY)
+
+    monkeypatch.setenv("PAIR_FILTER", "BTC/USDT,ETH/USDT")
+    monkeypatch.setenv("TIMEFRAME_FILTER", "5m,15m")
+    monkeypatch.setenv("SOURCE_FILTER", "binance,kraken")
+    monkeypatch.setenv("OWNER_ADDRS", "0x123,0x124")