def test_gen_user_secret(self): """Generation of a user secret which randomly salts and hashes.""" secret_hasher = Mock() secret_hasher.digest_size = 3 when(secret_hasher).hexdigest().thenReturn('ababa') secret_generator = secrets.SecretGenerator(lambda: secret_hasher, lambda x: 'aaa') self.assertEqual(secret_generator.gen_user_secret(1), u'ababa') verify(secret_hasher).update(u'aaa#1')
def test_gen_for_two_users(self): """Generation of a user secret from two user ids.""" secret_hasher = Mock() secret_hasher.digest_size = 3 secret_generator = secrets.SecretGenerator(lambda: secret_hasher, lambda x: 'aaa') when(secret_hasher).hexdigest().thenReturn('ababa') self.assertEqual(secret_generator.gen_user_secret(1), u'ababa') verify(secret_hasher).update(u'aaa#1') when(secret_hasher).hexdigest().thenReturn('bcax') self.assertEqual(secret_generator.gen_user_secret(2), u'bcax') verify(secret_hasher).update(u'aaa#2')