示例#1
0
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())
示例#2
0
 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())
示例#3
0
 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())
示例#4
0
 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())
示例#5
0
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
示例#6
0
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
示例#7
0
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
示例#8
0
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
示例#9
0
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
示例#10
0
 def test_valid_user_returnFalse(self):
     self.user = Service()
     self.user.name = "John"
     self.assertFalse(self.user.valid())
示例#11
0
def get_service_by_uuid(uuid):
    """Get service by uuid."""
    return Service.getbyuuid(uuid)
示例#12
0
def list_all_services():
    """List all services."""
    return [(serv.service, serv.name, serv.uuid) for serv in Service.getall()]
示例#13
0
def list_all_services():
    return [(serv.service, serv.name, serv.uuid) for serv in Service.getall()]
示例#14
0
 def test_valid_userPassKWARGS_returnTrue(self):
     self.user = Service(_name='John', _password='******')
     self.assertTrue(self.user.valid())
示例#15
0
 def test_valid_userPass_returnTrue(self):
     self.user = Service()
     self.user.name = 'John'
     self.user.password = '******'
     self.assertTrue(self.user.valid())
示例#16
0
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())
示例#17
0
def find_by_service(service):
    """Find service."""
    for serv in Service.getall():
        if serv.service == service:
            return serv
    return None
示例#18
0
def delete_service(uuid):
    serv = Service.getbyuuid(uuid)
    serv.delete()
    return True
示例#19
0
def delete_service(uuid):
    """Delete service."""
    serv = Service.getbyuuid(uuid)
    serv.delete()
    return True
示例#20
0
def get_service_by_uuid(uuid):
    return Service.getbyuuid(uuid)
示例#21
0
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
示例#22
0
 def test_valid__returnFalse(self):
     self.user = Service()
     self.assertFalse(self.user.valid())