Example #1
0
    def test_allocation_error(self, kwargs, mocker):
        """Test an error when no TCP port can be allocated from the given range"""
        mock_socket = mocker.patch("socket.socket")
        mock_socket.return_value.connect_ex.return_value = 0  # any port is unavailable

        pattern = "Cannot allocate an open TCP port for Kedro-Viz"
        with pytest.raises(ValueError, match=re.escape(pattern)):
            _allocate_port(**kwargs)
Example #2
0
    def test_allocate_from_viz_processes(self, mocker, viz_processes, kwargs,
                                         expected_port):
        """Test allocation of the port from the one that was already captured
        in _VIZ_PROCESSES"""
        mocker.patch.dict("kedro_viz.server._VIZ_PROCESSES",
                          {k: None
                           for k in viz_processes})

        allocated_port = _allocate_port(**kwargs)
        assert allocated_port == expected_port
Example #3
0
    def test_allocate_from_available_ports(self, mocker, kwargs,
                                           expected_port):
        """Test allocation of one of unoccupied ports"""
        def _mock_even_port_available(host_and_port):
            # Mock availability of every even port number
            port = host_and_port[1]
            return 1 - port % 2

        mock_socket = mocker.patch("socket.socket")
        mock_socket.return_value.connect_ex.side_effect = _mock_even_port_available

        allocated_port = _allocate_port(**kwargs)
        assert allocated_port == expected_port