funambol-citadel-connector/src/test/java/net/bionicmessage/funambol/citadel/store/EmailObjectStoreTest.java

   1 /* This program is free software: you can redistribute it and/or modify
   2  * it under the terms of the GNU Affero General Public License as
   3  * published by the Free Software Foundation, either version 3 of the
   4  * License, or (at your option) any later version.
   5  *
   6  * This program is distributed in the hope that it will be useful,
   7  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   8  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   9  * GNU Affero General Public License for more details.
  10  *
  11  * You should have received a copy of the GNU Affero General Public License
  12  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  13  */
  14 package net.bionicmessage.funambol.citadel.store;
  15
  16 import java.io.File;
  17 import java.io.FileInputStream;
  18 import java.util.Iterator;
  19 import java.util.List;
  20 import java.util.Properties;
  21 import org.junit.After;
  22 import org.junit.AfterClass;
  23 import org.junit.Before;
  24 import org.junit.BeforeClass;
  25 import org.junit.Test;
  26 import static org.junit.Assert.*;
  27
  28 /**
  29  * A test class for the EmailObjectStore
  30  * @author matt
  31  */
  32 public class EmailObjectStoreTest {
  33
  34     Properties props = null;
  35
  36     public EmailObjectStoreTest() {
  37     }
  38
  39     @BeforeClass
  40     public static void setUpClass() throws Exception {
  41     }
  42
  43     @AfterClass
  44     public static void tearDownClass() throws Exception {
  45     }
  46
  47     @Before
  48     public void setUp() throws Exception {
  49         props = new Properties();
  50         /* Set up your config here */
  51         File configFile = new File("/home/matt/.emailobjectstore/eosprops");
  52         FileInputStream confStream = new FileInputStream(configFile);
  53         props.load(confStream);
  54     }
  55
  56     @After
  57     public void tearDown() {
  58     }
  59
  60     @Test
  61     public void syncHeaderTest() throws Exception {
  62         props.setProperty(CtdlFnblConstants.PURGE_DB_OPTION, "");
  63         EmailObjectStore eos = new EmailObjectStore(props);
  64         eos.startSync(0);
  65         List<String> msgsInMail = eos.listMessagesInRoom("Mail");
  66         Iterator<String> it = msgsInMail.iterator();
  67         while (it.hasNext()) {
  68             String key = it.next();
  69             CitadelMailObject cmo = eos.getMessageByPointer(key);
  70             System.out.format("Message %s, From: %s\n", key, cmo.getFrom()).flush();
  71         }
  72         eos.close();
  73     }
  74
  75     @Test
  76     public void syncAddBodies() throws Exception {
  77         EmailObjectStore eos = new EmailObjectStore(props);
  78         eos.startSync(0);
  79         List<String> msgsInMail = eos.listMessagesInRoom("Mail");
  80         Iterator<String> it = msgsInMail.iterator();
  81         while (it.hasNext()) {
  82             String key = it.next();
  83             CitadelMailObject cmo = eos.getFilledMessageByPointer(key);
  84             System.out.format("Message %s, From: %s\n", key, cmo.getFrom()).flush();
  85             System.out.format("Message body size: %d\n", cmo.getData().length()).flush();
  86         }
  87         eos.close();
  88     }
  89     @Test
  90     public void stillHaveBodies() throws Exception {
  91         System.out.println("stillHaveBodies()");
  92         EmailObjectStore eos = new EmailObjectStore(props);
  93         props.remove(CtdlFnblConstants.PURGE_DB_OPTION);
  94         eos.startSync(0);
  95         List<String> msgsInMail = eos.listMessagesInRoom("Mail");
  96         Iterator<String> it = msgsInMail.iterator();
  97         while (it.hasNext()) {
  98             String key = it.next();
  99             CitadelMailObject cmo = eos.getFilledMessageByPointer(key);
 100             System.out.println("Current key:"+key);
 101             if (!cmo.hasData()) {
 102                 eos.close(); // Keep the db intact
 103             }
 104             assertTrue(cmo.hasData());
 105         }
 106         eos.close();
 107     }
 108     @Test
 109     public void syncAddParts() throws Exception {
 110         EmailObjectStore eos = new EmailObjectStore(props);
 111         eos.startSync(0);
 112         List<String> msgsInMail = eos.listMessagesInRoom("Mail");
 113         Iterator<String> it = msgsInMail.iterator();
 114         while (it.hasNext()) {
 115             String key = it.next();
 116             CitadelMailObject cmo = eos.getFilledMessageByPointer(key);
 117             eos.fillPartsForMessage(cmo);
 118             System.out.println("Object "+key+ " has "+cmo.getAttachedParts().size() + " attached parts");
 119         }
 120         eos.close();
 121     }
 122 }