File indexing completed on 2024-05-12 05:09:57

 /***************************************************************************
     Copyright (C) 2013 Robby Stephenson <robby@periapsis.org>
  ***************************************************************************/
 
 /***************************************************************************
  *                                                                         *
  *   This program 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 2 of        *
  *   the License or (at your option) version 3 or any later version        *
  *   accepted by the membership of KDE e.V. (or its successor approved     *
  *   by the membership of KDE e.V.), which shall act as a proxy            *
  *   defined in Section 14 of version 3 of the license.                    *
  *                                                                         *
  *   This program 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 this program.  If not, see <http://www.gnu.org/licenses/>. *
  *                                                                         *
  ***************************************************************************/
 
 #undef QT_NO_CAST_FROM_ASCII
 
 #include "entrezfetchertest.h"
 
 #include "../fetch/entrezfetcher.h"
 #include "../entry.h"
 #include "../collections/bibtexcollection.h"
 #include "../collectionfactory.h"
 #include "../utils/datafileregistry.h"
 
 #include <KSharedConfig>
 
 #include <QTest>
 
 QTEST_GUILESS_MAIN( EntrezFetcherTest )
 
 EntrezFetcherTest::EntrezFetcherTest() : AbstractFetcherTest() {
 }
 
 void EntrezFetcherTest::initTestCase() {
   Tellico::DataFileRegistry::self()->addDataLocation(QFINDTESTDATA("../../xslt/pubmed2tellico.xsl"));
   Tellico::RegisterCollection<Tellico::Data::BibtexCollection> registerBibtex(Tellico::Data::Collection::Bibtex, "bibtex");
 
   m_config = KSharedConfig::openConfig(QString(), KConfig::SimpleConfig)->group(QStringLiteral("entrez"));
   m_config.writeEntry("Custom Fields", QStringLiteral("abstract"));
 
   m_fieldValues.insert(QStringLiteral("doi"), QStringLiteral("10.1016/j.tcb.2013.03.001"));
   m_fieldValues.insert(QStringLiteral("pmid"), QStringLiteral("23597843"));
   m_fieldValues.insert(QStringLiteral("title"), QStringLiteral("Stem cell competition: finding balance in the niche."));
   m_fieldValues.insert(QStringLiteral("author"), QStringLiteral("Rachel R Stine; Erika L Matunis"));
 //  m_fieldValues.insert(QStringLiteral("volume"), QStringLiteral("85"));
   m_fieldValues.insert(QStringLiteral("journal"), QStringLiteral("Trends in cell biology"));
 //  m_fieldValues.insert(QStringLiteral("publisher"), QStringLiteral("Springer"));
   m_fieldValues.insert(QStringLiteral("year"), QStringLiteral("2013"));
   m_fieldValues.insert(QStringLiteral("month"), QStringLiteral("8"));
   m_fieldValues.insert(QStringLiteral("entry-type"), QStringLiteral("article"));
 }
 
 void EntrezFetcherTest::testTitle() {
   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Bibtex, Tellico::Fetch::Title,
                                        m_fieldValues.value(QStringLiteral("title")));
   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::EntrezFetcher(this));
   fetcher->readConfig(m_config);
 
   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
 
   QCOMPARE(results.size(), 1);
   Tellico::Data::EntryPtr entry = results.at(0);
 
   QHashIterator<QString, QString> i(m_fieldValues);
   while(i.hasNext()) {
     i.next();
     QCOMPARE(entry->field(i.key()), i.value());
   }
   QVERIFY(entry->field(QStringLiteral("abstract")).contains(QStringLiteral("Drosophila")));
 }
 
 void EntrezFetcherTest::testAuthor() {
   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Bibtex, Tellico::Fetch::Person,
                                        QStringLiteral("Rachel R Stine"));
   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::EntrezFetcher(this));
 
   Tellico::Data::EntryList results = DO_FETCH(fetcher, request);
 
   QVERIFY(results.size() > 1);
 }
 
 void EntrezFetcherTest::testKeyword() {
   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Bibtex, Tellico::Fetch::Keyword,
                                        QStringLiteral("Drosophila"));
   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::EntrezFetcher(this));
 
   // fetcher defaults to 25 at a time, ask for 26 to check search continue
   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 26);
   QCOMPARE(results.size(), 26);
 
   // and make sure it's 26 different
   QSet<QString> titles;
   foreach(Tellico::Data::EntryPtr entry, results) {
     titles.insert(entry->title());
   }
   QCOMPARE(titles.size(), results.size());
 
 }
 
 void EntrezFetcherTest::testPMID() {
   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Bibtex, Tellico::Fetch::PubmedID,
                                        m_fieldValues.value(QStringLiteral("pmid")));
   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::EntrezFetcher(this));
 
   // there are several results for the same ISBN here
   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
 
   QCOMPARE(results.size(), 1);
 }
 
 void EntrezFetcherTest::testDOI() {
   Tellico::Fetch::FetchRequest request(Tellico::Data::Collection::Bibtex, Tellico::Fetch::DOI,
                                        m_fieldValues.value(QStringLiteral("doi")));
   Tellico::Fetch::Fetcher::Ptr fetcher(new Tellico::Fetch::EntrezFetcher(this));
   fetcher->readConfig(m_config);
 
   Tellico::Data::EntryList results = DO_FETCH1(fetcher, request, 1);
 
   QCOMPARE(results.size(), 1);
   Tellico::Data::EntryPtr entry = results.at(0);
 
   QHashIterator<QString, QString> i(m_fieldValues);
   while(i.hasNext()) {
     i.next();
     QCOMPARE(entry->field(i.key()), i.value());
   }
   QVERIFY(entry->field(QStringLiteral("abstract")).contains(QStringLiteral("Drosophila")));
 }