use test.utils.mark_live_test rather than unittest.skipIf(...)

Author: bmos

test/test_azure/test_azure_blob_storage.py

@@ -7,6 +7,7 @@
 
 from parsons import AzureBlobStorage, Table
 from parsons.utilities import files
+from test.utils import mark_live_test
 
 TEST_ACCOUNT_NAME = os.getenv("PARSONS_AZURE_ACCOUNT_NAME")
 TEST_CREDENTIAL = os.getenv("PARSONS_AZURE_CREDENTIAL")
@@ -15,7 +16,7 @@
 TEST_FILE_CONTENTS = "Test"
 
 
-@unittest.skipIf(not os.getenv("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestAzureBlobStorage(unittest.TestCase):
     def setUp(self):
         self.azure_blob = AzureBlobStorage(

test/test_box/test_box_storage.py

@@ -9,6 +9,7 @@
 from boxsdk.exception import BoxAPIException, BoxOAuthException
 
 from parsons import Box, Table
+from test.utils import mark_live_test
 
 """Prior to running, you should ensure that the relevant environment
 variables have been set, e.g. via
@@ -28,7 +29,7 @@ def generate_random_string(length):
     return "".join(random.choice(string.ascii_letters) for i in range(length))
 
 
-@unittest.skipIf(not os.getenv("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestBoxStorage(unittest.TestCase):
     def setUp(self) -> None:
         warnings.filterwarnings(action="ignore", message="unclosed", category=ResourceWarning)

test/test_civis.py

@@ -1,12 +1,12 @@
-import os
 import unittest
 
 from parsons import CivisClient, Table
+from test.utils import mark_live_test
 
 # from . import scratch_creds
 
 
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestCivisClient(unittest.TestCase):
     def setUp(self):
         self.civis = CivisClient()

test/test_databases/test_dbsync.py

@@ -10,7 +10,7 @@
 from parsons.databases.database_connector import DatabaseConnector
 from parsons.databases.sqlite import Sqlite
 from test.test_databases.fakes import FakeDatabase
-from test.utils import assert_matching_tables
+from test.utils import assert_matching_tables, mark_live_test
 
 _dir = os.path.dirname(__file__)
 
@@ -195,7 +195,7 @@ def initialize_db_connections(self) -> None:
 
 # These tests interact directly with the Postgres database. In order to run, set the
 # env to LIVE_TEST='TRUE'.
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestPostgresDBSync(TestDBSync):
     db = Postgres
     setup_sql = f"""
@@ -209,6 +209,6 @@ class TestPostgresDBSync(TestDBSync):
 
 # These tests interact directly with the Postgres database. In order to run, set the
 # env to LIVE_TEST='TRUE'.
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestRedshiftDBSync(TestPostgresDBSync):
     db = Redshift

test/test_databases/test_mysql.py

@@ -1,13 +1,12 @@
-import os
 import unittest
 
 from parsons import MySQL, Table
 from parsons.databases.mysql.create_table import MySQLCreateTable
-from test.utils import assert_matching_tables
+from test.utils import assert_matching_tables, mark_live_test
 
 
 # These tests interact directly with the MySQL database. To run, set env variable "LIVE_TEST=True"
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestMySQLLive(unittest.TestCase):
     def setUp(self):
         self.mysql = MySQL()
@@ -41,7 +40,7 @@ def test_insert_data(self):
 
 
 # These tests interact directly with the MySQL database. To run, set env variable "LIVE_TEST=True"
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestMySQL(unittest.TestCase):
     def setUp(self):
         self.mysql = MySQL()

test/test_databases/test_postgres.py

@@ -4,7 +4,7 @@
 import pytest
 
 from parsons import Postgres, Table
-from test.utils import assert_matching_tables
+from test.utils import assert_matching_tables, mark_live_test
 
 # The name of the schema and will be temporarily created for the tests
 TEMP_SCHEMA = "parsons_test"
@@ -149,7 +149,7 @@ def test_create_statement(self):
 # These tests interact directly with the Postgres database
 
 
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestPostgresDB(unittest.TestCase):
     def setUp(self):
         self.temp_schema = TEMP_SCHEMA

test/test_databases/test_redshift.py

@@ -6,7 +6,7 @@
 from testfixtures import LogCapture
 
 from parsons import S3, Redshift, Table
-from test.utils import assert_matching_tables, validate_list
+from test.utils import assert_matching_tables, mark_live_test, validate_list
 
 # The name of the schema and will be temporarily created for the tests
 TEMP_SCHEMA = "parsons_test2"
@@ -381,7 +381,7 @@ def test_copy_statement_columns(self):
 # These tests interact directly with the Redshift database
 
 
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestRedshiftDB(unittest.TestCase):
     def setUp(self):
         self.temp_schema = TEMP_SCHEMA

test/test_donorbox/test_donorbox.py

@@ -42,7 +42,6 @@ def test_get_campaigns(self, m):
         ]
         assert len(result.columns) == len(columns)
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_campaigns_live_test(self):
         result = self.donorbox.get_campaigns()
@@ -74,7 +73,6 @@ def test_get_campaigns_with_id_filter(self, m):
         assert result.to_dicts()[0]["id"] == 366590
         assert result.to_dicts()[0]["name"] == "Membership Campaign"
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_campaigns_with_id_filter_live_test(self):
         result = self.donorbox.get_campaigns(id=366590)
@@ -95,7 +93,6 @@ def test_get_campaigns_with_name_filter(self, m):
         assert result.to_dicts()[0]["id"] == 366590
         assert result.to_dicts()[0]["name"] == "Membership Campaign"
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_campaigns_with_name_filter_live_test(self):
         result = self.donorbox.get_campaigns(name="Membership Campaign")
@@ -120,7 +117,6 @@ def test_get_campaigns_with_order_filter(self, m):
         result = self.donorbox.get_campaigns(order="asc")
         assert result["id"] == [366172, 366590]
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_campaigns_with_order_filter_live_test(self):
         # check order of the ids without looking at IDs. or maybe look at updated/created date
@@ -182,7 +178,6 @@ def test_get_donations(self, m):
         ]
         assert len(result.columns) == len(columns)
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donations_live_test(self):
         result = self.donorbox.get_donations()
@@ -228,7 +223,6 @@ def test_get_donations_live_test(self):
         assert result.columns == columns
         assert result.num_rows == 3
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donations_with_date_from_filter_live_test(self):
         # Correct formats (YYYY-mm-dd YYYY/mm/dd YYYYmmdd dd-mm-YYYY) successfully filter
@@ -243,7 +237,6 @@ def test_get_donations_with_date_from_filter_live_test(self):
         with pytest.raises(ValueError):  # noqa: PT011
             self.donorbox.get_donations(date_from="10 20 2022")
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donations_with_date_to_filter_live_test(self):
         # Correct formats (YYYY-mm-dd YYYY/mm/dd YYYYmmdd dd-mm-YYYY) successfully filter
@@ -279,7 +272,6 @@ def test_get_donations_with_amount_min_filter(self, m):
         result = self.donorbox.get_donations(amount_min="5")
         assert result.num_rows == 0
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donations_with_amount_min_filter_live_test(self):
         result = self.donorbox.get_donations(amount_min="3")
@@ -310,7 +302,6 @@ def test_get_donations_with_amount_max_filter(self, m):
         result = self.donorbox.get_donations(amount_max="2")
         assert result.num_rows == 0
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donations_with_amount_max_filter_live_test(self):
         result = self.donorbox.get_donations(amount_max="3")
@@ -349,7 +340,6 @@ def test_get_donors(self, m):
         ]
         assert len(result.columns) == len(columns)
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_donors_live_test(self):
         result = self.donorbox.get_donors()
