1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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
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
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
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 }