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 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())
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 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 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
def export_data(path='unipass_export.json'): try: with open(path, 'w') as outfile: json.dump([s.__dict__ for s in Service.getall()], outfile, indent=4) return True except: 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 update_service(uuid, service, name, password, note): serv = Service.getbyuuid(uuid) serv.service = service serv.name = name serv.password = password serv.note = note if serv.valid(): serv.update() return True else: return False
def test_valid_user_returnFalse(self): self.user = Service() self.user.name = "John" self.assertFalse(self.user.valid())
def get_service_by_uuid(uuid): """Get service by uuid.""" return Service.getbyuuid(uuid)
def list_all_services(): """List all services.""" return [(serv.service, serv.name, serv.uuid) for serv in Service.getall()]
def list_all_services(): return [(serv.service, serv.name, serv.uuid) for serv in Service.getall()]
def test_valid_userPassKWARGS_returnTrue(self): self.user = Service(_name='John', _password='******') self.assertTrue(self.user.valid())
def test_valid_userPass_returnTrue(self): self.user = Service() self.user.name = 'John' self.user.password = '******' self.assertTrue(self.user.valid())
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())
def find_by_service(service): """Find service.""" for serv in Service.getall(): if serv.service == service: return serv return None
def delete_service(uuid): serv = Service.getbyuuid(uuid) serv.delete() return True
def delete_service(uuid): """Delete service.""" serv = Service.getbyuuid(uuid) serv.delete() return True
def get_service_by_uuid(uuid): return Service.getbyuuid(uuid)
def login(username, password): """User login.""" for user in Service.getall(): if (user.name == username and user.password == password) and user.admin: return True return False
def test_valid__returnFalse(self): self.user = Service() self.assertFalse(self.user.valid())