add copy_db to postgres

.pylintrc

@@ -375,7 +375,7 @@ known-third-party=enchant
 max-args=6
 
 # Maximum number of attributes for a class (see R0902).
-max-attributes=7
+max-attributes=12
 
 # Maximum number of boolean expressions in a if statement
 max-bool-expr=5

aiida/control/postgres.py

@@ -13,6 +13,7 @@
 _GRANT_PRIV_COMMAND = 'GRANT ALL PRIVILEGES ON DATABASE "{}" TO "{}"'
 _GET_USERS_COMMAND = "SELECT usename FROM pg_user WHERE usename='{}'"
 _CHECK_DB_EXISTS_COMMAND = "SELECT datname FROM pg_database WHERE datname='{}'"
+_COPY_DB_COMMAND = 'CREATE DATABASE "{}" WITH TEMPLATE "{}" OWNER "{}"'
 
 
 class Postgres(object):
@@ -157,6 +158,10 @@ def drop_db(self, dbname):
         """
         self.pg_execute(_DROP_DB_COMMAND.format(dbname), **self.dbinfo)
 
+    def copy_db(self, src_db, dest_db, dbuser):
+        self.pg_execute(
+            _COPY_DB_COMMAND.format(dest_db, src_db, dbuser), **self.dbinfo)
+
     def db_exists(self, dbname):
         """
         Check wether a postgres database with dbname exists

aiida/utils/fixtures.py

@@ -54,6 +54,8 @@ def __init__(self):
         }
         self.pg_cluster = None
         self.postgres = None
+        self.__is_running_on_test_db = False
+        self.__is_running_on_test_profile = False
 
     def create_db_cluster(self):
         self.pg_cluster = PGTest()
@@ -76,6 +78,7 @@ def create_aiida_db(self):
                 'Could not connect to the test postgres instance')
         self.postgres.create_dbuser(self.db_user, self.db_pass)
         self.postgres.create_db(self.db_user, self.db_name)
+        self.__is_running_on_test_db = True
 
     def create_root_dir(self):
         self.root_dir = tempfile.mkdtemp()
@@ -104,6 +107,7 @@ def create_profile(self):
             **self.profile)
         aiida_cfg.set_default_profile('verdi', profile_name)
         aiida_cfg.set_default_profile('daemon', profile_name)
+        self.__is_running_on_test_profile = True
 
     @property
     def profile(self):