def test_delete_private_key_does(self): self.assertEquals(len(OneTimePad.objects.all()), 0) sheet = Sheet() user = User(username='******') user.save() sheet.owner = user sheet.create_private_key() self.assertEquals(len(OneTimePad.objects.all()), 1) sheet._delete_private_key() self.assertEquals(len(OneTimePad.objects.all()), 0)
def test_create_private_key_uses_onetimepad(self): sheet = Sheet() sheet.version = 155 user = User(username='******') user.set_password('glurk') user.save() sheet.owner = user self.assertEquals(len(OneTimePad.objects.all()), 0) self.assertEquals( sheet.create_private_key(), OneTimePad.objects.get(user=sheet.owner).guid) sheet._delete_private_key()
def test_calculate_always_deletes_private_key_in_finally_block( self, mock_calculate ): def raiser(*a, **kw): raise Exception() mock_calculate.side_effect = raiser sheet = Sheet() user = User(username='******') user.save() sheet.owner = user sheet._delete_private_key = Mock() self.assertRaises(Exception, sheet.calculate) self.assertCalledOnce(sheet._delete_private_key)