def test_0001(self, mock_get_backup_list, mock_get_system_ip, mock_remove_file): """ Test with patter 100 backups """ data = self._genBackupList(100) named = [x['file'] for x in sorted(data, key=lambda x: x['timestamp'])] mock_get_backup_list.return_value = (True, data) mock_get_system_ip.return_value = (True, "192.168.1.1") mock_remove_file.return_value = (True, '') success, result = rotate_backups('local', 10) self.assertEqual(success, True) self.assertEqual(result, "Remove 90 backups") self.assertEqual(mock_remove_file.called, True) # Per algoritm backups mas be present backups # [18, 30, 44, 59, 70, 84, 92, 96, 98, 99]. self.assertEqual(len(mock_remove_file.call_args_list), 1) firstcall = mock_remove_file.call_args_list[0] self.assertEqual(firstcall[0][0], ['192.168.1.1']) deletedone = firstcall[0][1].split(" ") self.assertEqual(len(deletedone), 90) test_vector = [18, 30, 44, 59, 70, 84, 92, 96, 98, 99] for fname in deletedone: index = named.index(fname) self.assertTrue(index not in test_vector)
def test_0002(self, mock_get_backup_list, mock_get_system_ip, mock_remove_file): """ 10 backups must no fired part of the code """ data = self._genBackupList(10) # ordered = sorted(data, key=lambda x: x['timestamp']) mock_get_backup_list.return_value = (True, data) mock_get_system_ip.return_value = (True, "192.168.1.1") mock_remove_file.return_value = (True, '') success, _ = rotate_backups('local', 10) self.assertEqual(success, True) self.assertEqual(mock_remove_file.called, False)