class UserTest(unittest.TestCase):
def setUp(self):
initdb()
def tearDown(self):
try:
os.remove("sqlite3.db")
except OSError:
pass
def test_valid__returnFalse(self):
self.user = Service()
self.assertFalse(self.user.valid())
def test_valid_user_returnFalse(self):
self.user = Service()
self.user.name = "John"
self.assertFalse(self.user.valid())
def test_valid_userPass_returnTrue(self):
self.user = Service()
self.user.name = 'John'
self.user.password = '******'
self.assertTrue(self.user.valid())
def test_valid_userPassKWARGS_returnTrue(self):
self.user = Service(_name='John', _password='******')
self.assertTrue(self.user.valid())
def import_data(path='unipass_export.json'):
initdb() # If db not exists, create one
with open(path, 'r') as fp:
try:
for s in json.load(fp):
service = Service(**s)
if service.valid():
service.create()
return True
except ValueError:
return False
def add_service(service, name, password, note):
serv = Service()
serv.service = service
serv.name = name
serv.password = password
serv.note = note
if serv.valid():
serv.create()
return True
else:
return False
def create_user(username, password):
user = Service()
user.name = username
user.password = password
user.service = 'UniPass'
user.note = 'Entry for UniPass'
user.admin = True
if user.valid():
user.create()
return True
else:
return False
class DBTest(unittest.TestCase):
def setUp(self):
initdb()
def tearDown(self):
try:
os.remove("sqlite3.db")
except OSError:
pass
def test_create_none(self):
self.user = Service()
self.user.admin = True
self.user.name = 'John'
self.user.password = '******'
self.user.create()
self.assertTrue(self.user.valid())
self.assertTrue(self.user in Service.getall())
def test_update_none(self):
self.user = Service()
self.user.admin = True
self.user.name = 'John'
self.user.password = '******'
self.user.create()
self.user = Service.getbyuuid(self.user._uuid)
self.user.name = 'Kalle'
self.user.update()
self.assertTrue(self.user in Service.getall())
def test_delete_none(self):
self.user = Service()
self.user.admin = True
self.user.name = 'John'
self.user.password = '******'
self.user.create()
self.user.delete()
self.assertFalse(self.user in Service.getall())