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)