tests/org/cacert/gigi/testUtils/ConfiguredTest.java

   1 package org.cacert.gigi.testUtils;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.io.File;
   6 import java.io.FileInputStream;
   7 import java.io.FileOutputStream;
   8 import java.io.IOException;
   9 import java.io.ObjectInputStream;
  10 import java.io.ObjectOutputStream;
  11 import java.security.GeneralSecurityException;
  12 import java.security.KeyPair;
  13 import java.security.KeyPairGenerator;
  14 import java.security.Signature;
  15 import java.sql.SQLException;
  16 import java.text.SimpleDateFormat;
  17 import java.util.Calendar;
  18 import java.util.Date;
  19 import java.util.Properties;
  20 import java.util.TimeZone;
  21 import java.util.regex.Matcher;
  22 import java.util.regex.Pattern;
  23
  24 import org.cacert.gigi.GigiApiException;
  25 import org.cacert.gigi.database.DatabaseConnection;
  26 import org.cacert.gigi.database.DatabaseConnection.Link;
  27 import org.cacert.gigi.database.GigiPreparedStatement;
  28 import org.cacert.gigi.database.SQLFileManager.ImportType;
  29 import org.cacert.gigi.dbObjects.CATS.CATSType;
  30 import org.cacert.gigi.dbObjects.Domain;
  31 import org.cacert.gigi.dbObjects.DomainPingType;
  32 import org.cacert.gigi.dbObjects.User;
  33 import org.cacert.gigi.testUtils.TestEmailReceiver.TestMail;
  34 import org.cacert.gigi.util.DatabaseManager;
  35 import org.cacert.gigi.util.DomainAssessment;
  36 import org.cacert.gigi.util.Notary;
  37 import org.cacert.gigi.util.PEM;
  38 import org.cacert.gigi.util.ServerConstants;
  39 import org.cacert.gigi.util.TimeConditions;
  40 import org.junit.BeforeClass;
  41
  42 import sun.security.pkcs10.PKCS10;
  43 import sun.security.pkcs10.PKCS10Attributes;
  44 import sun.security.x509.X500Name;
  45
  46 /**
  47  * Base class for a Testsuite that makes use of the config variables that define
  48  * the environment.
  49  */
  50 public abstract class ConfiguredTest {
  51
  52     static Properties testProps = new Properties();
  53
  54     public static Properties getTestProps() {
  55         return testProps;
  56     }
  57
  58     private static boolean envInited = false;
  59
  60     /**
  61      * Some password that fulfills the password criteria.
  62      */
  63     public static final String TEST_PASSWORD = "xvXV12°§";
  64
  65     public static final String DIFFICULT_CHARS = "ÜÖÄß𐀀";
  66
  67     @BeforeClass
  68     public static void initEnvironmentHook() throws IOException {
  69         initEnvironment();
  70     }
  71
  72     public static Properties initEnvironment() throws IOException {
  73         TimeZone.setDefault(TimeZone.getTimeZone("UTC"));
  74         if (envInited) {
  75             return generateProps();
  76         }
  77         envInited = true;
  78         try (FileInputStream inStream = new FileInputStream("config/test.properties")) {
  79             testProps.load(inStream);
  80         }
  81         Properties props = generateProps();
  82         ServerConstants.init(props);
  83         TimeConditions.init(props);
  84         DomainAssessment.init(props);
  85
  86         if ( !DatabaseConnection.isInited()) {
  87             DatabaseConnection.init(testProps);
  88             try {
  89                 l = DatabaseConnection.newLink(false);
  90             } catch (InterruptedException e) {
  91                 throw new Error(e);
  92             }
  93         }
  94         return props;
  95
  96     }
  97
  98     private static Properties generateProps() throws Error {
  99         Properties mainProps = new Properties();
 100         mainProps.setProperty("name.secure", testProps.getProperty("name.secure"));
 101         mainProps.setProperty("name.www", testProps.getProperty("name.www"));
 102         mainProps.setProperty("name.static", testProps.getProperty("name.static"));
 103         mainProps.setProperty("name.api", testProps.getProperty("name.api"));
 104
 105         mainProps.setProperty("https.port", testProps.getProperty("serverPort.https"));
 106         mainProps.setProperty("http.port", testProps.getProperty("serverPort.http"));
 107
 108         File out = new File("financial.dat");
 109         if ( !out.exists()) {
 110             try (FileOutputStream fos = new FileOutputStream(out)) {
 111                 fos.write("google.com\ntwitter.com\n".getBytes("UTF-8"));
 112             } catch (IOException e) {
 113                 throw new Error(e);
 114             }
 115         }
 116         mainProps.setProperty("highFinancialValue", out.getAbsolutePath());
 117         return mainProps;
 118     }
 119
 120     public static KeyPair generateKeypair() throws GeneralSecurityException {
 121         KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA");
 122         kpg.initialize(4096);
 123         KeyPair keyPair = null;
 124         File f = new File("testKeypair");
 125         if (f.exists()) {
 126             try (ObjectInputStream ois = new ObjectInputStream(new FileInputStream(f))) {
 127                 keyPair = (KeyPair) ois.readObject();
 128             } catch (ClassNotFoundException e) {
 129                 e.printStackTrace();
 130             } catch (IOException e) {
 131                 e.printStackTrace();
 132             }
 133         } else {
 134             keyPair = kpg.generateKeyPair();
 135             try (ObjectOutputStream oos = new ObjectOutputStream(new FileOutputStream(f))) {
 136                 oos.writeObject(keyPair);
 137                 oos.close();
 138             } catch (IOException e) {
 139                 e.printStackTrace();
 140             }
 141         }
 142         return keyPair;
 143     }
 144
 145     public static String generatePEMCSR(KeyPair kp, String dn) throws GeneralSecurityException, IOException {
 146         return generatePEMCSR(kp, dn, new PKCS10Attributes());
 147     }
 148
 149     public static String generatePEMCSR(KeyPair kp, String dn, PKCS10Attributes atts) throws GeneralSecurityException, IOException {
 150         return generatePEMCSR(kp, dn, atts, "SHA256WithRSA");
 151     }
 152
 153     public static String generatePEMCSR(KeyPair kp, String dn, PKCS10Attributes atts, String signature) throws GeneralSecurityException, IOException {
 154         PKCS10 p10 = new PKCS10(kp.getPublic(), atts);
 155         Signature s = Signature.getInstance(signature);
 156         s.initSign(kp.getPrivate());
 157         p10.encodeAndSign(new X500Name(dn), s);
 158         return PEM.encode("CERTIFICATE REQUEST", p10.getEncoded());
 159     }
 160
 161     static int count = 0;
 162
 163     private static Link l;
 164
 165     public static String createUniqueName() {
 166         return "test" + System.currentTimeMillis() + "a" + (count++) + "u";
 167     }
 168
 169     public static int countRegex(String text, String pattern) {
 170         Pattern p = Pattern.compile(pattern);
 171         Matcher m = p.matcher(text);
 172         int i = 0;
 173         while (m.find()) {
 174             i++;
 175         }
 176         return i;
 177     }
 178
 179     public static void makeAssurer(int uid) {
 180         try (GigiPreparedStatement ps1 = new GigiPreparedStatement("INSERT INTO cats_passed SET user_id=?, variant_id=?, language='en_EN', version='1'")) {
 181             ps1.setInt(1, uid);
 182             ps1.setInt(2, CATSType.ASSURER_CHALLENGE.getId());
 183             ps1.execute();
 184         }
 185
 186         try (GigiPreparedStatement ps2 = new GigiPreparedStatement("INSERT INTO `notary` SET `from`=?, `to`=?, points='100'")) {
 187             ps2.setInt(1, uid);
 188             ps2.setInt(2, User.getById(uid).getPreferredName().getId());
 189             ps2.execute();
 190         }
 191     }
 192
 193     public MailReceiver getMailReceiver() {
 194         throw new Error("Feature requires Business or ManagedTest.");
 195     }
 196
 197     public void verify(Domain d) {
 198         try {
 199             d.addPing(DomainPingType.EMAIL, "admin");
 200             TestMail testMail = getMailReceiver().receive();
 201             testMail.verify();
 202             assertTrue(d.isVerified());
 203         } catch (GigiApiException e) {
 204             throw new Error(e);
 205         } catch (IOException e) {
 206             throw new Error(e);
 207         }
 208     }
 209
 210     public static void purgeOnlyDB() throws SQLException, IOException {
 211         System.out.println("... resetting Database");
 212         long ms = System.currentTimeMillis();
 213         try {
 214             DatabaseManager.run(new String[] {
 215                     testProps.getProperty("sql.driver"), testProps.getProperty("sql.url"), testProps.getProperty("sql.user"), testProps.getProperty("sql.password")
 216             }, ImportType.TRUNCATE);
 217         } catch (ClassNotFoundException e) {
 218             e.printStackTrace();
 219         }
 220         System.out.println("Database reset complete in " + (System.currentTimeMillis() - ms) + " ms.");
 221     }
 222
 223     public static String validVerificationDateString() {
 224         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
 225         Calendar c = Calendar.getInstance();
 226         c.setTimeInMillis(System.currentTimeMillis());
 227         c.add(Calendar.MONTH, -Notary.LIMIT_MAX_MONTHS_VERIFICATION + 1);
 228         return sdf.format(new Date(c.getTimeInMillis()));
 229     }
 230
 231 }