Added lxml dependency and covid_test.py to demo how to work with xml responses

Gaurav Singh · Gaurav Singh · commit 657cf980ad11 · 2020-12-14T08:47:24.000+05:30
diff --git a/Pipfile b/Pipfile
@@ -9,6 +9,7 @@ verify_ssl = true
 requests = "*"
 pytest = "*"
 assertpy = "*"
+lxml = "*"
 
 [requires]
 python_version = "3.8"
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/config.py b/config.py
@@ -1 +1,2 @@
-BASE_URI = 'http://0.0.0.0:5000/api/people'
+BASE_URI = 'http://0.0.0.0:5000/api/people'
+COVID_TRACKER_HOST = 'http://127.0.0.1:5000'
diff --git a/tests/covid_test.py b/tests/covid_test.py
@@ -0,0 +1,17 @@
+import requests
+from assertpy import assert_that
+from lxml import etree
+
+from config import COVID_TRACKER_HOST
+from utils.print_helpers import pretty_print
+
+
+def test_covid_cases_have_crossed_a_million():
+    response = requests.get(f'{COVID_TRACKER_HOST}/api/v1/summary/latest')
+    pretty_print(response.headers)
+
+    response_xml = response.text
+    tree = etree.fromstring(bytes(response_xml, encoding='utf8'))
+    total_cases = tree.xpath("//data/summary/total_cases")[0].text
+
+    assert_that(int(total_cases)).is_greater_than(1000000)
