def setUp(self):
        from letsencrypt.auth_handler import challb_to_achall
        from letsencrypt.auth_handler import AuthHandler

        # Account and network are mocked...
        self.mock_net = mock.MagicMock()
        self.handler = AuthHandler(
            None, None, self.mock_net, mock.Mock(key="mock_key"))

        self.doms = ["0", "1", "2"]
        self.handler.authzr[self.doms[0]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[0],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.handler.authzr[self.doms[1]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[1],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.handler.authzr[self.doms[2]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[2],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.chall_update = {}
        for dom in self.doms:
            self.chall_update[dom] = [
                challb_to_achall(challb, "dummy_key", dom)
                for challb in self.handler.authzr[dom].body.challenges]
    def setUp(self):
        from letsencrypt.auth_handler import challb_to_achall
        from letsencrypt.auth_handler import AuthHandler

        # Account and network are mocked...
        self.mock_net = mock.MagicMock()
        self.handler = AuthHandler(
            None, None, self.mock_net, mock.Mock(key="mock_key"))

        self.doms = ["0", "1", "2"]
        self.handler.authzr[self.doms[0]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[0],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.handler.authzr[self.doms[1]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[1],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.handler.authzr[self.doms[2]] = acme_util.gen_authzr(
            messages.STATUS_PENDING, self.doms[2],
            acme_util.DV_CHALLENGES, [messages.STATUS_PENDING]*3, False)

        self.chall_update = {}
        for dom in self.doms:
            self.chall_update[dom] = [
                challb_to_achall(challb, mock.Mock(key="dummy_key"), dom)
                for challb in self.handler.authzr[dom].body.challenges]
 def test_unable_to_find_challenge_status(self, unused_mock_time):
     from letsencrypt.auth_handler import challb_to_achall
     self.mock_net.poll.side_effect = self._mock_poll_solve_one_valid
     self.chall_update[self.doms[0]].append(
         challb_to_achall(acme_util.RECOVERY_CONTACT_P, "key", self.doms[0]))
     self.assertRaises(
         errors.AuthorizationError,
         self.handler._poll_challenges, self.chall_update, False)
 def test_unable_to_find_challenge_status(self, unused_mock_time):
     from letsencrypt.auth_handler import challb_to_achall
     self.mock_net.poll.side_effect = self._mock_poll_solve_one_valid
     self.chall_update[self.doms[0]].append(
         challb_to_achall(acme_util.RECOVERY_CONTACT_P, "key", self.doms[0]))
     self.assertRaises(
         errors.AuthorizationError, self.handler._poll_challenges,
         self.chall_update, False)
 def _call(self, challb):
     from letsencrypt.auth_handler import challb_to_achall
     return challb_to_achall(challb, "account_key", "domain")