fix tests

Author: SamR1

twootfeed/tests/test_default_route.py

@@ -1,6 +1,13 @@
+from unittest.mock import patch
+
 from flask import Flask
 
+from .utils import MastodonApi, Tweepy, TwitterApi
+
 
+@patch('twootfeed.twitter.generate_tweets_feed.tweepy', Tweepy([]))
+@patch('twootfeed.twitter.routes.twitter_api', TwitterApi())
+@patch('twootfeed.mastodon.routes.mastodon_api', MastodonApi([]))
 def test_default_route(app: Flask) -> None:
     client = app.test_client()
     response = client.get('/')

twootfeed/tests/test_mastodon_routes.py

@@ -1,10 +1,11 @@
+from unittest.mock import patch
 from uuid import uuid4
 
 import pytest
 from flask import Flask
 
 from .data import TEST_TOKEN
-from .test_twitter_routes import TestTwitterRoutes
+from .utils import MastodonApi
 
 MASTODON_ENDPOINTS = [
     '/toots/favorites',
@@ -13,6 +14,7 @@
 ]
 
 
+@patch('twootfeed.mastodon.routes.mastodon_api', MastodonApi([]))
 class TestMastodonTagsRoute:
     endpoint = '/toots/tags/{tag}'
 
@@ -52,10 +54,54 @@ def test_it_returns_200_when_tag_is_provided(self, app: Flask) -> None:
         assert '<rss version="2.0">' in data
 
 
-class TestMastodonSearchRoute(TestTwitterRoutes):
+@patch('twootfeed.mastodon.routes.mastodon_api', MastodonApi([]))
+class TestMastodonSearchRoute:
     endpoint = '/toots/search'
 
+    def test_it_returns_401_when_token_is_missing(
+        self, app_missing_token: Flask
+    ) -> None:
+        client = app_missing_token.test_client()
+
+        response = client.get(self.endpoint)
+
+        assert response.status_code == 401
+        data = response.data.decode()
+        assert data == 'missing token'
+
+    def test_it_returns_403_when_token_is_invalid(
+        self, app_invalid_token: Flask
+    ) -> None:
+        client = app_invalid_token.test_client()
+
+        response = client.get(f'{self.endpoint}?token=invalid')
+
+        assert response.status_code == 403
+        data = response.data.decode()
+        assert data == 'invalid token'
+
+    def test_it_returns_400_when_query_is_missing(self, app: Flask) -> None:
+        client = app.test_client()
+
+        response = client.get(f'{self.endpoint}?token={TEST_TOKEN}')
+
+        assert response.status_code == 400
+        data = response.data.decode()
+        assert data == 'missing query'
+
+    def test_it_returns_200_when_query_is_provided(self, app: Flask) -> None:
+        client = app.test_client()
+
+        response = client.get(
+            f'{self.endpoint}?q={uuid4().hex}&token={TEST_TOKEN}'
+        )
+
+        assert response.status_code == 200
+        data = response.data.decode()
+        assert '<rss version="2.0">' in data
+
 
+@patch('twootfeed.mastodon.routes.mastodon_api', MastodonApi([]))
 class TestMastodonRoutes:
     @pytest.mark.parametrize('input_endpoint', MASTODON_ENDPOINTS)
     def test_it_returns_401_when_token_is_missing(

twootfeed/tests/test_twitter_routes.py

@@ -1,10 +1,14 @@
+from unittest.mock import patch
 from uuid import uuid4
 
 from flask import Flask
 
 from .data import TEST_TOKEN
+from .utils import Tweepy, TwitterApi
 
 
+@patch('twootfeed.twitter.generate_tweets_feed.tweepy', Tweepy([]))
+@patch('twootfeed.twitter.routes.twitter_api', TwitterApi())
 class TestTwitterRoutes:
     endpoint = '/tweets'
 
@@ -39,7 +43,7 @@ def test_it_returns_400_when_query_is_missing(self, app: Flask) -> None:
         data = response.data.decode()
         assert data == 'missing query'
 
-    def test_it_returns_200_when_query_valid(self, app: Flask) -> None:
+    def test_it_returns_200_when_query_is_provided(self, app: Flask) -> None:
         client = app.test_client()
 
         response = client.get(

twootfeed/tests/utils.py

@@ -50,6 +50,9 @@ def __init__(self, tweets: List):
         self.tweet_pages = self.items_in_pages(tweets)
         self.tweets = [ToDotNotationTweepy(tweet) for tweet in tweets]
 
+    def Cursor(self, *args: Any, **kwargs: Any) -> 'Tweepy':
+        return self
+
     def pages(self) -> List[List[ToDotNotationTweepy]]:
         return self.tweet_pages
 

twootfeed/utils/decorator.py

@@ -1,8 +1,7 @@
 from functools import wraps
 from typing import Any, Callable, Tuple, Union
 
-from flask import request
-from twootfeed import param
+from flask import current_app, request
 
 
 def require_token(f: Callable) -> Callable:
@@ -13,7 +12,7 @@ def decorated_function(
         token = request.args.get('token')
         if not token:
             return 'missing token', 401
-        if token != param['feed']['token']:
+        if token != current_app.config['FEED_CONFIG']['token']:
             return 'invalid token', 403
 
         return f(*args, **kwargs)