def test_assign_ip(self, m_allocate, m_pools):
     """
     Test assign_ip assigns an IP address.
     """
     m_pools.return_value = [IPNetwork("1.2.3.0/24"), IPNetwork("2.3.4.5/32")]
     m_allocate.return_value = IPAddress("1.2.3.6")
     ip = docker_plugin.assign_ip(4)
     assert_equal(ip, IPNetwork("1.2.3.6"))
     m_pools.assert_called_once_with(4)
     m_allocate.assert_called_once_with(ANY, IPNetwork("1.2.3.0/24"))
Exemplo n.º 2
0
 def test_assign_ip(self, m_allocate, m_pools):
     """
     Test assign_ip assigns an IP address.
     """
     m_pools.return_value = [IPNetwork("1.2.3.0/24"), IPNetwork("2.3.4.5/32")]
     m_allocate.return_value = IPAddress("1.2.3.6")
     ip = docker_plugin.assign_ip(4)
     assert_equal(ip, IPNetwork("1.2.3.6"))
     m_pools.assert_called_once_with(4)
     m_allocate.assert_called_once_with(ANY, IPNetwork("1.2.3.0/24"))
    def test_assign_ip_no_ip(self, m_allocate, m_pools):
        """
        Test assign_ip when no IP addresses can be allocated.
        """
        m_pools.return_value = [IPNetwork("1.2.3.0/24"), IPNetwork("2.3.4.5/32")]
        m_allocate.return_value = None
        ip = docker_plugin.assign_ip(4)
        assert_equal(ip, None)
        m_pools.assert_called_once_with(4)

        # We should have attempted to allocate for each pool.
        m_allocate.assert_has_calls([call(ANY, IPNetwork("1.2.3.0/24")), call(ANY, IPNetwork("2.3.4.5/32"))])
Exemplo n.º 4
0
    def test_assign_ip_no_ip(self, m_allocate, m_pools):
        """
        Test assign_ip when no IP addresses can be allocated.
        """
        m_pools.return_value = [IPNetwork("1.2.3.0/24"),
                                IPNetwork("2.3.4.5/32")]
        m_allocate.return_value = None
        ip = docker_plugin.assign_ip(4)
        assert_equal(ip, None)
        m_pools.assert_called_once_with(4)

        # We should have attempted to allocate for each pool.
        m_allocate.assert_has_calls([call(ANY, IPNetwork("1.2.3.0/24")),
                                     call(ANY, IPNetwork("2.3.4.5/32"))])