def setUp(self):
if not os.path.exists(os.path.abspath('migrations')):
os.mkdir(os.path.abspath('migrations'))
self.config = create_config(migrations_dir='.:migrations')
self.test_migration_files = []
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115100_01_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115200_02_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115300_03_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115400_04_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115500_05_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115600_06_test_migration.migration', 'foo', 'bar')))
def setUp(self):
super(SimpleDBMigrateTest, self).setUp()
if not os.path.exists(os.path.abspath('migrations')):
os.mkdir(os.path.abspath('migrations'))
self.config = create_config(migrations_dir='.:migrations')
self.test_migration_files = []
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115100_01_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115200_02_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115300_03_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('20090214115400_04_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115500_05_test_migration.migration', 'foo', 'bar')))
self.test_migration_files.append(os.path.abspath(create_migration_file('migrations/20090214115600_06_test_migration.migration', 'foo', 'bar')))
def setUp(self):
targets = io.StringIO("127.0.0.1\n")
options_agent = Namespace(name='data/data_agent',
s='127.0.0.1',
p=9011,
cmd='agent')
options_srv = Namespace(name='data',
b='127.0.0.1',
p=9011,
cmd='srv',
targets=targets)
self.cfg = create_config()
self.settings_srv = Config(self.cfg, options_srv)
self.settings_srv.target_optimization(targets)
self.settings_agent = Config(self.cfg, options_agent)
def setUp(self):
options = Namespace(name='data', b='127.0.0.1', p=9011,
cmd='srv', targets='foofile')
self.cfg = create_config()
self.patcher = patch('os.makedirs')
mos_isfile = patch('os.path.isfile')
mos_isfile.return_value = True
mos_isfile.start()
self.mock_makedirs = self.patcher.start()
self.settings = Config(self.cfg, options)
# ctx = Context(self.settings)
self.ctx = MagicMock(spect=Context)
self.ctx.pop.return_value = ("127.0.0.1", "-sV -Pn -p1-1000")
self.ctx.secret_key = self.settings.secret_key
self.ctx.is_finished = False
self.mock_server = MagicMock(spect=DScanServer)
self.mock_server.secret_key = self.settings.secret_key
self.mock_terminate = MagicMock()
self.mock_terminate.is_set.return_value = False
self.hmac_patch = patch.object(hmac, 'compare_digest',
return_value=True)
self.challenge = b'4%x8p\x8d\xda\x04\xe5r\xfb\xc1Si8[' \
b'\xcb\x1a\x1c\x84\xf5\xb5%\x15[' \
b'\xea\x10\x96)!n\xe3\xadit\x0f\x15e\xc2\x06\xd1' \
b'\xd8\xb0' \
b'\xc5\x81\x87\xf2s\xe3\xd8\x95\xd1\x9c\xbdM\x8f' \
b'\x9c\xd5' \
b'\x14\xb3\x8e\xdd\x8eQ\xffw\x10Y8\xa5\xa5\x83\xf3' \
b'\xeeQ' \
b'\xa1\xfcOP\x9d\xd6\x80x\x80\x9eh\x11\xa7\xd7\xce' \
b'\xcf' \
b'.\xec\x01\x94S\xd4\x1d\x7f\xef\x83e\xe8\xfa\xf9' \
b'`\xfb' \
b'\xc6:SB\xeff\x15\r\xcb\xe9\xa4\xefO\x03i\xe9' \
b'\xefoMz\x8b'
self.hmac_patch.start()
self.addCleanup(self.patcher.stop)
self.addCleanup(self.hmac_patch.stop)
self.addCleanup(mos_isfile.stop)
def setUp(self):
super(SimpleVirtuosoMigrateTest, self).setUp()
self.config = create_config(migrations_dir='.')
