mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-13 17:23:02 -05:00
609 lines
21 KiB
C#
609 lines
21 KiB
C#
//
|
|
// Copyright (c) Microsoft. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
//
|
|
|
|
using Microsoft.SqlTools.ServiceLayer.QueryExecution.DataStorage;
|
|
using Microsoft.SqlTools.ServiceLayer.Test.Common;
|
|
using Moq;
|
|
using System;
|
|
using System.IO;
|
|
using System.IO.Compression;
|
|
using System.Text.RegularExpressions;
|
|
using System.Xml;
|
|
using Microsoft.SqlTools.ServiceLayer.QueryExecution.Contracts;
|
|
using NUnit.Framework;
|
|
|
|
namespace Microsoft.SqlTools.ServiceLayer.UnitTests.QueryExecution.DataStorage
|
|
{
|
|
public partial class SaveAsExcelFileStreamWriterHelperTests
|
|
{
|
|
[Test]
|
|
public void DefaultSheet()
|
|
{
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
ContentMatch(stream, "[Content_Types].xml");
|
|
ContentMatch(stream, "_rels/.rels");
|
|
ContentMatch(stream, "xl/_rels/workbook.xml.rels");
|
|
ContentMatch(stream, "xl/styles.xml");
|
|
ContentMatch(stream, "xl/workbook.xml");
|
|
ContentMatch(stream, "xl/worksheets/sheet1.xml");
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWithFrozenHeaderRow()
|
|
{
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
sheet.FreezeHeaderRow();
|
|
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
ContentMatch(stream, "[Content_Types].xml");
|
|
ContentMatch(stream, "_rels/.rels");
|
|
ContentMatch(stream, "xl/_rels/workbook.xml.rels");
|
|
ContentMatch(stream, "xl/styles.xml");
|
|
ContentMatch(stream, "xl/workbook.xml");
|
|
ContentMatch(stream, "xl/worksheets/sheet1.xml", "xl/worksheets/sheet1-headerRowFrozen.xml");
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWithFrozenHeaderRowCalledTooLate()
|
|
{
|
|
var expectedException = new InvalidOperationException("Must be called before calling AddRow");
|
|
var actualException = new Exception();
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
sheet.AddRow();
|
|
|
|
try
|
|
{
|
|
sheet.FreezeHeaderRow();
|
|
|
|
Assert.Fail("Did not throw an exception when calling FreezeHeaderRow too late");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
actualException = e;
|
|
}
|
|
}
|
|
|
|
Assert.AreEqual(expectedException.GetType(), actualException.GetType());
|
|
Assert.AreEqual(expectedException.Message, actualException.Message);
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWithBoldHeaderRow()
|
|
{
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
ContentMatch(stream, "[Content_Types].xml");
|
|
ContentMatch(stream, "_rels/.rels");
|
|
ContentMatch(stream, "xl/_rels/workbook.xml.rels");
|
|
ContentMatch(stream, "xl/styles.xml");
|
|
ContentMatch(stream, "xl/workbook.xml");
|
|
ContentMatch(stream, "xl/worksheets/sheet1.xml", "xl/worksheets/sheet1-headerRowBold.xml");
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWithAutoFilterEnabled()
|
|
{
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
sheet.EnableAutoFilter();
|
|
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
ContentMatch(stream, "[Content_Types].xml");
|
|
ContentMatch(stream, "_rels/.rels");
|
|
ContentMatch(stream, "xl/_rels/workbook.xml.rels");
|
|
ContentMatch(stream, "xl/styles.xml");
|
|
ContentMatch(stream, "xl/workbook.xml");
|
|
ContentMatch(stream, "xl/worksheets/sheet1.xml", "xl/worksheets/sheet1-autoFilterEnabled.xml");
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWriteColumnInformationCalledTooLate()
|
|
{
|
|
var expectedException = new InvalidOperationException("Must be called before calling AddRow");
|
|
var actualException = new Exception();
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
sheet.AddRow();
|
|
|
|
try
|
|
{
|
|
sheet.WriteColumnInformation(new []{ 1F, 2F, 3F, 4F, 5F, 6F, 7F });
|
|
|
|
Assert.Fail("Did not throw an exception when calling WriteColumnInformation too late");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
actualException = e;
|
|
}
|
|
}
|
|
|
|
Assert.AreEqual(expectedException.GetType(), actualException.GetType());
|
|
Assert.AreEqual(expectedException.Message, actualException.Message);
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWriteColumnInformationWithWrongAmount()
|
|
{
|
|
var expectedException = new InvalidOperationException("Column count mismatch");
|
|
var actualException = new Exception();
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
try
|
|
{
|
|
sheet.WriteColumnInformation(new[] { 1F, 2F, 3F, 4F, 5F });
|
|
|
|
Assert.Fail("Did not throw an exception when calling WriteColumnInformation with the wrong number of columns");
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
actualException = e;
|
|
}
|
|
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
Assert.AreEqual(expectedException.GetType(), actualException.GetType());
|
|
Assert.AreEqual(expectedException.Message, actualException.Message);
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[Test]
|
|
public void SheetWriteColumnInformation()
|
|
{
|
|
var stream = new MemoryStream();
|
|
using (var helper = new SaveAsExcelFileStreamWriterHelper(stream, true))
|
|
using (var sheet = helper.AddSheet(null, 7))
|
|
{
|
|
sheet.WriteColumnInformation(new[] { 1F, 2F, 3F, 4F, 5F, 6F, 7F });
|
|
|
|
var value = new DbCellValue();
|
|
sheet.AddRow();
|
|
|
|
value.IsNull = true;
|
|
sheet.AddCell(value);
|
|
|
|
value.IsNull = false;
|
|
value.RawObject = "";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = "test string";
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = 3.5;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = false;
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = true;
|
|
sheet.AddCell(value);
|
|
|
|
sheet.AddRow();
|
|
|
|
value.RawObject = new DateTime(1900, 2, 28);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1900, 3, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new DateTime(1, 1, 1, 15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(15, 00, 00);
|
|
sheet.AddCell(value);
|
|
|
|
value.RawObject = new TimeSpan(24, 00, 00);
|
|
sheet.AddCell(value);
|
|
}
|
|
|
|
ContentMatch(stream, "[Content_Types].xml");
|
|
ContentMatch(stream, "_rels/.rels");
|
|
ContentMatch(stream, "xl/_rels/workbook.xml.rels");
|
|
ContentMatch(stream, "xl/styles.xml");
|
|
ContentMatch(stream, "xl/workbook.xml");
|
|
ContentMatch(stream, "xl/worksheets/sheet1.xml", "xl/worksheets/sheet1-withColumns.xml");
|
|
|
|
stream.Dispose();
|
|
}
|
|
|
|
[GeneratedRegex("\\r?\\n\\s*")]
|
|
private static partial Regex GetContentRemoveLinebreakLeadingSpaceRegex();
|
|
|
|
private void ContentMatch(Stream stream, string fileName)
|
|
{
|
|
ContentMatch(stream, fileName, fileName);
|
|
}
|
|
|
|
private void ContentMatch(Stream stream, string realFileName, string referenceFileName)
|
|
{
|
|
string referencePath = Path.Combine(RunEnvironmentInfo.GetTestDataLocation(),
|
|
"DataStorage",
|
|
"SaveAsExcelFileStreamWriterHelperTests",
|
|
referenceFileName);
|
|
string referenceContent = File.ReadAllText(referencePath);
|
|
referenceContent = GetContentRemoveLinebreakLeadingSpaceRegex().Replace(referenceContent, "");
|
|
|
|
using (var zip = new ZipArchive(stream, ZipArchiveMode.Read, true))
|
|
{
|
|
using (var reader = new StreamReader(zip?.GetEntry(realFileName)?.Open()!))
|
|
{
|
|
string realContent = reader.ReadToEnd();
|
|
Assert.AreEqual(referenceContent, realContent);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public class SaveAsExcelFileStreamWriterHelperReferenceManagerTests
|
|
{
|
|
private Mock<XmlWriter> _xmlWriterMock;
|
|
private string? LastWrittenReference { get; set; }
|
|
private int LastWrittenRow { get; set; }
|
|
|
|
public SaveAsExcelFileStreamWriterHelperReferenceManagerTests()
|
|
{
|
|
_xmlWriterMock = new Mock<XmlWriter>(MockBehavior.Strict);
|
|
_xmlWriterMock
|
|
.Setup(_xmlWriter => _xmlWriter.WriteChars(
|
|
It.IsAny<char[]>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Callback<char[], int, int>((array, index, count) =>
|
|
LastWrittenReference = new string(array, index, count));
|
|
_xmlWriterMock.Setup(a => a.WriteStartAttribute(null, "r", null));
|
|
_xmlWriterMock.Setup(a => a.WriteEndAttribute());
|
|
_xmlWriterMock.Setup(a => a.WriteValue(It.IsAny<int>()))
|
|
.Callback<int>(row => LastWrittenRow = row);
|
|
}
|
|
|
|
[Test]
|
|
public void ReferenceA1()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
manager.WriteAndIncreaseRowReference();
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("A1", LastWrittenReference);
|
|
}
|
|
[Test]
|
|
public void ReferenceZ1()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
manager.WriteAndIncreaseRowReference();
|
|
for (int i = 0; i < 26 - 1; i++)
|
|
{
|
|
manager.IncreaseColumnReference();
|
|
}
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("Z1", LastWrittenReference);
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("AA1", LastWrittenReference);
|
|
}
|
|
[Test]
|
|
public void ReferenceZZ1()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
manager.WriteAndIncreaseRowReference();
|
|
|
|
for (int i = 0; i < 27 * 26 - 1; i++)
|
|
{
|
|
manager.IncreaseColumnReference();
|
|
}
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("ZZ1", LastWrittenReference);
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("AAA1", LastWrittenReference);
|
|
}
|
|
[Test]
|
|
public void ReferenceXFD()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
manager.WriteAndIncreaseRowReference();
|
|
|
|
for (int i = 0; i < 16384 - 1; i++)
|
|
{
|
|
manager.IncreaseColumnReference();
|
|
}
|
|
//The 16384 should be the maximal column and not throw
|
|
manager.AssureColumnReference();
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("XFD1", LastWrittenReference);
|
|
var ex = Assert.Throws<InvalidOperationException>(manager.AssureColumnReference);
|
|
Assert.That(ex.Message, Does.Contain("max column number is 16384"));
|
|
}
|
|
[Test]
|
|
public void ReferenceRowReset()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
manager.WriteAndIncreaseRowReference();
|
|
Assert.AreEqual(1, LastWrittenRow);
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("A1", LastWrittenReference);
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("B1", LastWrittenReference);
|
|
|
|
//add row should reset column reference
|
|
manager.WriteAndIncreaseRowReference();
|
|
Assert.AreEqual(2, LastWrittenRow);
|
|
manager.WriteAndIncreaseColumnReference();
|
|
Assert.AreEqual("A2", LastWrittenReference);
|
|
}
|
|
|
|
[Test]
|
|
public void AddRowMustBeCalledBeforeAddCellException()
|
|
{
|
|
var xmlWriter = _xmlWriterMock.Object;
|
|
var manager = new SaveAsExcelFileStreamWriterHelper.ReferenceManager(xmlWriter);
|
|
|
|
var ex = Assert.Throws<InvalidOperationException>(manager.AssureColumnReference);
|
|
Assert.That(ex.Message, Does.Contain("AddRow must be called before AddCell"));
|
|
}
|
|
|
|
}
|
|
}
|