tdf#113861: Unittest for chart datatable labels expansion

Change-Id: Icfa10b1e75c801a9a025c0173bd8a34e362f049e
Reviewed-on: https://gerrit.libreoffice.org/45670
Tested-by: Jenkins <ci@libreoffice.org>
Reviewed-by: Eike Rathke <erack@redhat.com>

2 files changed, 122 insertions, 0 deletions

diff --git a/sc/qa/unit/chart2dataprovider.cxx b/sc/qa/unit/chart2dataprovider.cxx
new file mode 100644
index 000000000000..d7fd25122797
--- /dev/null
+++ b/sc/qa/unit/chart2dataprovider.cxx
@@ -0,0 +1,122 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/*
+ * This file is part of the LibreOffice project.
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+#include <sal/config.h>
+#include <test/bootstrapfixture.hxx>
+
+#include <docsh.hxx>
+#include <chart2uno.hxx>
+
+#include <com/sun/star/frame/Desktop.hpp>
+#include <com/sun/star/chart2/data/XLabeledDataSequence2.hpp>
+#include <com/sun/star/chart2/data/XDataSource.hpp>
+#include <com/sun/star/chart2/data/XDataSink.hpp>
+
+#include "helper/qahelper.hxx"
+
+using namespace ::com::sun::star;
+using namespace ::com::sun::star::uno;
+
+class ScChart2DataProviderTest : public ScBootstrapFixture
+{
+public:
+    ScChart2DataProviderTest();
+
+    virtual void setUp() override;
+    virtual void tearDown() override;
+
+    void testHeaderExpansion();
+
+    CPPUNIT_TEST_SUITE(ScChart2DataProviderTest);
+    CPPUNIT_TEST(testHeaderExpansion);
+    CPPUNIT_TEST_SUITE_END();
+
+private:
+    uno::Reference<uno::XInterface> m_xCalcComponent;
+};
+
+void lcl_createAndCheckDataProvider(ScDocument& rDoc, const OUString& cellRange, bool hasCategories,
+                                    bool firstCellAsLabel, sal_Int32 expectedRows,
+                                    sal_Int32 expectedCols)
+{
+    uno::Reference<chart2::data::XDataProvider> xDataProvider = new ScChart2DataProvider(&rDoc);
+    CPPUNIT_ASSERT(xDataProvider.is());
+
+    uno::Sequence<beans::PropertyValue> aArgs(4);
+
+    aArgs[0].Name = "CellRangeRepresentation";
+    aArgs[0].Value <<= cellRange;
+
+    aArgs[1].Name = "HasCategories";
+    aArgs[1].Value <<= hasCategories;
+
+    aArgs[2].Name = "FirstCellAsLabel";
+    aArgs[2].Value <<= firstCellAsLabel;
+
+    aArgs[3].Name = "DataRowSource";
+    aArgs[3].Value <<= chart::ChartDataRowSource_COLUMNS;
+
+    uno::Reference<chart2::data::XDataSource> xDataSource = xDataProvider->createDataSource(aArgs);
+    CPPUNIT_ASSERT(xDataSource.is());
+
+    css::uno::Sequence<uno::Reference<chart2::data::XLabeledDataSequence>> xSequences
+        = xDataSource->getDataSequences();
+
+    CPPUNIT_ASSERT_EQUAL(expectedRows, xSequences.getLength());
+
+    for (sal_Int32 nIdx = 0; nIdx < xSequences.getLength(); ++nIdx)
+    {
+        Reference<chart2::data::XDataSequence> xValues(xSequences[nIdx]->getValues());
+        if (xValues.is())
+            CPPUNIT_ASSERT_EQUAL(expectedCols, xValues->getData().getLength());
+    }
+}
+
+void ScChart2DataProviderTest::testHeaderExpansion()
+{
+    ScDocShellRef xDocSh = loadDoc("chart2dataprovider.", FORMAT_ODS);
+    CPPUNIT_ASSERT_MESSAGE("Failed to load ch.ods.", xDocSh.is());
+
+    ScDocument& rDoc = xDocSh->GetDocument();
+
+    lcl_createAndCheckDataProvider(rDoc, "$Sheet1.$A$1:$D$4", false, false, 4, 4);
+    lcl_createAndCheckDataProvider(rDoc, "$Sheet1.$A$1:$D$4", true, true, 4, 3);
+
+    lcl_createAndCheckDataProvider(rDoc, "$Sheet1.$A$9:$D$12", true, true, 1, 2);
+
+    xDocSh->DoClose();
+}
+
+ScChart2DataProviderTest::ScChart2DataProviderTest()
+    : ScBootstrapFixture("sc/qa/unit/data")
+{
+}
+
+void ScChart2DataProviderTest::setUp()
+{
+    test::BootstrapFixture::setUp();
+
+    // This is a bit of a fudge, we do this to ensure that ScGlobals::ensure,
+    // which is a private symbol to us, gets called
+    m_xCalcComponent
+        = getMultiServiceFactory()->createInstance("com.sun.star.comp.Calc.SpreadsheetDocument");
+    CPPUNIT_ASSERT_MESSAGE("no calc component!", m_xCalcComponent.is());
+}
+
+void ScChart2DataProviderTest::tearDown()
+{
+    uno::Reference<lang::XComponent>(m_xCalcComponent, UNO_QUERY_THROW)->dispose();
+    test::BootstrapFixture::tearDown();
+}
+
+CPPUNIT_TEST_SUITE_REGISTRATION(ScChart2DataProviderTest);
+
+CPPUNIT_PLUGIN_IMPLEMENT();
+
+/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
diff --git a/sc/qa/unit/data/ods/chart2dataprovider.ods b/sc/qa/unit/data/ods/chart2dataprovider.ods
new file mode 100644
index 000000000000..d27b78c83e82
--- /dev/null
+++ b/sc/qa/unit/data/ods/chart2dataprovider.ods