1   /*
2    * $Id: $
3    *
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *  http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  package org.apache.struts.apps.mailreader.dao.impl.memory;
23  
24  import java.io.File;
25  
26  import org.apache.struts.apps.mailreader.dao.BaseTestUserDatabase;
27  import org.apache.struts.apps.mailreader.dao.Subscription;
28  import org.apache.struts.apps.mailreader.dao.User;
29  import org.apache.struts.apps.mailreader.dao.UserDatabase;
30  
31  
32  
33  public class MemoryUserDatabaseTest extends BaseTestUserDatabase {
34  
35      protected String defaultPathName = "test-database.xml";
36      private boolean deleteDatabaseFile = true;
37  
38      public boolean isDeleteDatabaseFile() {
39          return deleteDatabaseFile;
40      }
41  
42      public void setDeleteDatabaseFile(boolean deleteDatabaseFile) {
43          this.deleteDatabaseFile = deleteDatabaseFile;
44      }
45  
46      protected UserDatabase getNewUserDatabase() {
47          // using default impl
48          MemoryUserDatabase memoryUserDatabase = new MemoryUserDatabase();
49          memoryUserDatabase.setPathname(defaultPathName);
50          userDatabase = memoryUserDatabase;
51          return memoryUserDatabase;
52  
53      }
54      protected User getNewUser(UserDatabase db, String userName){
55          // using default impl
56          return db.createUser(userName);
57      }
58      protected Subscription getNewSubscription(User user, String host) {
59          return user.createSubscription(host);
60      }
61  
62      protected void setUp() throws Exception {
63          super.setUp();
64          // force write to disk
65         userDatabase.close();
66      }
67  
68      protected void tearDown() throws Exception {
69          super.tearDown();
70          if (isDeleteDatabaseFile()) {
71              File file = new File(defaultPathName);
72              file.delete();
73          }
74      }
75  }