def setUp(self):
self.fb = FbQuery()
class PasswordTests(unittest.TestCase):
def setUp(self):
self.fb = FbQuery()
# User tests
def testUserCreate(self):
self.assertTrue(self.fb.user_create(
'testuser',
'randomPassword'
))
self.fb.user_delete('testuser')
def testUserExists(self):
self.fb.user_create('testUser', 'randomPassword')
self.assertTrue(self.fb.user_exist('testUser'))
self.fb.user_delete('testUser')
def testUserExistsInvalid(self):
self.assertFalse(self.fb.user_exist('invalidUser'))
def testChangePassword(self):
self.fb.user_create('testuser', 'randomPassword')
self.assertTrue(self.fb.user_change_password('testuser', 'somepassword'))
self.fb.user_delete('testuser')
def testChangePasswordNonExistingUser(self):
self.assertFalse(self.fb.user_change_password('jkhkjhkjh', 'somepassword'))
def testUserDelete(self):
self.fb.user_create('userDeleteTest', 'randomPassword')
self.assertTrue(self.fb.user_delete('userDeleteTest'))
def testUserDeleteNonExisting(self):
self.assertFalse(self.fb.user_delete('non-existing-user'))
def testUserGet(self):
u = 'testuser'
self.fb.user_create(u, 'randomPassword')
user = self.fb.user_get(u)
self.assertEqual(user.username, u)
self.fb.user_delete(u)
def testUserGetNonExisting(self):
self.assertIsNone(self.fb.user_get('unkownUser'))
def testUserListFiles(self):
user = '******'
file_hash = '1337'
file_name = 'random_file_name'
self.fb.user_create(user, 'password')
user_data = self.fb.user_get(user)
self.fb.file_add(file_hash, user_data.id, file_name, 0, False, False)
data = self.fb.user_list_files(user)
self.assertEqual(data['message'][file_hash]['filename'], file_name)
self.fb.file_remove(file_hash, file_name)
self.fb.user_delete(user)
def testUserNonExistingListFiles(self):
self.assertIsNone(self.fb.user_list_files('invalid-user'))
# File tests
def testFileAdd(self):
self.assertTrue(self.fb.file_add('123', 1, 'testfile', 0, False, False))
def testFileGetNoneExisting(self):
self.assertIsNone(self.fb.file_get('unkown_hash'))
def testSetFileExpiry(self):
test_date = '2000-01-01 00:00:00'
hash = '8765'
self.fb.file_add(hash, 1, 'testfile', 0, False, False)
self.fb.file_set_expiry(hash, test_date)
f = self.fb.file_get(hash)
self.assertEqual(f.expire, test_date)
self.fb.file_remove(hash, 'testfile')
def testFileDelete(self):
self.assertTrue(self.fb.file_remove('123', 'testfile'))
def testFileDeleteUnknownFile(self):
self.assertFalse(self.fb.file_remove('9999', 'unknownFile'))
def testExpiredFile(self):
date_from_the_past = '20110101010101'
self.assertTrue(self.fb.file_expired(date_from_the_past))
def testFileExpiredValueError(self):
invalid_date = 'abc'
self.assertRaises(self.fb.file_expired(invalid_date))
def testNonExpiredFile(self):
date_in_the_future = '20700101010101'
self.assertFalse(self.fb.file_expired(date_in_the_future))
def testFileRemoveExpired(self):
expire_date = datetime.now().strftime('%Y%m%d%H%M%S')
hash = 'SJG8F90G8S09'
testfile = 'testFileRemoveExpired'
self.fb.user_create('testuser', 'password')
self.fb.file_add(hash, 1, testfile, expire_date, False, False)
self.assertEqual(testfile, self.fb.file_remove_expired()['message'][hash]['filename'])
self.fb.file_remove(hash, testfile)
self.fb.user_delete('testuser')
