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_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_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)
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)
