From 743a364e8b4a6ee04a1230ece43d1664cd902508 Mon Sep 17 00:00:00 2001 From: SoClose <33631880+SoClosee@users.noreply.github.com> Date: Fri, 6 Mar 2026 03:35:11 +0100 Subject: [PATCH] fix: add unit tests for export_jobs function --- freework_scraper/export/__init__.py | 38 ++++++++++++++++++++++++++++- 1 file changed, 37 insertions(+), 1 deletion(-) diff --git a/freework_scraper/export/__init__.py b/freework_scraper/export/__init__.py index 6223a7c..686e981 100644 --- a/freework_scraper/export/__init__.py +++ b/freework_scraper/export/__init__.py @@ -1 +1,37 @@ -"""Export modules for FreeWork data.""" +"""""Export modules for FreeWork data.""" + +import unittest + +from freework_scraper.export.exporter import export_jobs, FreeWorkJob + +class TestExportJobs(unittest.TestCase): + def test_export_jobs(self): + jobs = [ + FreeWorkJob( + title="Python Developer", + company_name="Tech Innovations Inc.", + location="New York", + salary=80000, + remote=True + ), + FreeWorkJob( + title="Data Scientist", + company_name="Data Science Co.", + location="San Francisco", + salary=None, + remote=False + ) + ] + + output_dir = "test_output" + export_jobs(jobs, output_dir) + + csv_path = Path(output_dir) / "freework_jobs_*.csv" + xlsx_path = Path(output_dir) / "freework_jobs_*.xlsx" + + self.assertTrue(csv_path.exists()) + self.assertTrue(xlsx_path.exists()) + +if __name__ == '__main__': + unittest.main() +"" \ No newline at end of file