def test_getProfileByID(self): db = PSTMirrorDB() # 821 - pst_id for pmargani globalId = 821 info = db.getStaticContactInfoByID(globalId) # expected values emails = ['*****@*****.**' , '*****@*****.**' , '*****@*****.**'] emailDescs = ['[email protected] (Work)' , '[email protected] (Other)' , '[email protected] (Personal)'] phones = ['304-456-2202', '304-456-2206'] phoneDescs = ['304-456-2202 (Work)', '304-456-2206 (Other)'] postals = \ ['NRAO, PO Box 2, Green Bank, West Virginia, 24944, USA, (Office)' , '49 columbus Ave., W. Bridgewater, Massachusetts, 02379, United States, (Other)'] affiliations = [u'National Radio Astronomy Observatory' , u'Oregon, University of'] self.assertEquals(emails, info['emails']) self.assertEquals(emailDescs, info['emailDescs']) self.assertEquals(phones, info['phones']) self.assertEquals(phoneDescs, info['phoneDescs']) self.assertEquals(postals, info['postals']) self.assertEquals(affiliations, info['affiliations']) self.assertEquals('pmargani', info['username']) self.assertEquals(True, info['status']) self.assertEquals(globalId, info['person_id']) self.assertEquals(823, info['personAuth_id'])
def test_getUserAuthenticationIdFromId(self): db = PSTMirrorDB() globalId = 821 userAuthId = 823 username = '******' id = db.getUserAuthenticationIdFromId(globalId) self.assertEquals(userAuthId, id)
def test_getIdFromUsername(self): db = PSTMirrorDB() globalId = 821 username = '******' id = db.getIdFromUsername(username) self.assertEquals(globalId, id) un = db.getUsername(globalId) self.assertEquals(username, un)
from nell.utilities.database.external import UserInfo, PSTMirrorDB from scheduler.models import * # Utility for taking in a list of usernames from computing, and # seeing who should be deleted and who shouldn't pst = PSTMirrorDB() filename = 'oldUsers.txt' f = open(filename, 'r') lines = f.readlines() notInPst = [] notInDss = [] scheduled = [] notScheduled = [] notComplete = [] for l in lines: username = l.split(" ")[0].strip() pst_id = None u = None try: pst_id = pst.getIdFromUsername(username) except: pst_id = None notInPst.append(username) if pst_id is not None: try: u = User.objects.get(pst_id = pst_id) except:
def test_getBadProfile(self): "Make sure we can handle bogus info." db = PSTMirrorDB() username = db.getUsername(0) self.assertEquals(None, username)