From f366c6c0596610094c0053bb609099bb358888a9 Mon Sep 17 00:00:00 2001
From: Dan Palmer <dan@danpalmer.me>
Date: Thu, 4 Jan 2018 14:12:00 +0000
Subject: [PATCH] Implement and test setup for feeds

---
 routemaster/feeds.py            |  2 +-
 routemaster/tests/test_feeds.py | 71 +++++++++++++++++++++++++++++++++
 2 files changed, 72 insertions(+), 1 deletion(-)
 create mode 100644 routemaster/tests/test_feeds.py

diff --git a/routemaster/feeds.py b/routemaster/feeds.py
index ae0d9bf4..49227d0a 100644
--- a/routemaster/feeds.py
+++ b/routemaster/feeds.py
@@ -8,7 +8,7 @@
 
 def feeds_for_state_machine(state_machine) -> Dict[str, 'Feed']:
     """Get a mapping of feed prefixes to unfetched feeds."""
-    pass
+    return {x.name: Feed(x.url) for x in state_machine.feeds}
 
 
 class FeedNotFetched(Exception):
diff --git a/routemaster/tests/test_feeds.py b/routemaster/tests/test_feeds.py
new file mode 100644
index 00000000..a92991a5
--- /dev/null
+++ b/routemaster/tests/test_feeds.py
@@ -0,0 +1,71 @@
+import mock
+import pytest
+import httpretty
+
+from routemaster.feeds import Feed, FeedNotFetched, feeds_for_state_machine
+from routemaster.config import Feed as FeedConfig
+from routemaster.config import Gate, NoNextStates, StateMachine
+from routemaster.exit_conditions import ExitConditionProgram
+
+
+def test_feeds_for_state_machine():
+    state_machine = StateMachine(
+        name='example',
+        feeds=[
+            FeedConfig(name='test_feed', url='http://localhost/<label>'),
+        ],
+        states=[
+            Gate(
+                name='start',
+                triggers=[],
+                next_states=NoNextStates(),
+                exit_condition=ExitConditionProgram('false'),
+            ),
+        ]
+    )
+
+    feeds = feeds_for_state_machine(state_machine)
+
+    assert 'test_feed' in feeds
+    assert feeds['test_feed'].data is None
+    assert feeds['test_feed'].url == 'http://localhost/<label>'
+
+
+@httpretty.activate
+def test_fetch_only_once():
+    httpretty.register_uri(
+        httpretty.GET,
+        'http://example.com/label1',
+        body='{"foo": "bar"}',
+        content_type='application/json',
+    )
+
+    feed = Feed('http://example.com/<label>')
+
+    with mock.patch('requests.Response.json') as json:
+        feed.fetch('label1')
+        feed.fetch('label1')
+        feed.fetch('label1')
+
+        assert json.call_count == 1
+
+
+@httpretty.activate
+def test_lookup():
+    httpretty.register_uri(
+        httpretty.GET,
+        'http://example.com/label1',
+        body='{"foo": "bar"}',
+        content_type='application/json',
+    )
+
+    feed = Feed('http://example.com/<label>')
+    feed.fetch('label1')
+
+    assert feed.lookup(('foo',)) == 'bar'
+
+
+def test_lookup_fails_on_unfetched_feed():
+    feed = Feed('http://example.com/<label>')
+    with pytest.raises(FeedNotFetched):
+        feed.lookup(('foo',))