def test_base_undoes_all_but_last_when_more_than_one(self): AlterUtil.create_alters([1, 2]) AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 2) sys.argv = make_argv(['base']) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 1)
def test_n_option_runs_down_given_number_of_alters(self): AlterUtil.create_alters([1, 2, 3, 4]) AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 4) sys.argv = make_argv(['-n2']) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 2)
def test_all_undoes_all_current_alters_when_alters(self): AlterUtil.create_alters([1]) AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 1) sys.argv = make_argv(['all']) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 0)
def test_base_undoes_none_when_one_alter(self): AlterUtil.create_alters([1]) AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 1) sys.argv = make_argv(['base']) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 1)
def test_ref_undoes_nothing_when_ref_doesnt_exist(self): AlterUtil.create_alters([1, 2, 3, 4]) AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 4) sys.argv = make_argv([str(10)]) try: self.downCommand.run() except MissingRefError: pass self.assertEqual(len(MemoryDb.data), 4)
def help_test_missing_alter(self, search, error): AlterUtil.create_alters([1]) AlterUtil.run_alters() alter_files = ChainUtil.get_alter_files() self.assertEqual(len(alter_files), 2) # up and down os.remove([i for i in alter_files if search in i].pop()) try: sys.argv = make_argv([]) self.checkCommand.run() except error: pass
def test_ref_undoes_all_alters_including_ref(self): AlterUtil.create_alters([1,2,3]) ids = AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 3) sys.argv = make_argv([str(ids[1])]) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 1)
def test_ref_undoes_all_alters_including_ref(self): AlterUtil.create_alters([1, 2, 3]) ids = AlterUtil.run_alters() self.assertEqual(len(MemoryDb.data), 3) sys.argv = make_argv([str(ids[1])]) self.downCommand.run() self.assertEqual(len(MemoryDb.data), 1)
def test_order_reverse(self): id1, id2 = AlterUtil.create_alters([1, 2]) AlterUtil.run_alters() sys.argv = make_argv(['-r']) result = self.listCommand.run() self.assertEqual([id2, id1], result)
def test_valid_chain(self): AlterUtil.create_alters([1, 2]) AlterUtil.run_alters() sys.argv = make_argv([]) self.checkCommand.run()