def setUp(self):
EnvironmentUtil.setup_fresh_test_env()
self.context = CommandContext.via({'type': 'memory-db'})
self.downCommand = DownCommand(self.context)
self.newCommand = NewCommand(self.context)
self.upCommand = UpCommand(self.context)
self.checkCommand = CheckCommand(self.context)
def setUp(self):
EnvironmentUtil.setup_fresh_test_env()
self.context = CommandContext.via({
'type': 'memory-db'})
self.upCommand = UpCommand(self.context)
class UpTest(unittest.TestCase):
def setUp(self):
EnvironmentUtil.setup_fresh_test_env()
self.context = CommandContext.via({
'type': 'memory-db'})
self.upCommand = UpCommand(self.context)
def tearDown(self):
EnvironmentUtil.teardown_fresh_test_env()
def test_no_arg_does_all_current_alters_when_none(self):
self.assertEqual(len(MemoryDb.data), 0)
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 0)
def test_no_arg_does_all_current_alters_when_alters(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
def test_ref_does_all_alters_including_ref(self):
id1, id2 = AlterUtil.create_alters([1,2])
sys.argv = make_argv([str(id1)])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
def test_ref_does_nothing_when_ref_doesnt_exist(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([str(10)])
try:
self.upCommand.run()
except MissingRefError:
pass
def test_n_option_runs_up_given_number_of_alters(self):
AlterUtil.create_alters([1, 2, 3, 4])
sys.argv = make_argv(['-n2'])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 2)
def test_up_resumes_after_last_executed_alter(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
AlterUtil.create_alters([2])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 2)
def test_stop_on_error(self):
AlterUtil.create_alters([1, 'error', 2])
sys.argv = make_argv([])
try:
self.upCommand.run()
except AppliedAlterError:
pass
self.assertEqual(len(MemoryDb.data), 1)
def test_continue_on_error_with_force(self):
AlterUtil.create_alters([1, 'error', 2])
sys.argv = make_argv(['-f'])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 3)
def init(cls):
if not cls.initialized:
cls.context = CommandContext.via({'type': 'memory-db'})
cls.newCommand = NewCommand(cls.context)
cls.upCommand = UpCommand(cls.context)
cls.initialized = True
class UpTest(unittest.TestCase):
def setUp(self):
EnvironmentUtil.setup_fresh_test_env()
self.context = CommandContext.via({'type': 'memory-db'})
self.upCommand = UpCommand(self.context)
def tearDown(self):
EnvironmentUtil.teardown_fresh_test_env()
def test_no_arg_does_all_current_alters_when_none(self):
self.assertEqual(len(MemoryDb.data), 0)
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 0)
def test_no_arg_does_all_current_alters_when_alters(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
def test_ref_does_all_alters_including_ref(self):
id1, id2 = AlterUtil.create_alters([1, 2])
sys.argv = make_argv([str(id1)])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
def test_ref_does_nothing_when_ref_doesnt_exist(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([str(10)])
try:
self.upCommand.run()
except MissingRefError:
pass
def test_n_option_runs_up_given_number_of_alters(self):
AlterUtil.create_alters([1, 2, 3, 4])
sys.argv = make_argv(['-n2'])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 2)
def test_up_resumes_after_last_executed_alter(self):
AlterUtil.create_alters([1])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 1)
AlterUtil.create_alters([2])
sys.argv = make_argv([])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 2)
def test_stop_on_error(self):
AlterUtil.create_alters([1, 'error', 2])
sys.argv = make_argv([])
try:
self.upCommand.run()
except AppliedAlterError:
pass
self.assertEqual(len(MemoryDb.data), 1)
def test_continue_on_error_with_force(self):
AlterUtil.create_alters([1, 'error', 2])
sys.argv = make_argv(['-f'])
self.upCommand.run()
self.assertEqual(len(MemoryDb.data), 3)
