def test_csv(self): init_stat(self.cur) rows = get_data(config.TEST_FTP_URI) logs = {} cnt = 0 for row in rows: cnt += 1 self.assertIsInstance(row[0], datetime) self.assertIsInstance(row[1], str) logs[row[1]] = row[0] if cnt == 3: self.assertFalse(check_passwd(row[2])) else: self.assertTrue(check_passwd(row[2])) save_log(logs, config.TEST_LOG) with open(config.TEST_LOG) as f: x = f.read() self.assertEqual(x, 'user2\nuser3\nuser1') f.closed
def test_passwd(self): """ Test good/bad password """ self.assertTrue(check_passwd('11Qaaa')) self.assertFalse(check_passwd('aaaQaaa')) self.assertFalse(check_passwd('111111')) self.assertFalse(check_passwd('111aaa')) self.assertFalse(check_passwd('1aQaa')) self.assertFalse(check_passwd('1aQ'))