class TestSavingPasswordData(unittest.TestCase): def test_FullFunctionality(self): print("Running the FullFunctionality test for the TestSavingPasswordData.") self.gen = SimpleLabelledKeyGenerator() self.pickler = PicklingMethodology() self.data = Definitions() lclFactory = FactoryProgramPasswordData() self.passwordData = lclFactory.build() #self.passwordData.addPassword(self.lclpwd1) #self.passwordData.addPassword(self.lclpwd2) self.passwordData.addPassword(self.data.lclpwd3) print("Status: " + self.passwordData.toString()) self.passwordData.addInput(self.data.passwordListA2) print("Status: " + self.passwordData.toString()) self.passwordData.addPassword(self.data.lclpwd4) print("Status: " + self.passwordData.toString()) self.passwordData.addPassword(self.data.lclpwd4d) print("Status: " + self.passwordData.toString()) self.passwordData.addInputWithPasswords(self.data.passwordListB4) print("Status: " + self.passwordData.toString()) self.pickler.saveVault(self.passwordData, "ExampleVault.vlt")
class ProgramState(object): ''' classdocs ''' def __init__(self): ''' Constructor ''' self.vaultFileName = "" self.passwordData = ProgramPasswordData() self.pickler = PicklingMethodology() # def isOpen(self): # if -1 in self.list.identifier(): # return True # return False def addPassword(self, pPasswordTuple): self.passwordData.addPassword(pPasswordTuple) def removePassword(self, pPasswordTuple): self.passwordData.removePassword(pPasswordTuple) # Remove keys associated with password pair? def addInputPlusPasswords(self, pInputList): self.passwordData.addInputWithPasswords(pInputList) #self.addInput(pInputList) #for i in pInputList.getList(): # self.addPassword(i) def addInput(self, pInputList): self.passwordData.addInput(pInputList) def setFileName(self, pFileName): self.vaultFileName = pFileName def saveVault(self): self.pickler.saveVault(self.passwordData, self.vaultFileName) def loadVault(self): #Check for malicious pickling return self.pickler.getVaultFromFile(self.vaultFileName) def removeInput(self, pInputList): self.passwordData.removeInput(pInputList) def displayedPasswords(self): lclPasswordList = self.passwordData.getPasswordList() del lclPasswordList["Intermediate Constant: 3161379751047332383565"] return lclPasswordList def toString(self): print("Password Data contents:") print(self.vault.toString()) #print("The recovered password list:") #print(self.recoveredList())
def __init__(self): ''' Constructor ''' self.vaultFileName = "" self.passwordData = ProgramPasswordData() self.pickler = PicklingMethodology()
def test_FullFunctionality(self): self.data = Definitions() self.pickler = PicklingMethodology() lclVaultXX = self.pickler.getVaultFromFile("ExampleVault.vlt") print(lclVaultXX.toString()) lclResults = lclVaultXX.recover(self.data.passwordListA3) print(lclResults)