def test_should_process_row_within_cuttoff_date(self):
"""Test that we correctly determine a row should be processed."""
today = self.date_accessor.today_with_timezone('UTC')
row = {'lineItem/UsageStartDate': today.isoformat()}
processor = AWSReportProcessor(
schema_name=self.schema,
report_path=self.test_report,
compression=UNCOMPRESSED,
provider_uuid=self.aws_provider_uuid,
manifest_id=self.manifest.id,
)
should_process = processor._should_process_row(row, 'lineItem/UsageStartDate', False)
self.assertTrue(should_process)
def test_should_process_is_full_month(self):
"""Test that we correctly determine a row should be processed."""
today = self.date_accessor.today_with_timezone('UTC')
usage_start = today - relativedelta(days=10)
row = {'lineItem/UsageStartDate': usage_start.isoformat()}
processor = AWSReportProcessor(
schema_name=self.schema,
report_path=self.test_report,
compression=UNCOMPRESSED,
provider_uuid=self.aws_provider_uuid,
manifest_id=self.manifest.id,
)
should_process = processor._should_process_row(row, 'lineItem/UsageStartDate', True)
self.assertTrue(should_process)
def test_should_process_row_outside_cuttoff_date(self):
"""Test that we correctly determine a row should be processed."""
today = self.date_accessor.today_with_timezone("UTC")
usage_start = today - relativedelta(days=10)
row = {"lineItem/UsageStartDate": usage_start.isoformat()}
processor = AWSReportProcessor(
schema_name=self.schema,
report_path=self.test_report,
compression=UNCOMPRESSED,
provider_uuid=self.aws_provider_uuid,
manifest_id=self.manifest.id,
)
should_process = processor._should_process_row(
row, "lineItem/UsageStartDate", False)
self.assertFalse(should_process)
