1   package org.weda.store;
2   
3   import java.util.HashMap;
4   import java.util.Map;
5   import org.weda.test.WedaTestCase;
6   import org.weda.domain.Address;
7   import org.weda.domain.Company;
8   
9   public class InsertDataTestCase extends WedaTestCase {
10      
11      public InsertDataTestCase(String name) throws Exception {
12          super(name);
13      }
14      
15      public void test() throws Exception{
16          ObjectStore pserv = 
17              (ObjectStore)registry.getService(ObjectStore.class);
18          ObjectSourceRegistry dsr = 
19              (ObjectSourceRegistry)registry.getService(ObjectSourceRegistry.class);
20          ObjectSource ds = dsr.getObjectSource("company2");
21          //
22          pserv.beginTransaction();
23          pserv.deleteAll(Company.class);
24          pserv.deleteAll(Address.class);
25          Company comp;
26          Map<String, Company> companies = new HashMap<String, Company>();
27          for (int i=0; i<100; i++){
28              comp = new Company();
29              comp.setName("name "+i);
30              Address addr = new Address();
31              addr.setStreet("street "+i);
32              pserv.save(addr);
33              pserv.save(comp);
34              companies.put(comp.getName(), comp);
35          }        
36          for (int i=0; i<300; i++){
37              comp = new Company();
38              comp.setName("name "+i);
39              Address addr = new Address();
40              addr.setStreet("street "+i);
41              comp.setAddress(addr);
42              pserv.save(addr);
43              pserv.save(comp);
44              companies.put(comp.getName(), comp);
45          }        
46          pserv.commit();
47      }
48      
49  }