@@ -429,7 +419,6 @@ def test_get_plans(self, m):
         assert result.columns == columns
         assert result.num_rows == 3
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_plans_live_test(self):
         result = self.donorbox.get_plans()
@@ -450,7 +439,6 @@ def test_get_plans_live_test(self):
         assert result.columns == columns
         assert result.num_rows == 3
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_plans_with_date_from_filter_live_test(self):
         # Correct formats (YYYY-mm-dd YYYY/mm/dd YYYYmmdd dd-mm-YYYY) successfully filter
@@ -465,7 +453,6 @@ def test_get_plans_with_date_from_filter_live_test(self):
         with pytest.raises(ValueError):  # noqa: PT011
             result = self.donorbox.get_plans(date_from="10 20 2022")
 
-    @unittest.skip("requires live account setup")
     @mark_live_test
     def test_get_plans_with_date_to_filter_live_test(self):
         # Correct formats (YYYY-mm-dd YYYY/mm/dd YYYYmmdd dd-mm-YYYY) successfully filter

test/test_facebook_ads.py

@@ -1,9 +1,9 @@
-import os
 import unittest
 
 import pytest
 
 from parsons import FacebookAds, Table
+from test.utils import mark_live_test
 
 users_table = Table(
     [
@@ -27,7 +27,7 @@
 )
 
 
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestFacebookAdsIntegration(unittest.TestCase):
     def setUp(self):
         self.fb_ads = FacebookAds()

test/test_geocoder/test_census_geocoder.py

@@ -1,15 +1,14 @@
-import os
 import unittest
 from unittest import mock
 
 import petl
 from test_responses import batch_resp, coord_resp, geographies_resp, locations_resp
 
 from parsons import CensusGeocoder, Table
-from test.utils import assert_matching_tables
+from test.utils import assert_matching_tables, mark_live_test
 
 
-@unittest.skipIf(not os.environ.get("LIVE_TEST"), "Skipping because not running live test")
+@mark_live_test
 class TestCensusGeocoder(unittest.TestCase):
     def setUp(self):
         self.cg = CensusGeocoder()