transport/sqlite/test/aaSqliteTransportTestModule.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
   3 /*
   4  * Copyright (C) 2008 Sergey Yanovich <ynvich@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public
  17  * License along with this program; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include "xpcom-config.h"
  23
  24 #include "nsIGenericFactory.h"
  25 #include "nsStringAPI.h"
  26
  27 /* Unfrozen API */
  28 #include "nsIStringEnumerator.h"
  29 #include "nsITest.h"
  30 #include "nsITestRunner.h"
  31
  32 /* Project includes */
  33 #include "aaSqliteChannelTest.h"
  34 #include "aaSqliteLoadEntityTest.h"
  35 #include "aaSqliteCreateTable1Test.h"
  36
  37 #define AA_SQLITE_TRANSPORT_TEST_MODULE_CID \
  38 {0x8ebe8443, 0xe59b, 0x4ffc, {0xa5, 0x40, 0x93, 0xf5, 0x26, 0xee, 0x1d, 0x5b}}
  39 #define AA_SQLITE_TRANSPORT_TEST_MODULE_CONTRACT \
  40 "@aasii.org/transport/sqlite/unit;1"
  41
  42 class aaSqliteTransportTestModule: public nsITest,
  43                                    public nsIUTF8StringEnumerator
  44 {
  45 public:
  46   aaSqliteTransportTestModule() :mSubtest(0) {;}
  47   NS_DECL_ISUPPORTS
  48   NS_DECL_NSITEST
  49   NS_DECL_NSIUTF8STRINGENUMERATOR
  50 private:
  51   ~aaSqliteTransportTestModule() {;}
  52   PRUint32 mSubtest;
  53 };
  54
  55 NS_IMPL_ISUPPORTS2(aaSqliteTransportTestModule,
  56                    nsITest,
  57                    nsIUTF8StringEnumerator)
  58
  59 NS_IMETHODIMP
  60 aaSqliteTransportTestModule::Test(nsITestRunner *aTestRunner)
  61 {
  62   return NS_OK;
  63 }
  64
  65 /* nsIUTF8StringEnumerator */
  66 static const char* subtests[] =
  67 {
  68   AA_SQLITE_CREATE_TABLE1_CONTRACT
  69   ,"@aasii.org/transport/xml/sqlite;1"
  70 };
  71 #define subtestCount (sizeof(subtests) / sizeof(char*))
  72
  73 NS_IMETHODIMP
  74 aaSqliteTransportTestModule::HasMore(PRBool *aHasMore)
  75 {
  76   NS_ENSURE_ARG_POINTER(aHasMore);
  77   *aHasMore = (mSubtest < subtestCount);
  78   return NS_OK;
  79 }
  80
  81 NS_IMETHODIMP
  82 aaSqliteTransportTestModule::GetNext(nsACString &aContractID)
  83 {
  84   NS_ENSURE_TRUE(mSubtest < subtestCount, NS_ERROR_FAILURE);
  85   aContractID.Assign(subtests[mSubtest++]);
  86   return NS_OK;
  87 }
  88
  89 /* Module and Factory code */
  90 NS_GENERIC_FACTORY_CONSTRUCTOR(aaSqliteTransportTestModule)
  91
  92 static NS_IMETHODIMP
  93 aaSqliteLoadEntityTestConstructor(nsISupports *aOuter, REFNSIID aIID,
  94                                   void **aResult)
  95 {
  96     nsresult rv;
  97     aaSqliteChannelTest *inst;
  98
  99     *aResult = NULL;
 100     NS_ENSURE_TRUE(NULL == aOuter, NS_ERROR_NO_AGGREGATION);
 101
 102     inst = new aaSqliteChannelTest(AA_SQLITE_LOAD_ENTITY_QUERY,
 103         AA_SQLITE_LOAD_ENTITY_RESULT);
 104     NS_ENSURE_TRUE(NULL != inst, NS_ERROR_OUT_OF_MEMORY);
 105
 106     NS_ADDREF(inst);
 107     rv = inst->QueryInterface(aIID, aResult);
 108     NS_RELEASE(inst);
 109
 110     return rv;
 111 }
 112
 113 static NS_IMETHODIMP
 114 aaSqliteCreateTable1TestConstructor(nsISupports *aOuter, REFNSIID aIID,
 115                                   void **aResult)
 116 {
 117     nsresult rv;
 118     aaSqliteChannelTest *inst;
 119
 120     *aResult = NULL;
 121     NS_ENSURE_TRUE(NULL == aOuter, NS_ERROR_NO_AGGREGATION);
 122
 123     inst = new aaSqliteChannelTest(AA_SQLITE_CREATE_TABLE1_QUERY,
 124         AA_SQLITE_CREATE_TABLE1_RESULT);
 125     NS_ENSURE_TRUE(NULL != inst, NS_ERROR_OUT_OF_MEMORY);
 126
 127     NS_ADDREF(inst);
 128     rv = inst->QueryInterface(aIID, aResult);
 129     NS_RELEASE(inst);
 130
 131     return rv;
 132 }
 133
 134 static const nsModuleComponentInfo kComponents[] =
 135 {
 136   { "Sqlite Transport Module Test Container",
 137     AA_SQLITE_TRANSPORT_TEST_MODULE_CID,
 138     AA_SQLITE_TRANSPORT_TEST_MODULE_CONTRACT,
 139     aaSqliteTransportTestModuleConstructor}
 140   ,{"Sqlite Channel Load Entity Test",
 141     AA_SQLITE_LOAD_ENTITY_CID,
 142     AA_SQLITE_LOAD_ENTITY_CONTRACT,
 143     aaSqliteLoadEntityTestConstructor}
 144   ,{"Sqlite Channel Create Table Test",
 145     AA_SQLITE_CREATE_TABLE1_CID,
 146     AA_SQLITE_CREATE_TABLE1_CONTRACT,
 147     aaSqliteCreateTable1TestConstructor}
 148 };
 149 NS_IMPL_NSGETMODULE(aaSqliteTransportTest, kComponents)