Remove XML exercises, update GraphQL exercise

Author: basdijkstra

.github/workflows/ci.yml

@@ -34,4 +34,7 @@ jobs:
         run: pytest ./answers/answers_02.py
 
       - name: Run Answers 03
-        run: pytest ./answers/answers_03.py
+        run: pytest ./answers/answers_03.py
+
+      - name: Run Answers 04
+        run: pytest ./answers/answers_04.py

answers/answers_04.py

@@ -1,91 +1,71 @@
-from lxml import etree
-
 import requests
+import pytest
 
 
 # Exercise 4.1
-# Create a function create_xml_body_from_string()
-# that returns a docstring (with triple double quotes)
-# containing the following XML document:
-# <payee>
-#     <name>John Smith</name>
-#     <address>
-#         <street>My street</street>
-#         <city>My city</city>
-#         <state>My state</state>
-#         <zipCode>90210</zipCode>
-#     </address>
-#     <phoneNumber>0123456789</phoneNumber>
-#     <accountNumber>12345</accountNumber>
-# </payee>
-def create_xml_body_from_string():
-    return """
-    <payee>
-        <name>John Smith</name>
-        <address>
-            <street>My street</street>
-            <city>My city</city>
-            <state>My state</state>
-            <zipCode>90210</zipCode>
-        </address>
-        <phoneNumber>0123456789</phoneNumber>
-        <accountNumber>12345</accountNumber>
-    </payee>
-    """
-
+# Create a new GraphQL query as a String with value { company { name ceo coo } }
+# POST this object to https://spacex-production.up.railway.app/
+# Assert that the name of the CEO is Elon Musk
+# The name can be found using ['data']['company']['ceo']
+def test_get_company_data_check_ceo_should_be_elon_musk():
 
-# Exercise 4.2
-# Write a test that POSTs the object created in 4.1
-# to https://parabank.parasoft.com/parabank/services/bank/billpay?accountId=12345&amount=500
-# Set the request header 'Content-Type' to 'application/xml'
-# Then check that the response status code is 200
-# and that the value of the response header 'Content-Type' is also equal to 'application/xml'
-def test_send_xml_body_from_docstring_check_status_code_is_200_and_name_is_correct():
     response = requests.post(
-        "https://parabank.parasoft.com/parabank/services/bank/billpay?accountId=12345&amount=500",
-        headers={"Content-Type": "application/xml"},
-        data=create_xml_body_from_string(),
+        'https://spacex-production.up.railway.app/',
+        json={'query': '{ company { name ceo coo } }'}
     )
-    assert response.status_code == 200
-    assert response.headers["Content-Type"] == "application/xml"
+    response_body = response.json()
+    assert response_body['data']['company']['ceo'] == 'Elon Musk'
+
+
+# Exercise 4.2
+# Create a test data source (a list of test data tuples)
+# containing the following test data:
+# --------------------------------------------------------------------------
+# rocket id                | rocket name  | country
+# --------------------------------------------------------------------------
+# 5e9d0d95eda69955f709d1eb | Falcon 1     | Republic of the Marshall Islands
+# 5e9d0d95eda69974db09d1ed | Falcon Heavy | United States
+# 5e9d0d96eda699382d09d1ee | Starship     | United States
+test_data_rockets = [
+    ('5e9d0d95eda69955f709d1eb', 'Falcon 1', 'Republic of the Marshall Islands'),
+    ('5e9d0d95eda69974db09d1ed', 'Falcon Heavy', 'United States'),
+    ('5e9d0d96eda699382d09d1ee', 'Starship', 'United States')
+]
 
 
 # Exercise 4.3
-# Write a method create_xml_body_using_elementtree() that returns
-# the same request body as in Exercise 4.1, but now uses the
-# etree library from lxml (I've imported that for you already, it's available as 'etree')
-# Make your life a little easier by specifying all element values as strings
-def create_xml_body_using_elementtree():
-    payee = etree.Element("payee")
-    name = etree.SubElement(payee, "name")
-    name.text = "John Smith"
-    address = etree.SubElement(payee, "address")
-    street = etree.SubElement(address, "street")
-    street.text = "My street"
-    city = etree.SubElement(address, "city")
-    city.text = "My city"
-    state = etree.SubElement(address, "state")
-    state.text = "My state"
-    zip_code = etree.SubElement(address, "zipCode")
-    zip_code.text = "90210"
-    phone_number = etree.SubElement(payee, "phoneNumber")
-    phone_number.text = "0123456789"
-    account_number = etree.SubElement(payee, "accountNumber")
-    account_number.text = "12345"
-    return payee
+# Write a test that POSTs the given parameterized GraphQL query to
+# https://spacex-production.up.railway.app/, together with the rocket id as
+# the value for the id variable, for all test cases in the test data source.
+#
+# Assert that the name of the rocket is equal to the value in the data source
+# Use ['data']['rocket']['name'] to extract it from the JSON response body.
+#
+# Assert that the country where the rocket was launched is equal to the value in the data source
+# Use ['data']['rocket']['country'] to extract it from the JSON response body.
+query_rocket_parameterized = """
+query getRocketData($id: ID!)
+{
+  rocket(id: $id) {
+    name
+    country
+  }
+}
+"""
+
 
+@pytest.mark.parametrize('rocket_id, rocket_name, country', test_data_rockets)
+def test_get_rocket_data_check_name_and_country_should_equal_expected(rocket_id, rocket_name, country):
 
-# Exercise 4.4
-# Repeat Exercise 4.2, but now use the XML document created in Exercise 4.3
-# Don't forget to convert the XML document to a string before sending it!
-def test_send_xml_body_from_elementtree_check_status_code_is_200_and_name_is_correct():
-    xml = create_xml_body_using_elementtree()
-    xml_as_string = etree.tostring(xml)
     response = requests.post(
-        "https://parabank.parasoft.com/parabank/services/bank/billpay?accountId=12345&amount=500",
-        headers={"Content-Type": "application/xml"},
-        data=xml_as_string,
+        'https://spacex-production.up.railway.app/',
+        json={
+            'query': query_rocket_parameterized,
+            'variables': {
+                'id': rocket_id
+            }
+        }
     )
-    print(response.request.body)
-    assert response.status_code == 200
-    assert response.headers["Content-Type"] == "application/xml"
+    response_body = response.json()
+    assert response_body['data']['rocket']['name'] == rocket_name
+    assert response_body['data']['rocket']['country'] == country