public void FiltersForLineItemType()
        {
            // Arrange
            var sw = new StringWriter();

            using (var csv = new CsvWriter(sw))
            {
                var lineItems = new[]
                {
                    new LineItem {
                        RecordType = "Dummy"
                    },
                    new LineItem {
                        RecordType = "LineItem", RecordId = "Hoojey"
                    }
                };
                csv.WriteHeader <LineItem>();
                csv.WriteRecords(lineItems);
            }

            // Act
            List <LineItem> parsedItems;

            using (var reader = new StringReader(sw.ToString()))
            {
                using (var parser = new LineItemCsvParser(reader))
                {
                    parsedItems = parser.GetLineItems().ToList();
                }
            }

            // Assert
            parsedItems.Count.Should().Be(1);
            parsedItems[0].RecordId.Should().Be("Hoojey");
        }
Ejemplo n.º 2
0
        private void RefreshDataForPeriod(AwsProfile profile, string period, IAwsClient client, DateTime currentTime)
        {
            string s3Url = MonthlyCsvUrl.Create(profile.DetailedBillingS3Bucket, profile.Account, period);

            var lastModified = new DateTime();

            if (profile.BillingMetadata.ContainsKey(period))
            {
                lastModified = profile.BillingMetadata[period].LastModified;
            }

            DateTime newLastModified;
            Stream   file = client.StorageService.GetFileIfChangedSince(s3Url, lastModified, out newLastModified);

            if (file == null)
            {
                return;
            }

            using (file)
            {
                using (Stream zipStream = OpenFirstZipEntry(file))
                {
                    using (var streamReader = new StreamReader(zipStream, Encoding.UTF8, detectEncodingFromByteOrderMarks: true, bufferSize: 32768, leaveOpen: true))
                    {
                        using (var parser = new LineItemCsvParser(streamReader))
                        {
                            _billingManager.LoadLineItems(parser.GetLineItems(), period);
                        }
                    }
                }
            }

            BillingMetadata metadata;

            if (!profile.BillingMetadata.ContainsKey(period))
            {
                metadata = new BillingMetadata();
                profile.BillingMetadata.Add(period, metadata);
            }
            else
            {
                metadata = profile.BillingMetadata[period];
            }
            metadata.LastModified = newLastModified;
            metadata.LastLoaded   = currentTime;
        }
 public void Parse_Ok()
 {
     using (var stream = new MemoryStream(Resources.LineItems))
     {
         using (var streamReader = new StreamReader(stream))
         {
             using (var parser = new LineItemCsvParser(streamReader))
             {
                 List <LineItem> items = parser.GetLineItems().Take(5).ToList();
                 items.Count.Should().Be(5);
                 foreach (LineItem lineItem in items)
                 {
                     lineItem.RecordType.Should().Be("LineItem");
                     lineItem.RecordId.Should().NotBeEmpty();
                     lineItem.ResourceId.Should().NotBeEmpty();
                     lineItem.UsageStartDate.Year.Should().BeGreaterOrEqualTo(2014);
                     lineItem.UsageEndDate.Year.Should().BeGreaterOrEqualTo(2014);
                 }
             }
         }
     }
 }