File indexing completed on 2024-05-12 16:46:23

0001 /***************************************************************************
0002     Copyright (C) 2010-2011 Robby Stephenson <robby@periapsis.org>
0003  ***************************************************************************/
0004 
0005 /***************************************************************************
0006  *                                                                         *
0007  *   This program is free software; you can redistribute it and/or         *
0008  *   modify it under the terms of the GNU General Public License as        *
0009  *   published by the Free Software Foundation; either version 2 of        *
0010  *   the License or (at your option) version 3 or any later version        *
0011  *   accepted by the membership of KDE e.V. (or its successor approved     *
0012  *   by the membership of KDE e.V.), which shall act as a proxy            *
0013  *   defined in Section 14 of version 3 of the license.                    *
0014  *                                                                         *
0015  *   This program is distributed in the hope that it will be useful,       *
0016  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
0017  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
0018  *   GNU General Public License for more details.                          *
0019  *                                                                         *
0020  *   You should have received a copy of the GNU General Public License     *
0021  *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
0022  *                                                                         *
0023  ***************************************************************************/
0024 
0025 #undef QT_NO_CAST_FROM_ASCII
0026 
0027 #include "musicbrainzfetchertest.h"
0028 
0029 #include "../fetch/musicbrainzfetcher.h"
0030 #include "../collections/musiccollection.h"
0031 #include "../collectionfactory.h"
0032 #include "../entry.h"
0033 #include "../images/imagefactory.h"
0034 #include "../utils/datafileregistry.h"
0035 
0036 #include <KSharedConfig>
0037 
0038 #include <QTest>
0039 
0040 QTEST_GUILESS_MAIN( MusicBrainzFetcherTest )
0041 
0042 MusicBrainzFetcherTest::MusicBrainzFetcherTest() : AbstractFetcherTest() {
0043 }
0044 
0045 void MusicBrainzFetcherTest::initTestCase() {
0046   Tellico::RegisterCollection<Tellico::Data::MusicCollection> registerMusic(Tellico::Data::Collection::Album, "album");
0047   // since we use the importer
0048   Tellico::DataFileRegistry::self()->addDataLocation(QFINDTESTDATA("../../xslt/musicbrainz2tellico.xsl"));
0049   Tellico::ImageFactory::init();
0050 
0051   m_fieldValues.insert(QStringLiteral("title"), QStringLiteral("carried along"));
0052   m_fieldValues.insert(QStringLiteral("artist"), QStringLiteral("Andrew Peterson"));
0053   m_fieldValues.insert(QStringLiteral("label"), QStringLiteral("essential records"));
0054   m_fieldValues.insert(QStringLiteral("year"), QStringLiteral("2000"));
0055   m_fieldValues.insert(QStringLiteral("medium"), QStringLiteral("compact disc"));
0056 }
0057 
0058 void MusicBrainzFetcherTest::testTitle() {
0059   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Title,
0060                                        m_fieldValues.value(QStringLiteral("title")));
0061   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0062 
0063   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0064   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0065 
0066   QCOMPARE(results.size(), 1);
0067 
0068   Tellico::Data::EntryPtr entry = results.at(0);
0069   QHashIterator<QString, QString> i(m_fieldValues);
0070   while(i.hasNext()) {
0071     i.next();
0072     QString result = entry->field(i.key()).toLower();
0073     QCOMPARE(result, i.value().toLower());
0074   }
0075   QVERIFY(!entry->field(QStringLiteral("track")).isEmpty());
0076   QVERIFY(!entry->field(QStringLiteral("cover")).isEmpty());
0077   QVERIFY(!entry->field(QStringLiteral("cover")).contains(QLatin1Char('/')));
0078 }
0079 
0080 void MusicBrainzFetcherTest::testKeyword() {
0081   // the total test case ends up exceeding the throttle limit so pause for a second
0082   QTest::qWait(1000);
0083 
0084   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Keyword,
0085                                        m_fieldValues.value(QStringLiteral("title")));
0086   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0087 
0088   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0089   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0090 
0091   QCOMPARE(results.size(), 1);
0092 
0093   Tellico::Data::EntryPtr entry = results.at(0);
0094   QHashIterator<QString, QString> i(m_fieldValues);
0095   while(i.hasNext()) {
0096     i.next();
0097     QString result = entry->field(i.key()).toLower();
0098     QCOMPARE(result, i.value().toLower());
0099   }
0100   QVERIFY(!entry->field(QStringLiteral("track")).isEmpty());
0101   QVERIFY(!entry->field(QStringLiteral("cover")).isEmpty());
0102   QVERIFY(!entry->field(QStringLiteral("cover")).contains(QLatin1Char('/')));
0103 }
0104 
0105 void MusicBrainzFetcherTest::testBug426560() {
0106   // the total test case ends up exceeding the throttle limit so pause for a second
0107   QTest::qWait(1000);
0108 
0109   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Keyword,
0110                                        QStringLiteral("lily allen - no shame"));
0111   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0112 
0113   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0114   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0115 
0116   QCOMPARE(results.size(), 1);
0117 
0118   Tellico::Data::EntryPtr entry = results.at(0);
0119   QVERIFY(entry);
0120   QCOMPARE(entry->title(), QStringLiteral("No Shame"));
0121   QCOMPARE(entry->field(QStringLiteral("artist")), QStringLiteral("Lily Allen"));
0122 }
0123 
0124 void MusicBrainzFetcherTest::testPerson() {
0125   const QString artist(QStringLiteral("artist"));
0126   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Person,
0127                                        m_fieldValues.value(artist));
0128   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0129 
0130   // TODO: Fetcher::setLimit should be virtual in Fetcher class
0131   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0132   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0133 
0134   QVERIFY(results.size() > 0);
0135 
0136   Tellico::Data::EntryPtr entry = results.at(0);
0137   QVERIFY(entry);
0138   QCOMPARE(entry->field(artist), m_fieldValues.value(artist));
0139 }
0140 
0141 void MusicBrainzFetcherTest::testACDC() {
0142   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Person,
0143                                        QStringLiteral("AC/DC"));
0144   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0145 
0146   // TODO: Fetcher::setLimit should be virtual in Fetcher class
0147   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0148   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0149 
0150   QVERIFY(results.size() > 0);
0151 
0152   Tellico::Data::EntryPtr entry = results.at(0);
0153   QVERIFY(entry);
0154   QCOMPARE(entry->field(QStringLiteral("artist")), QStringLiteral("AC/DC"));
0155 }
0156 
0157 // test grabbing cover art from coverartarchive.org
0158 void MusicBrainzFetcherTest::testCoverArt() {
0159   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Title,
0160                                        QStringLiteral("Laulut ja tarinat"));
0161   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0162 
0163   static_cast<Tellico::Fetch::MusicBrainzFetcher*>(fetcher.data())->setLimit(1);
0164   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0165 
0166   QVERIFY(!results.isEmpty());
0167 
0168   Tellico::Data::EntryPtr entry = results.at(0);
0169   QCOMPARE(entry->title(), QStringLiteral("Laulut ja tarinat"));
0170   QVERIFY(!entry->field(QStringLiteral("cover")).isEmpty());
0171   QVERIFY(!entry->field(QStringLiteral("cover")).contains(QLatin1Char('/')));
0172 }
0173 
0174 void MusicBrainzFetcherTest::testSoundtrack() {
0175   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::Title,
0176                                        QStringLiteral("legend of bagger vance"));
0177   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0178 
0179   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0180 
0181   QVERIFY(!results.isEmpty());
0182 
0183   Tellico::Data::EntryPtr entry = results.at(0);
0184   QCOMPARE(entry->title(), QStringLiteral("The Legend of Bagger Vance"));
0185   // sound tracks are the only genre tag that is read
0186   QCOMPARE(entry->field(QStringLiteral("genre")), QStringLiteral("Soundtrack"));
0187 }
0188 
0189 void MusicBrainzFetcherTest::testBarcode() {
0190   KConfigGroup cg = KSharedConfig::openConfig(QString(), KConfig::SimpleConfig)->group(QStringLiteral("musicbrainz"));
0191   cg.writeEntry("Custom Fields", QStringLiteral("barcode"));
0192 
0193   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Album, Tellico::Fetch::UPC,
0194                                        QStringLiteral("8024391054123"));
0195   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::MusicBrainzFetcher(this));
0196   fetcher->readConfig(cg);
0197 
0198   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
0199 
0200   QVERIFY(!results.isEmpty());
0201 
0202   Tellico::Data::EntryPtr entry = results.at(0);
0203   QCOMPARE(entry->title(), QStringLiteral("The Old Man and the Spirit"));
0204   QCOMPARE(entry->field(QStringLiteral("barcode")), QStringLiteral("8024391054123"));
0205 }