From 44785255b04879edaeff43337c82d5c6ca73029a Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Mon, 15 Jun 2020 02:14:04 +0200
Subject: [PATCH] Add Flickr avalaibility test

---
 app/forms.py                |  4 ++++
 tests/preprocessing_test.py | 10 +++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/app/forms.py b/app/forms.py
index 6631a79..c6d735b 100644
--- a/app/forms.py
+++ b/app/forms.py
@@ -4,5 +4,9 @@ from wtforms import SelectField, SubmitField
 
 
 class DatasetForm(FlaskForm):
+    """
+    Web form to select a dataset
+    """
+
     dataset = SelectField(choices=CHOICES)
     submit = SubmitField("Submit")
diff --git a/tests/preprocessing_test.py b/tests/preprocessing_test.py
index 74d7886..7080220 100644
--- a/tests/preprocessing_test.py
+++ b/tests/preprocessing_test.py
@@ -3,7 +3,7 @@ from requests import get
 
 from app.preprocessing import create_dataframe
 from app.data_request import request_dataset
-from constants import COLUMNS, DATASETS, DATASET_URL
+from constants import COLUMNS, DATASETS, DATASET_URL, FLICKR_URL
 
 
 def test_dataset_request():
@@ -23,3 +23,11 @@ def test_dataframe_creation():
         df = create_dataframe(dataset)
         assert isinstance(df, DataFrame)
         assert all(df.columns == COLUMNS[dataset])
+
+
+def test_flickr_request():
+    """
+    Checks that Flickr search is avalaible
+    """
+    response = get(FLICKR_URL.format("paris coronavirus"))
+    assert response.status_code == 200