From 413cc873379fa15964575047e3c76f99439a8e54 Mon Sep 17 00:00:00 2001
From: Olivier Gayot <olivier.gayot@canonical.com>
Date: Wed, 20 Apr 2022 19:09:30 +0200
Subject: [PATCH] Revert "Move to aiohttp for GeoIP requests"

This reverts commit ebdf1db63619424d06f7597f0c358e72cae4eedc.
---
 apt-deps.txt                         |  1 -
 subiquity/server/geoip.py            | 14 ++++----
 subiquity/server/tests/test_geoip.py | 48 ++++++++++++++++------------
 3 files changed, 35 insertions(+), 28 deletions(-)

diff --git a/apt-deps.txt b/apt-deps.txt
index b019f991..06c7a1b5 100644
--- a/apt-deps.txt
+++ b/apt-deps.txt
@@ -15,7 +15,6 @@ os-prober
 pep8
 pkg-config
 python3-aiohttp
-python3-aioresponses
 python3-apport
 python3-async-timeout
 python3-attr
diff --git a/subiquity/server/geoip.py b/subiquity/server/geoip.py
index 1d38fc68..0f9eeef0 100644
--- a/subiquity/server/geoip.py
+++ b/subiquity/server/geoip.py
@@ -14,12 +14,13 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 from abc import ABC, abstractmethod
-import aiohttp
 import logging
 import enum
+import requests
 from xml.etree import ElementTree
 
 from subiquitycore.async_helpers import (
+    run_in_thread,
     SingleInstanceTask,
 )
 
@@ -75,14 +76,13 @@ class HTTPGeoIPStrategy(GeoIPStrategy):
     """ HTTP implementation to retrieve GeoIP information. We use the
     geoip.ubuntu.com service. """
     async def get_response(self) -> str:
-        url = "https://geoip.ubuntu.com/lookup"
         try:
-            async with aiohttp.ClientSession() as session:
-                async with session.get(url) as response:
-                    response.raise_for_status()
-                    return await response.text()
-        except aiohttp.ClientError as e:
+            response = await run_in_thread(
+                requests.get, "https://geoip.ubuntu.com/lookup")
+            response.raise_for_status()
+        except requests.exceptions.RequestException as e:
             raise LookupError from e
+        return response.text
 
 
 class GeoIP:
diff --git a/subiquity/server/tests/test_geoip.py b/subiquity/server/tests/test_geoip.py
index 835292e8..ae15e5f8 100644
--- a/subiquity/server/tests/test_geoip.py
+++ b/subiquity/server/tests/test_geoip.py
@@ -13,7 +13,7 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from aioresponses import aioresponses
+import mock
 
 from subiquitycore.tests import SubiTestCase
 from subiquitycore.tests.mocks import make_app
@@ -47,17 +47,30 @@ empty_tz = '<Response><TimeZone></TimeZone></Response>'
 empty_cc = '<Response><CountryCode></CountryCode></Response>'
 
 
+class MockGeoIPResponse:
+    def __init__(self, text, status_code=200):
+        self.text = text
+        self.status_code = status_code
+
+    def raise_for_status(self, *args, **kwargs):
+        pass
+
+
+def requests_get_factory(text):
+    def requests_get(*args, **kwargs):
+        return MockGeoIPResponse(text)
+    return requests_get
+
+
 class TestGeoIP(SubiTestCase):
+    @mock.patch('requests.get', new=requests_get_factory(xml))
     def setUp(self):
         strategy = HTTPGeoIPStrategy()
         self.geoip = GeoIP(make_app(), strategy)
 
         async def fn():
             self.assertTrue(await self.geoip.lookup())
-
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=xml)
-            run_coro(fn())
+        run_coro(fn())
 
     def test_countrycode(self):
         self.assertEqual("us", self.geoip.countrycode)
@@ -71,42 +84,37 @@ class TestGeoIPBadData(SubiTestCase):
         strategy = HTTPGeoIPStrategy()
         self.geoip = GeoIP(make_app(), strategy)
 
+    @mock.patch('requests.get', new=requests_get_factory(partial))
     def test_partial_reponse(self):
         async def fn():
             self.assertFalse(await self.geoip.lookup())
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=partial)
-            run_coro(fn())
+        run_coro(fn())
 
+    @mock.patch('requests.get', new=requests_get_factory(incomplete))
     def test_incomplete(self):
         async def fn():
             self.assertFalse(await self.geoip.lookup())
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=incomplete)
-            run_coro(fn())
+        run_coro(fn())
         self.assertIsNone(self.geoip.countrycode)
         self.assertIsNone(self.geoip.timezone)
 
+    @mock.patch('requests.get', new=requests_get_factory(long_cc))
     def test_long_cc(self):
         async def fn():
             self.assertFalse(await self.geoip.lookup())
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=long_cc)
-            run_coro(fn())
+        run_coro(fn())
         self.assertIsNone(self.geoip.countrycode)
 
+    @mock.patch('requests.get', new=requests_get_factory(empty_cc))
     def test_empty_cc(self):
         async def fn():
             self.assertFalse(await self.geoip.lookup())
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=empty_cc)
-            run_coro(fn())
+        run_coro(fn())
         self.assertIsNone(self.geoip.countrycode)
 
+    @mock.patch('requests.get', new=requests_get_factory(empty_tz))
     def test_empty_tz(self):
         async def fn():
             self.assertFalse(await self.geoip.lookup())
-        with aioresponses() as mocked:
-            mocked.get("https://geoip.ubuntu.com/lookup", body=empty_tz)
-            run_coro(fn())
+        run_coro(fn())
         self.assertIsNone(self.geoip.timezone)