def test_private_link_response(self):
        discovery = HazelcastCloudDiscovery(
            *get_params(HOST, self.server.port, CLOUD_URL, PRIVATE_LINK_TOKEN))
        discovery._ctx = self.ctx
        addresses = discovery.discover_nodes()

        six.assertCountEqual(self, PRIVATE_LINK_ADDRESSES, addresses)
    def test_found_response(self):
        discovery = HazelcastCloudDiscovery(
            *get_params(HOST, self.server.port, CLOUD_URL, TOKEN))
        discovery._ctx = self.ctx
        addresses = discovery.discover_nodes()

        six.assertCountEqual(self, ADDRESSES, addresses)
 def test_invalid_url(self):
     discovery = HazelcastCloudDiscovery(
         *get_params(HOST, self.server.port, "/INVALID_URL", ""))
     discovery._ctx = self.ctx
     with self.assertRaises(IOError):
         discovery.discover_nodes()
 def test_not_found_response(self):
     discovery = HazelcastCloudDiscovery(
         *get_params(HOST, self.server.port, CLOUD_URL, "INVALID_TOKEN"))
     discovery._ctx = self.ctx
     with self.assertRaises(IOError):
         discovery.discover_nodes()