def setUp(self): super(WebSocketTestCase, self).setUp() DatabaseManager.add('default', ':memory:') User._users = UserManager() Token._tokens = TokenManager() self.username_0 = '*****@*****.**' self.password_0 = 'asdfasdf' self.username_1 = '*****@*****.**' self.password_1 = 'qwerqwer' self.user_0 = User(self.username_0, self.password_0, True) self.user_0.save() self.user_1 = User(self.username_1, self.password_1, True) self.user_1.save() self.token_0 = Token(user=self.user_0) self.token_0.save() self.token_1 = Token(user=self.user_1) self.token_1.save()
def setUp(self): # Create a temporary directory for writing the config file to # as well as for creating testing databases. self.temp_dir = tempfile.mkdtemp() # Create a config file programmatically. config = ConfigParser.RawConfigParser() config.add_section('sremote') config.set('sremote', 'tick', '3') config.set('sremote', 'send_update_tick', '5') config.set('sremote', 'max_wait_seconds_before_shutdown', '3') config.set('sremote', 'port_number', '8080') config.set('sremote', 'daemonize', 'False') config.set('sremote', 'log_level', 'DEBUG') config.set('sremote', 'log_file', os.path.join(self.temp_dir, 'sremote.log')) config.set('sremote', 'database_dir', self.temp_dir) # Writing our configuration file to 'temp_dir/sremote.conf' with open(os.path.join(self.temp_dir, 'sremote.conf'), 'wb') as configfile: config.write(configfile) parser = argparse.ArgumentParser(prog='sremote.py') parser.add_argument("--config", help="path to the configuration file.") # Create arguments programmatically self.args = parser.parse_args(['--config', os.path.join(self.temp_dir, 'sremote.conf')]) self.app = Application() # Overwrite default database with one in memory. # from sremote.models.database import DatabaseManager DatabaseManager.add('default', os.path.join(self.temp_dir, 'db.sqlite'))
def setUp(self): super(HTTPTestCase, self).setUp() DatabaseManager.add("default", ":memory:") User._users = UserManager() Token._tokens = TokenManager() self.username_0 = "*****@*****.**" self.password_0 = "asdfasdf" self.username_1 = "*****@*****.**" self.password_1 = "qwerqwer" self.user_0 = User(self.username_0, self.password_0, True) self.user_0.save() self.user_1 = User(self.username_1, self.password_1, True) self.user_1.save() self.token_0 = Token(user=self.user_0) self.token_0.save() self.token_1 = Token(user=self.user_1) self.token_1.save()
def setUp(self): DatabaseManager.add('default', ':memory:')
def testDatabaseManager(self): DatabaseManager.add("default", ":memory:") db = DatabaseManager.instance("default") self.assertTrue(db != None, "Database exists.") DatabaseManager.remove("default") self.assertRaises(KeyError, DatabaseManager.instance, "DatabaseManager raises KeyError when no db exists.")