Sources availability

scrapper/data_scrapper.py

@@ -5,4 +5,7 @@
 
 class DataScrapper:
     def __init__(self, source) -> None:
-        self.source:str = {section: dict(config.items(section)) for section in config.sections()}['URLS'][source.lower()]
+        self.source:str = {section: dict(config.items(section)) for section in config.sections()}['URLS'][source.lower()]
+
+    def get_scrapper_sources(self):
+        return {section: dict(config.items(section)) for section in config.sections()}['URLS']

scrapper/supreme_court_united_states.py

@@ -21,6 +21,13 @@ def get_data(self, output_path:str, format:str='json', save_data_on_file:bool=Fa
 
         return 'Data was saved! on {}'.format(output_path) if save_data_on_file else response
 
+    def get_areas(self):
+        completed_url:str = self.source
+        headers = {
+          'Content-Type': 'application/json'
+        }   
+        return requests.request("GET", completed_url, headers=headers, data={}).json()      
+
     def __save_data(self, file_name:str, data,format:str, output_path:str):
       if format == 'json':
         formatted_output_path:str = '{}{}.{}'.format(output_path, file_name, format)

tests/data_scrapper_test.py

@@ -1,3 +1,4 @@
+from typing import List
 import unittest
 from scrapper import DataScrapper
 import configparser
@@ -12,6 +13,13 @@ class DataScrapperTestCase(unittest.TestCase):
     def setUp(self):
         self.DataScrapper = DataScrapper(source)
 
+    def test_get_scrapper_sources_counter(self):
+        self.assertEqual(len(self.DataScrapper.get_scrapper_sources()),2)
+
+    def test_get_scrapper_sources_options(self):
+        options:List = self.DataScrapper.get_scrapper_sources()
+        self.assertEqual([key for key in options.keys()][0],source.lower())
+
     def test_get_valid_url(self):
         object_source:str = self.DataScrapper.source
         valid_url:str = config['URLS'][source.lower()]

tests/supreme_court_united_states_test.py

@@ -16,6 +16,10 @@ class SupremeCourtUnitedStatesTestCase(unittest.TestCase):
     def setUp(self):
         self.SupremeCourtUnitedStates = SupremeCourtUnitedStates(source,area)
 
+    def test_get_areas(self):
+        options = self.SupremeCourtUnitedStates.get_areas()
+        self.assertGreater(len([key for key in options.keys()]),1)
+
     def test_get_first_element(self):
         result = self.SupremeCourtUnitedStates.get_data(output_path,'json',False)
         self.assertEqual(result['results'][0]['id'], 'scotus')