class Basic(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
nl_class = IPRoute
nl_kwarg = {}
ssh = ''
ipnets = []
ipranges = []
def ifaddr(self):
return str(self.ipranges[0].pop())
def ifname(self):
ret = uifname()
self.interfaces.append(ret)
return ret
def setup(self):
require_user('root')
self.interfaces = []
self.ipnets = [allocate_network() for _ in range(2)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
def teardown(self):
with self.nl_class(**self.nl_kwarg) as ipr:
for link in reversed(self.interfaces):
try:
ipr.link('del', index=ipr.link_lookup(ifname=link)[0])
except Exception:
pass
self.ndb.close()
for net in self.ipnets:
free_network(net)
class Basic(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
nl_class = IPRoute
nl_kwarg = {}
ssh = ''
ipnets = []
ipranges = []
def ifaddr(self):
return str(self.ipranges[0].pop())
def ifname(self):
ret = uifname()
self.interfaces.append(ret)
return ret
def setup(self):
require_user('root')
self.interfaces = []
self.ipnets = [allocate_network() for _ in range(2)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
def teardown(self):
with self.nl_class(**self.nl_kwarg) as ipr:
for link in reversed(self.interfaces):
try:
ipr.link('del', index=ipr.link_lookup(ifname=link)[0])
except Exception:
pass
self.ndb.close()
for net in self.ipnets:
free_network(net)
class TestRollback(TestBase):
def setup(self):
require_user('root')
self.ipnets = [allocate_network() for _ in range(5)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
self.interfaces = []
def test_simple_deps(self):
# register NDB handler to wait for the interface
self.if_simple = uifname()
ifaddr = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
#
# simple dummy interface with one address and
# one dependent route
#
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_simple, kind='dummy')
.set('state', 'up')
.commit()['index']))
(self
.ndb
.addresses
.create(address=ifaddr,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
iface = self.ndb.interfaces[self.if_simple]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert not grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
def test_bridge_deps(self):
self.if_br0 = uifname()
self.if_br0p0 = uifname()
self.if_br0p1 = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0,
kind='bridge',
state='up')
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0p0,
kind='dummy',
state='up',
master=self.ndb.interfaces[self.if_br0]['index'])
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0p1,
kind='dummy',
state='up',
master=self.ndb.interfaces[self.if_br0]['index'])
.commit()['index']))
(self
.ndb
.interfaces[self.if_br0]
.ipaddr
.create(address=ifaddr1, prefixlen=24)
.commit())
(self
.ndb
.interfaces[self.if_br0]
.ipaddr
.create(address=ifaddr2, prefixlen=24)
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
master = self.ndb.interfaces[self.if_br0]['index']
self.ndb.interfaces.wait(ifname=self.if_br0p0, master=master)
self.ndb.interfaces.wait(ifname=self.if_br0p1, master=master)
self.ndb.addresses.wait(address=ifaddr1)
self.ndb.addresses.wait(address=ifaddr2)
self.ndb.routes.wait(dst=dst, gateway=router)
iface = self.ndb.interfaces[self.if_br0]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert not grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
def test_vlan_deps(self):
if_host = uifname()
if_vlan = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=if_host,
kind='dummy',
state='up')
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=if_vlan,
kind='vlan',
link=self.interfaces[-1],
state='up',
vlan_id=1001)
.commit()['index']))
(self
.ndb
.addresses
.create(address=ifaddr1,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.addresses
.create(address=ifaddr2,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
iface = self.ndb.interfaces[if_host]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=if_host)
assert grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert grep('%s cat /proc/net/vlan/config' % self.ssh, pattern=if_vlan)
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=if_host)
assert not grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert not grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert not grep('%s cat /proc/net/vlan/config' % self.ssh,
pattern=if_vlan)
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=if_host)
assert grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert grep('%s cat /proc/net/vlan/config' % self.ssh, pattern=if_vlan)
class TestBase(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
nl_class = IPRoute
nl_kwarg = {}
ssh = ''
ipnets = []
ipranges = []
def create_interfaces(self):
# dummy interface
if_dummy = uifname()
if_vlan_stag = uifname()
if_vlan_ctag = uifname()
if_bridge = uifname()
if_port = uifname()
if_addr1 = self.ifaddr()
if_addr2 = self.ifaddr()
ret = []
ret.append(self
.ndb
.interfaces
.create(ifname=if_dummy, kind='dummy')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_vlan_stag,
link=self.ndb.interfaces[if_dummy]['index'],
vlan_id=101,
vlan_protocol=0x88a8,
kind='vlan')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_vlan_ctag,
link=self.ndb.interfaces[if_vlan_stag]['index'],
vlan_id=1001,
vlan_protocol=0x8100,
kind='vlan')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_bridge, kind='bridge')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_port,
master=self.ndb.interfaces[if_bridge]['index'],
kind='dummy')
.commit()['index'])
(self
.ndb
.interfaces[if_bridge]
.ipaddr
.create(address=if_addr1, prefixlen=24)
.commit())
(self
.ndb
.interfaces[if_bridge]
.ipaddr
.create(address=if_addr2, prefixlen=24)
.commit())
self.if_bridge = if_bridge
return ret
def ifaddr(self, r=0):
return str(self.ipranges[r].pop())
def setup(self):
require_user('root')
self.if_simple = None
self.ipnets = [allocate_network() for _ in range(5)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
self.interfaces = self.create_interfaces()
def teardown(self):
with self.nl_class(**self.nl_kwarg) as ipr:
for link in reversed(self.interfaces):
ipr.link('del', index=link)
self.ndb.close()
for net in self.ipnets:
free_network(net)
def fetch(self, request, values=[]):
return (self
.ndb
.schema
.fetch(request, values))
class TestNetNS(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
def setup(self):
require_user('root')
self.netns = str(uuid.uuid4())
self.ipnets = [allocate_network() for _ in range(3)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.sources = [{'target': 'localhost'},
{'netns': self.netns}]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
sources=self.sources,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
def ifaddr(self, r=0):
return str(self.ipranges[r].pop())
def teardown(self):
for net in self.ipnets:
free_network(net)
self.ndb.close()
netns.remove(self.netns)
def test_basic(self):
ifname = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
ifaddr3 = self.ifaddr()
(self
.ndb
.interfaces
.create(target=self.netns, ifname=ifname, kind='dummy')
.ipaddr
.create(address=ifaddr1, prefixlen=24)
.create(address=ifaddr2, prefixlen=24)
.create(address=ifaddr3, prefixlen=24)
.commit())
with NDB(sources=[{'target': 'localhost',
'netns': self.netns,
'kind': 'netns'}]) as ndb:
if_idx = ndb.interfaces[ifname]['index']
addr1_idx = ndb.addresses['%s/24' % ifaddr1]['index']
addr2_idx = ndb.addresses['%s/24' % ifaddr2]['index']
addr3_idx = ndb.addresses['%s/24' % ifaddr3]['index']
assert if_idx == addr1_idx == addr2_idx == addr3_idx
def _assert_test_view(self, ifname, ifaddr):
with NDB(sources=[{'target': 'localhost',
'netns': self.netns,
'kind': 'netns'}]) as ndb:
(if_idx,
if_state,
if_addr,
if_flags) = ndb.interfaces[ifname].fields('index',
'state',
'address',
'flags')
addr_idx = ndb.addresses['%s/24' % ifaddr]['index']
assert if_idx == addr_idx
assert if_state == 'up'
assert if_flags & 1
assert if_addr == '00:11:22:33:44:55'
def test_view_constraints_pipeline(self):
ifname = uifname()
ifaddr = self.ifaddr()
(self
.ndb
.interfaces
.constraint('target', self.netns)
.create(ifname=ifname, kind='dummy')
.set('address', '00:11:22:33:44:55')
.set('state', 'up')
.ipaddr
.create(address=ifaddr, prefixlen=24)
.commit())
self._assert_test_view(ifname, ifaddr)
def test_view_constraints_cmanager(self):
ifname = uifname()
ifaddr = self.ifaddr()
with self.ndb.interfaces as view:
view.constraints['target'] = self.netns
with view.create(ifname=ifname, kind='dummy') as interface:
interface['address'] = '00:11:22:33:44:55'
interface['state'] = 'up'
with interface.ipaddr as aview:
with aview.create(address=ifaddr, prefixlen=24):
pass
self._assert_test_view(ifname, ifaddr)
def test_move(self):
ifname = uifname()
ifaddr = self.ifaddr()
# create the interfaces
(self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy')
.commit())
# move it to a netns
(self
.ndb
.interfaces[ifname]
.set('net_ns_fd', self.netns)
.commit())
# setup the interface only when it is moved
(self
.ndb
.interfaces
.wait(target=self.netns, ifname=ifname)
.set('state', 'up')
.set('address', '00:11:22:33:44:55')
.ipaddr
.create(address=ifaddr, prefixlen=24)
.commit())
self._assert_test_view(ifname, ifaddr)
class TestCreate(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
nl_class = IPRoute
nl_kwarg = {}
ssh = ''
ipnets = []
ipranges = []
def ifaddr(self):
return str(self.ipranges[0].pop())
def ifname(self):
ret = uifname()
self.interfaces.append(ret)
return ret
def setup(self):
require_user('root')
self.interfaces = []
self.ipnets = [allocate_network() for _ in range(2)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
def teardown(self):
with self.nl_class(**self.nl_kwarg) as ipr:
for link in reversed(self.interfaces):
ipr.link('del', index=ipr.link_lookup(ifname=link)[0])
self.ndb.close()
for net in self.ipnets:
free_network(net)
def test_context_manager(self):
ifname = uifname()
address = '00:11:22:36:47:58'
ifobj = (self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy'))
with ifobj:
pass
assert grep('%s ip link show' % self.ssh, pattern=ifname)
with ifobj:
ifobj['state'] = 'up'
ifobj['address'] = address
assert grep('%s ip link show' % self.ssh, pattern=address)
assert self.ndb.interfaces[ifname]['state'] == 'up'
with ifobj:
ifobj.remove()
def test_fail(self):
ifname = uifname()
kind = uifname()
ifobj = (self
.ndb
.interfaces
.create(ifname=ifname, kind=kind))
save = dict(ifobj)
try:
ifobj.commit()
except NetlinkError as e:
assert e.code == 95 # Operation not supported
assert save == dict(ifobj)
assert ifobj.state == 'invalid'
def test_dummy(self):
ifname = self.ifname()
(self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy', address='00:11:22:33:44:55')
.commit())
assert grep('%s ip link show' % self.ssh, pattern=ifname)
assert self.ndb.interfaces[ifname]['address'] == '00:11:22:33:44:55'
def test_bridge(self):
bridge = self.ifname()
brport = self.ifname()
(self
.ndb
.interfaces
.create(ifname=bridge, kind='bridge')
.commit())
(self
.ndb
.interfaces
.create(ifname=brport, kind='dummy')
.set('master', self.ndb.interfaces[bridge]['index'])
.commit())
assert grep('%s ip link show' % self.ssh,
pattern=bridge)
assert grep('%s ip link show' % self.ssh,
pattern='%s.*%s' % (brport, bridge))
def test_vrf(self):
vrf = self.ifname()
(self
.ndb
.interfaces
.create(ifname=vrf, kind='vrf')
.set('vrf_table', 42)
.commit())
assert grep('%s ip link show' % self.ssh, pattern=vrf)
def test_vlan(self):
host = self.ifname()
vlan = self.ifname()
(self
.ndb
.interfaces
.create(ifname=host, kind='dummy')
.commit())
(self
.ndb
.interfaces
.create(ifname=vlan, kind='vlan')
.set('link', self.ndb.interfaces[host]['index'])
.set('vlan_id', 101)
.commit())
assert grep('%s ip link show' % self.ssh, pattern=vlan)
def test_vxlan(self):
host = self.ifname()
vxlan = self.ifname()
(self
.ndb
.interfaces
.create(ifname=host, kind='dummy')
.commit())
(self
.ndb
.interfaces
.create(ifname=vxlan, kind='vxlan')
.set('vxlan_link', self.ndb.interfaces[host]['index'])
.set('vxlan_id', 101)
.set('vxlan_group', '239.1.1.1')
.set('vxlan_ttl', 16)
.commit())
assert grep('%s ip link show' % self.ssh, pattern=vxlan)
def test_basic_address(self):
ifaddr = self.ifaddr()
ifname = self.ifname()
i = (self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy', state='up'))
i.commit()
a = (self
.ndb
.addresses
.create(index=i['index'],
address=ifaddr,
prefixlen=24))
a.commit()
assert grep('%s ip link show' % self.ssh,
pattern=ifname)
assert grep('%s ip addr show dev %s' % (self.ssh, ifname),
pattern=ifaddr)
def test_basic_route(self):
ifaddr = self.ifaddr()
router = self.ifaddr()
ifname = self.ifname()
i = (self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy', state='up'))
i.commit()
a = (self
.ndb
.addresses
.create(index=i['index'],
address=ifaddr,
prefixlen=24))
a.commit()
r = (self
.ndb
.routes
.create(dst_len=24,
dst=str(self.ipnets[1].network),
gateway=router))
r.commit()
assert grep('%s ip link show' % self.ssh,
pattern=ifname)
assert grep('%s ip addr show dev %s' % (self.ssh, ifname),
pattern=ifaddr)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (str(self.ipnets[1]), ifname))
class TestRollback(TestBase):
def setup(self):
require_user('root')
self.ipnets = [allocate_network() for _ in range(5)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
self.interfaces = []
def test_simple_deps(self):
# register NDB handler to wait for the interface
self.if_simple = uifname()
ifaddr = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
#
# simple dummy interface with one address and
# one dependent route
#
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_simple, kind='dummy')
.set('state', 'up')
.commit()['index']))
(self
.ndb
.addresses
.create(address=ifaddr,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
iface = self.ndb.interfaces[self.if_simple]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert not grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh, pattern=self.if_simple)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
def test_bridge_deps(self):
self.if_br0 = uifname()
self.if_br0p0 = uifname()
self.if_br0p1 = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0,
kind='bridge',
state='up')
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0p0,
kind='dummy',
state='up',
master=self.ndb.interfaces[self.if_br0]['index'])
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=self.if_br0p1,
kind='dummy',
state='up',
master=self.ndb.interfaces[self.if_br0]['index'])
.commit()['index']))
(self
.ndb
.interfaces[self.if_br0]
.ipaddr
.create(address=ifaddr1, prefixlen=24)
.commit())
(self
.ndb
.interfaces[self.if_br0]
.ipaddr
.create(address=ifaddr2, prefixlen=24)
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
master = self.ndb.interfaces[self.if_br0]['index']
self.ndb.interfaces.wait(ifname=self.if_br0p0, master=master)
self.ndb.interfaces.wait(ifname=self.if_br0p1, master=master)
self.ndb.addresses.wait(address=ifaddr1)
self.ndb.addresses.wait(address=ifaddr2)
self.ndb.routes.wait(dst=dst, gateway=router)
iface = self.ndb.interfaces[self.if_br0]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert not grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p0)
assert grep('%s ip link show' % self.ssh, pattern=self.if_br0p1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=self.if_br0)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
def test_vlan_deps(self):
if_host = uifname()
if_vlan = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
router = self.ifaddr()
dst = str(self.ipnets[1].network)
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=if_host,
kind='dummy',
state='up')
.commit()['index']))
(self
.interfaces
.append(self
.ndb
.interfaces
.create(ifname=if_vlan,
kind='vlan',
link=self.interfaces[-1],
state='up',
vlan_id=1001)
.commit()['index']))
(self
.ndb
.addresses
.create(address=ifaddr1,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.addresses
.create(address=ifaddr2,
prefixlen=24,
index=self.interfaces[-1])
.commit())
(self
.ndb
.routes
.create(dst=dst, dst_len=24, gateway=router)
.commit())
iface = self.ndb.interfaces[if_host]
# check everything is in place
assert grep('%s ip link show' % self.ssh, pattern=if_host)
assert grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert grep('%s cat /proc/net/vlan/config' % self.ssh, pattern=if_vlan)
# remove the interface
iface.remove()
iface.commit()
# check there is no interface, no route
assert not grep('%s ip link show' % self.ssh, pattern=if_host)
assert not grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert not grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert not grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert not grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert not grep('%s cat /proc/net/vlan/config' % self.ssh,
pattern=if_vlan)
# revert the changes using the implicit last_save
iface.rollback()
assert grep('%s ip link show' % self.ssh, pattern=if_host)
assert grep('%s ip link show' % self.ssh, pattern=if_vlan)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr1)
assert grep('%s ip addr show' % self.ssh, pattern=ifaddr2)
assert grep('%s ip route show' % self.ssh, pattern=if_vlan)
assert grep('%s ip route show' % self.ssh,
pattern='%s.*%s' % (dst, router))
assert grep('%s cat /proc/net/vlan/config' % self.ssh, pattern=if_vlan)
class TestNetNS(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
def setup(self):
require_user('root')
self.netns = str(uuid.uuid4())
self.ipnets = [allocate_network() for _ in range(3)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.sources = [{'target': 'localhost'},
{'netns': self.netns}]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
sources=self.sources,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
def ifaddr(self, r=0):
return str(self.ipranges[r].pop())
def teardown(self):
for net in self.ipnets:
free_network(net)
self.ndb.close()
netns.remove(self.netns)
def test_basic(self):
ifname = uifname()
ifaddr1 = self.ifaddr()
ifaddr2 = self.ifaddr()
ifaddr3 = self.ifaddr()
(self
.ndb
.interfaces
.create(target=self.netns, ifname=ifname, kind='dummy')
.ipaddr
.create(address=ifaddr1, prefixlen=24)
.create(address=ifaddr2, prefixlen=24)
.create(address=ifaddr3, prefixlen=24)
.commit())
with NDB(sources=[{'target': 'localhost',
'netns': self.netns,
'kind': 'netns'}]) as ndb:
if_idx = ndb.interfaces[ifname]['index']
addr1_idx = ndb.addresses['%s/24' % ifaddr1]['index']
addr2_idx = ndb.addresses['%s/24' % ifaddr2]['index']
addr3_idx = ndb.addresses['%s/24' % ifaddr3]['index']
assert if_idx == addr1_idx == addr2_idx == addr3_idx
def _assert_test_view(self, ifname, ifaddr):
with NDB(sources=[{'target': 'localhost',
'netns': self.netns,
'kind': 'netns'}]) as ndb:
(if_idx,
if_state,
if_addr,
if_flags) = ndb.interfaces[ifname].fields('index',
'state',
'address',
'flags')
addr_idx = ndb.addresses['%s/24' % ifaddr]['index']
assert if_idx == addr_idx
assert if_state == 'up'
assert if_flags & 1
assert if_addr == '00:11:22:33:44:55'
def test_view_constraints_pipeline(self):
ifname = uifname()
ifaddr = self.ifaddr()
(self
.ndb
.interfaces
.constraint('target', self.netns)
.create(ifname=ifname, kind='dummy')
.set('address', '00:11:22:33:44:55')
.set('state', 'up')
.ipaddr
.create(address=ifaddr, prefixlen=24)
.commit())
self._assert_test_view(ifname, ifaddr)
def test_view_constraints_cmanager(self):
ifname = uifname()
ifaddr = self.ifaddr()
with self.ndb.interfaces as view:
view.constraints['target'] = self.netns
with view.create(ifname=ifname, kind='dummy') as interface:
interface['address'] = '00:11:22:33:44:55'
interface['state'] = 'up'
with interface.ipaddr as aview:
with aview.create(address=ifaddr, prefixlen=24):
pass
self._assert_test_view(ifname, ifaddr)
def test_move(self):
ifname = uifname()
ifaddr = self.ifaddr()
# create the interfaces
(self
.ndb
.interfaces
.create(ifname=ifname, kind='dummy')
.commit())
# move it to a netns
(self
.ndb
.interfaces[ifname]
.set('net_ns_fd', self.netns)
.commit())
# setup the interface only when it is moved
(self
.ndb
.interfaces
.wait(target=self.netns, ifname=ifname)
.set('state', 'up')
.set('address', '00:11:22:33:44:55')
.ipaddr
.create(address=ifaddr, prefixlen=24)
.commit())
self._assert_test_view(ifname, ifaddr)
class TestBase(object):
db_provider = 'sqlite3'
db_spec = ':memory:'
nl_class = IPRoute
nl_kwarg = {}
ssh = ''
ipnets = []
ipranges = []
def create_interfaces(self):
# dummy interface
if_dummy = uifname()
if_vlan_stag = uifname()
if_vlan_ctag = uifname()
if_bridge = uifname()
if_port = uifname()
if_addr1 = self.ifaddr()
if_addr2 = self.ifaddr()
ret = []
ret.append(self
.ndb
.interfaces
.create(ifname=if_dummy, kind='dummy')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_vlan_stag,
link=self.ndb.interfaces[if_dummy]['index'],
vlan_id=101,
vlan_protocol=0x88a8,
kind='vlan')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_vlan_ctag,
link=self.ndb.interfaces[if_vlan_stag]['index'],
vlan_id=1001,
vlan_protocol=0x8100,
kind='vlan')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_bridge, kind='bridge')
.commit()['index'])
ret.append(self
.ndb
.interfaces
.create(ifname=if_port,
master=self.ndb.interfaces[if_bridge]['index'],
kind='dummy')
.commit()['index'])
(self
.ndb
.interfaces[if_bridge]
.ipaddr
.create(address=if_addr1, prefixlen=24)
.commit())
(self
.ndb
.interfaces[if_bridge]
.ipaddr
.create(address=if_addr2, prefixlen=24)
.commit())
self.if_bridge = if_bridge
return ret
def ifaddr(self, r=0):
return str(self.ipranges[r].pop())
def setup(self):
require_user('root')
self.if_simple = None
self.ipnets = [allocate_network() for _ in range(5)]
self.ipranges = [[str(x) for x in net] for net in self.ipnets]
self.ndb = NDB(db_provider=self.db_provider,
db_spec=self.db_spec,
debug=True)
self.ndb.log('../ndb-%s-%s.log' % (os.getpid(), id(self.ndb)))
self.interfaces = self.create_interfaces()
def teardown(self):
with self.nl_class(**self.nl_kwarg) as ipr:
for link in reversed(self.interfaces):
ipr.link('del', index=link)
self.ndb.close()
for net in self.ipnets:
free_network(net)
def fetch(self, request, values=[]):
return (self
.ndb
.schema
.fetch(request, values))
