280 lines
8.5 KiB
C++
280 lines
8.5 KiB
C++
/* === This file is part of Calamares - <https://github.com/calamares> ===
|
|
*
|
|
* SPDX-FileCopyrightText: 2018 Adriaan de Groot <groot@kde.org>
|
|
*
|
|
* Calamares is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Calamares is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Calamares. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
* SPDX-License-Identifier: GPL-3.0-or-later
|
|
* License-Filename: LICENSE
|
|
*
|
|
*/
|
|
|
|
#include "GeoIPTests.h"
|
|
|
|
#include "GeoIPFixed.h"
|
|
#include "GeoIPJSON.h"
|
|
#ifdef QT_XML_LIB
|
|
#include "GeoIPXML.h"
|
|
#endif
|
|
#include "Handler.h"
|
|
|
|
#include "network/Manager.h"
|
|
|
|
#include <QtTest/QtTest>
|
|
|
|
QTEST_GUILESS_MAIN( GeoIPTests )
|
|
|
|
using namespace CalamaresUtils::GeoIP;
|
|
|
|
GeoIPTests::GeoIPTests() {}
|
|
|
|
GeoIPTests::~GeoIPTests() {}
|
|
|
|
void
|
|
GeoIPTests::initTestCase()
|
|
{
|
|
}
|
|
|
|
static const char json_data_attribute[] = "{\"time_zone\":\"Europe/Amsterdam\"}";
|
|
|
|
void
|
|
GeoIPTests::testJSON()
|
|
{
|
|
GeoIPJSON handler;
|
|
auto tz = handler.processReply( json_data_attribute );
|
|
|
|
QCOMPARE( tz.first, QStringLiteral( "Europe" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Amsterdam" ) );
|
|
|
|
// JSON is quite tolerant
|
|
tz = handler.processReply( "time_zone: \"Europe/Brussels\"" );
|
|
QCOMPARE( tz.second, QStringLiteral( "Brussels" ) );
|
|
|
|
tz = handler.processReply( "time_zone: America/New_York\n" );
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testJSONalt()
|
|
{
|
|
GeoIPJSON handler( "zona_de_hora" );
|
|
|
|
auto tz = handler.processReply( json_data_attribute );
|
|
QCOMPARE( tz.first, QString() ); // Not found
|
|
|
|
tz = handler.processReply( "tarifa: 12\nzona_de_hora: Europe/Madrid" );
|
|
QCOMPARE( tz.first, QStringLiteral( "Europe" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Madrid" ) );
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testJSONbad()
|
|
{
|
|
static const char data[] = "time_zone: 1";
|
|
|
|
GeoIPJSON handler;
|
|
auto tz = handler.processReply( data );
|
|
|
|
tz = handler.processReply( data );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = handler.processReply( "" );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = handler.processReply( "<html><body>404 Forbidden</body></html>" );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = handler.processReply( "{ time zone = 'America/LosAngeles'}" );
|
|
QCOMPARE( tz.first, QString() );
|
|
}
|
|
|
|
|
|
static const char xml_data_ubiquity[] =
|
|
R"(<Response>
|
|
<Ip>85.150.1.1</Ip>
|
|
<Status>OK</Status>
|
|
<CountryCode>NL</CountryCode>
|
|
<CountryCode3>NLD</CountryCode3>
|
|
<CountryName>Netherlands</CountryName>
|
|
<RegionCode>None</RegionCode>
|
|
<RegionName>None</RegionName>
|
|
<City>None</City>
|
|
<ZipPostalCode/>
|
|
<Latitude>50.0</Latitude>
|
|
<Longitude>4.0</Longitude>
|
|
<AreaCode>0</AreaCode>
|
|
<TimeZone>Europe/Amsterdam</TimeZone>
|
|
</Response>)";
|
|
|
|
void
|
|
GeoIPTests::testXML()
|
|
{
|
|
#ifdef QT_XML_LIB
|
|
GeoIPXML handler;
|
|
auto tz = handler.processReply( xml_data_ubiquity );
|
|
|
|
QCOMPARE( tz.first, QStringLiteral( "Europe" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Amsterdam" ) );
|
|
#endif
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testXML2()
|
|
{
|
|
static const char data[]
|
|
= "<Response><TimeZone>America/North Dakota/Beulah</TimeZone></Response>"; // With a space!
|
|
|
|
#ifdef QT_XML_LIB
|
|
GeoIPXML handler;
|
|
auto tz = handler.processReply( data );
|
|
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "North_Dakota/Beulah" ) ); // Without space
|
|
#endif
|
|
}
|
|
|
|
|
|
void
|
|
GeoIPTests::testXMLalt()
|
|
{
|
|
#ifdef QT_XML_LIB
|
|
GeoIPXML handler( "ZT" );
|
|
|
|
auto tz = handler.processReply( "<A><B/><C><ZT>Moon/Dark_side</ZT></C></A>" );
|
|
QCOMPARE( tz.first, QStringLiteral( "Moon" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Dark_side" ) );
|
|
#endif
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testXMLbad()
|
|
{
|
|
#ifdef QT_XML_LIB
|
|
GeoIPXML handler;
|
|
auto tz = handler.processReply( "{time_zone: \"Europe/Paris\"}" );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = handler.processReply( "<Response></Response>" );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = handler.processReply( "fnord<html/>" );
|
|
QCOMPARE( tz.first, QString() );
|
|
#endif
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testSplitTZ()
|
|
{
|
|
using namespace CalamaresUtils::GeoIP;
|
|
auto tz = splitTZString( QStringLiteral( "Moon/Dark_side" ) );
|
|
QCOMPARE( tz.first, QStringLiteral( "Moon" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Dark_side" ) );
|
|
|
|
// Some providers return weirdly escaped data
|
|
tz = splitTZString( QStringLiteral( "America\\/NewYork" ) );
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "NewYork" ) ); // That's not actually the zone name
|
|
|
|
// Check that bogus data fails
|
|
tz = splitTZString( QString() );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
tz = splitTZString( QStringLiteral( "America.NewYork" ) );
|
|
QCOMPARE( tz.first, QString() );
|
|
|
|
// Check that three-level is split properly and space is replaced
|
|
tz = splitTZString( QStringLiteral( "America/North Dakota/Beulah" ) );
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "North_Dakota/Beulah" ) );
|
|
}
|
|
|
|
|
|
#define CHECK_GET( t, selector, url ) \
|
|
{ \
|
|
auto tz = GeoIP##t( selector ) \
|
|
.processReply( CalamaresUtils::Network::Manager::instance().synchronousGet( QUrl( url ) ) ); \
|
|
qDebug() << tz; \
|
|
QCOMPARE( default_tz, tz ); \
|
|
auto tz2 = CalamaresUtils::GeoIP::Handler( "" #t, url, selector ).get(); \
|
|
qDebug() << tz2; \
|
|
QCOMPARE( default_tz, tz2 ); \
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testGet()
|
|
{
|
|
if ( !QProcessEnvironment::systemEnvironment().contains( QStringLiteral( "TEST_HTTP_GET" ) ) )
|
|
{
|
|
qDebug() << "Skipping HTTP GET tests, set TEST_HTTP_GET environment variable to enable";
|
|
return;
|
|
}
|
|
|
|
GeoIPJSON default_handler;
|
|
// Call the KDE service the definitive source.
|
|
auto default_tz = default_handler.processReply(
|
|
CalamaresUtils::Network::Manager::instance().synchronousGet( QUrl( "https://geoip.kde.org/v1/calamares" ) ) );
|
|
|
|
// This is bogus, because the test isn't always run by me
|
|
// QCOMPARE( default_tz.first, QStringLiteral("Europe") );
|
|
// QCOMPARE( default_tz.second, QStringLiteral("Amsterdam") );
|
|
QVERIFY( !default_tz.first.isEmpty() );
|
|
QVERIFY( !default_tz.second.isEmpty() );
|
|
|
|
// Each expansion of CHECK_GET does a synchronous GET, then checks that
|
|
// the TZ data is the same as the default_tz; this is fragile if the
|
|
// services don't agree on the location of where the test is run.
|
|
CHECK_GET( JSON, QString(), "https://geoip.kde.org/v1/calamares" ) // Check it's consistent
|
|
CHECK_GET( JSON, QStringLiteral( "timezone" ), "https://ipapi.co/json" ) // Different JSON
|
|
CHECK_GET( JSON, QStringLiteral( "timezone" ), "http://ip-api.com/json" )
|
|
|
|
CHECK_GET( JSON, QStringLiteral( "Location.TimeZone" ), "https://geoip.kde.org/debug" ) // 2-level JSON
|
|
|
|
#ifdef QT_XML_LIB
|
|
CHECK_GET( XML, QString(), "http://geoip.ubuntu.com/lookup" ) // Ubiquity's XML format
|
|
CHECK_GET( XML, QString(), "https://geoip.kde.org/v1/ubiquity" ) // Temporary KDE service
|
|
#endif
|
|
}
|
|
|
|
void
|
|
GeoIPTests::testFixed()
|
|
{
|
|
{
|
|
GeoIPFixed f;
|
|
auto tz = f.processReply( QByteArray() );
|
|
QCOMPARE( tz.first, QStringLiteral( "Europe" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Amsterdam" ) );
|
|
|
|
QCOMPARE( f.processReply( xml_data_ubiquity ), tz );
|
|
QCOMPARE( f.processReply( QByteArray( "derp" ) ), tz );
|
|
}
|
|
{
|
|
GeoIPFixed f( QStringLiteral( "America/Vancouver" ) );
|
|
auto tz = f.processReply( QByteArray() );
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "Vancouver" ) );
|
|
|
|
QCOMPARE( f.processReply( xml_data_ubiquity ), tz );
|
|
QCOMPARE( f.processReply( QByteArray( "derp" ) ), tz );
|
|
}
|
|
{
|
|
GeoIPFixed f( QStringLiteral( "America/North Dakota/Beulah" ) );
|
|
auto tz = f.processReply( QByteArray() );
|
|
QCOMPARE( tz.first, QStringLiteral( "America" ) );
|
|
QCOMPARE( tz.second, QStringLiteral( "North_Dakota/Beulah" ) );
|
|
|
|
QCOMPARE( f.processReply( xml_data_ubiquity ), tz );
|
|
QCOMPARE( f.processReply( QByteArray( "derp" ) ), tz );
|
|
}
|
|
}
|