def test_set_databases_correct(self):
mysql = Mysql(config)
mysql.set_databases(config['databases']['mysql'])
self.assertEqual(mysql.databases, config['databases']['mysql'].split(','))
databases = 'db1, db2, db3,db4'
mysql.set_databases(databases)
self.assertEqual(mysql.databases, [x.strip() for x in databases.split(',')])
def test_dump_databases(self, mock_call, mock_create):
mysql = Mysql(config)
mysql.dump_databases()
self.assertTrue(mock_create.called)
self.assertTrue(mock_call.called)
def test_backup(self, mock_cleanup, mock_archive, mock_dump):
mysql = Mysql(config)
mysql.backup()
self.assertTrue(mock_cleanup.called)
self.assertTrue(mock_archive.called)
self.assertTrue(mock_dump.called)
def test_my_credentials_error(self):
mysql = Mysql(config)
with self.assertRaises(ValueError):
mysql.set_mysql_credentials([])
def test_set_databases_error(self):
mysql = Mysql(config)
with self.assertRaises(ValueError):
mysql.set_databases('')
