def test_create_dump(self, mock_dump_cmd):
connector = MysqlDumpConnector()
dump = connector.create_dump()
# Test dump
dump_content = dump.read()
self.assertTrue(dump_content)
self.assertEqual(dump_content, b'foo')
# Test cmd
self.assertTrue(mock_dump_cmd.called)
def test_restore_dump_password(self, mock_dump_cmd, mock_restore_cmd):
connector = MysqlDumpConnector()
dump = connector.create_dump()
# Without
connector.settings.pop('PASSWORD', None)
connector.restore_dump(dump)
self.assertNotIn(' --password='******'PASSWORD'] = '******'
connector.restore_dump(dump)
self.assertIn(' --password=foo', mock_restore_cmd.call_args[0][0])
def test_restore_dump_user(self, mock_dump_cmd, mock_restore_cmd):
connector = MysqlDumpConnector()
dump = connector.create_dump()
# Without
connector.settings.pop('USER', None)
connector.restore_dump(dump)
self.assertNotIn(' --user='******'USER'] = '******'
connector.restore_dump(dump)
self.assertIn(' --user=foo', mock_restore_cmd.call_args[0][0])
def test_restore_dump_port(self, mock_dump_cmd, mock_restore_cmd):
connector = MysqlDumpConnector()
dump = connector.create_dump()
# Without
connector.settings.pop('PORT', None)
connector.restore_dump(dump)
self.assertNotIn(' --port=', mock_restore_cmd.call_args[0][0])
# With
connector.settings['PORT'] = 42
connector.restore_dump(dump)
self.assertIn(' --port=42', mock_restore_cmd.call_args[0][0])
def test_create_dump_host(self, mock_dump_cmd):
connector = MysqlDumpConnector()
# Without
connector.settings.pop('HOST', None)
connector.create_dump()
self.assertNotIn(' --host=', mock_dump_cmd.call_args[0][0])
# With
connector.settings['HOST'] = 'foo'
connector.create_dump()
self.assertIn(' --host=foo', mock_dump_cmd.call_args[0][0])
def test_create_dump_password(self, mock_dump_cmd):
connector = MysqlDumpConnector()
# Without
connector.settings.pop('PASSWORD', None)
connector.create_dump()
self.assertNotIn(' --password='******'PASSWORD'] = '******'
connector.create_dump()
self.assertIn(' --password=\'foo bar\'', mock_dump_cmd.call_args[0][0])
def test_restore_dump(self, mock_dump_cmd, mock_restore_cmd):
connector = MysqlDumpConnector()
dump = connector.create_dump()
connector.restore_dump(dump)
# Test cmd
self.assertTrue(mock_restore_cmd.called)
def test_create_dump_exclude(self, mock_dump_cmd):
connector = MysqlDumpConnector()
connector.settings['NAME'] = 'db'
# Without
connector.create_dump()
self.assertNotIn(' --ignore-table=', mock_dump_cmd.call_args[0][0])
# With
connector.exclude = ('foo',)
connector.create_dump()
self.assertIn(' --ignore-table=db.foo', mock_dump_cmd.call_args[0][0])
# With serveral
connector.exclude = ('foo', 'bar')
connector.create_dump()
self.assertIn(' --ignore-table=db.foo', mock_dump_cmd.call_args[0][0])
self.assertIn(' --ignore-table=db.bar', mock_dump_cmd.call_args[0][0])