def setUp(self):
'''Creates with in-memory DB'''
self.conf = {}
self.conf['dbUrl'] = 'sqlite:///'
self.conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.conf)
self.session = self.ztpServer.dao.Session()
def setUp(self):
'''Creates with in-memory DB'''
self.conf = {}
self.conf['dbUrl'] = 'sqlite:///'
self.conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.conf)
self.session = self.ztpServer.dao.Session()
def setupZTP(self):
'''
Setup Zero Touch Provisioning
Generate DHCP config file
Install and restart DHCP server with new dhcp configuration
'''
ztpServer = ZtpServer()
ztpServer.createPodSpecificDhcpConfFile('anotherPod')
if jnpr.openclos.util.isPlatformUbuntu():
os.system('sudo apt-get -y install isc-dhcp-server')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/init.d/isc-dhcp-server restart")
elif jnpr.openclos.util.isPlatformCentos():
os.system('yum -y install dhcp')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/rc.d/init.d/dhcpd restart")
def setupZTP(self):
'''
Setup Zero Touch Provisioning
Generate DHCP config file
Install and restart DHCP server with new dhcp configuration
'''
ztpServer = ZtpServer()
ztpServer.createPodSpecificDhcpConfFile('anotherPod')
if jnpr.openclos.util.isPlatformUbuntu():
os.system('sudo apt-get -y install isc-dhcp-server')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/init.d/isc-dhcp-server restart")
elif jnpr.openclos.util.isPlatformCentos():
os.system('yum -y install dhcp')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/rc.d/init.d/dhcpd restart")
def setupZTP(self):
'''
Setup Zero Touch Provisioning
Generate DHCP config file
Install and restart DHCP server with new dhcp configuration
'''
ztpServer = ZtpServer()
with ztpServer._dao.getReadSession() as session:
ztpServer.createPodSpecificDhcpConfFile(session, self.pod.id)
print generatedDhcpConf
if jnpr.openclos.util.isPlatformUbuntu():
os.system('sudo apt-get -y install isc-dhcp-server')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system('sudo /usr/sbin/service isc-dhcp-server restart')
elif jnpr.openclos.util.isPlatformCentos():
os.system('yum -y install dhcp')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/rc.d/init.d/dhcpd restart")
def setupZTP(self):
'''
Setup Zero Touch Provisioning
Generate DHCP config file
Install and restart DHCP server with new dhcp configuration
'''
ztpServer = ZtpServer()
with ztpServer._dao.getReadSession() as session:
ztpServer.createPodSpecificDhcpConfFile(session, self.pod.id)
print generatedDhcpConf
if jnpr.openclos.util.isPlatformUbuntu():
os.system('sudo apt-get -y install isc-dhcp-server')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system('sudo /usr/sbin/service isc-dhcp-server restart')
elif jnpr.openclos.util.isPlatformCentos():
os.system('yum -y install dhcp')
os.system('sudo cp ' + generatedDhcpConf + ' ' + installedDhcpConf)
os.system("/etc/rc.d/init.d/dhcpd restart")
def setUp(self):
self.__conf = {}
self.__conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.__conf, daoClass = InMemoryDao)
self._dao = InMemoryDao.getInstance()
class TestZtp(unittest.TestCase):
def setUp(self):
self.__conf = {}
self.__conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.__conf, daoClass = InMemoryDao)
self._dao = InMemoryDao.getInstance()
def tearDown(self):
InMemoryDao._destroy()
def testGenerateDhcpConfWithNoPodDevice(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
with self._dao.getReadWriteSession() as session:
dhcpConf = self.ztpServer.generateSingleDhcpConf(session)
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(1, dhcpConf.count('host-name')) # 1 global + 0 device
def testGenerateSingleDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
with self._dao.getReadWriteSession() as session:
createDevice(session, 'dev1')
createDevice(session, 'dev2')
dhcpConf = self.ztpServer.generateSingleDhcpConf(session)
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(3, dhcpConf.count('host-name')) # 1 global + 2 device
def testGeneratePodSpecificDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(session, pod.id)
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertEquals(4, dhcpConf.count('host-name')) # 1 global + 3 device
def testGeneratePodSpecificDhcpConfWithSerial(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev2.macAddress = None
dev2.serialNumber = 'VB1234567890'
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dev3.serialNumber = 'VB1234567891'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(session, pod.id)
print dhcpConf
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertTrue('VB1234567890' in dhcpConf)
self.assertTrue('VB1234567891' not in dhcpConf)
self.assertEquals(5, dhcpConf.count('host-name')) # 1 global class + 1 subnet + 2 device mac + 1 device serial
def testGeneratePodSpecificDhcpConfFor2StageZtp(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
flexmock(util, isZtpStaged = True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
pod.leafSettings.append(LeafSetting('ex4300-24p', pod.id))
dev1 = createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dev4 = createPodDevice(session, 'dev4', pod)
dev4.role = 'leaf'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(session, pod.id)
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertEquals(3, dhcpConf.count('host-name')) # 1 global + 2 spine device
self.assertEquals(1, dhcpConf.count('pool'))
self.assertEquals(2, dhcpConf.count('class '))
self.assertEquals(4, dhcpConf.count('vendor-class-identifier'))
def testPopulateDhcpGlobalSettings(self):
from jnpr.openclos.l3Clos import util
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25', 'dhcpOptionRoute': '10.20.30.254', 'dhcpOptionRangeStart': '10.20.30.15','dhcpOptionRangeEnd': '10.20.30.20'}}
flexmock(util, loadClosDefinition = globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.0', globalSetting['network'])
self.assertEquals('255.255.255.128', globalSetting['netmask'])
self.assertEquals('10.20.30.254', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.15', globalSetting['rangeStart'])
self.assertEquals('10.20.30.20', globalSetting['rangeEnd'])
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25'}}
flexmock(util, loadClosDefinition = globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.1', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.2', globalSetting['rangeStart'])
self.assertEquals('10.20.30.126', globalSetting['rangeEnd'])
def setUp(self):
self.__conf = {}
self.__conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.__conf, daoClass=InMemoryDao)
self._dao = InMemoryDao.getInstance()
class TestZtp(unittest.TestCase):
def setUp(self):
self.__conf = {}
self.__conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.__conf, daoClass=InMemoryDao)
self._dao = InMemoryDao.getInstance()
def tearDown(self):
InMemoryDao._destroy()
def testGenerateDhcpConfWithNoPodDevice(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
with self._dao.getReadWriteSession() as session:
dhcpConf = self.ztpServer.generateSingleDhcpConf(session)
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(1,
dhcpConf.count('host-name')) # 1 global + 0 device
def testGenerateSingleDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
with self._dao.getReadWriteSession() as session:
createDevice(session, 'dev1')
createDevice(session, 'dev2')
dhcpConf = self.ztpServer.generateSingleDhcpConf(session)
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(3,
dhcpConf.count('host-name')) # 1 global + 2 device
def testGeneratePodSpecificDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(
session, pod.id)
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertEquals(4,
dhcpConf.count('host-name')) # 1 global + 3 device
def testGeneratePodSpecificDhcpConfWithSerial(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev2.macAddress = None
dev2.serialNumber = 'VB1234567890'
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dev3.serialNumber = 'VB1234567891'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(
session, pod.id)
print dhcpConf
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertTrue('VB1234567890' in dhcpConf)
self.assertTrue('VB1234567891' not in dhcpConf)
self.assertEquals(
5, dhcpConf.count('host-name')
) # 1 global class + 1 subnet + 2 device mac + 1 device serial
def testGeneratePodSpecificDhcpConfFor2StageZtp(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
flexmock(util, isZtpStaged=True)
with self._dao.getReadWriteSession() as session:
pod = createPod('pod1', session)
pod.spineJunosImage = 'testSpineImage'
pod.leafSettings.append(LeafSetting('ex4300-24p', pod.id))
dev1 = createPodDevice(session, 'dev1', pod)
dev2 = createPodDevice(session, 'dev2', pod)
dev3 = createPodDevice(session, 'dev3', pod)
dev3.role = 'leaf'
dev4 = createPodDevice(session, 'dev4', pod)
dev4.role = 'leaf'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf(
session, pod.id)
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertFalse('None' in dhcpConf)
self.assertEquals(
3, dhcpConf.count('host-name')) # 1 global + 2 spine device
self.assertEquals(1, dhcpConf.count('pool'))
self.assertEquals(2, dhcpConf.count('class '))
self.assertEquals(4, dhcpConf.count('vendor-class-identifier'))
def testPopulateDhcpGlobalSettings(self):
from jnpr.openclos.l3Clos import util
globalZtpConf = {
'ztp': {
'dhcpSubnet': '10.20.30.0/25',
'dhcpOptionRoute': '10.20.30.254',
'dhcpOptionRangeStart': '10.20.30.15',
'dhcpOptionRangeEnd': '10.20.30.20'
}
}
flexmock(util, loadClosDefinition=globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.0', globalSetting['network'])
self.assertEquals('255.255.255.128', globalSetting['netmask'])
self.assertEquals('10.20.30.254', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.15', globalSetting['rangeStart'])
self.assertEquals('10.20.30.20', globalSetting['rangeEnd'])
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25'}}
flexmock(util, loadClosDefinition=globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.1', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.2', globalSetting['rangeStart'])
self.assertEquals('10.20.30.126', globalSetting['rangeEnd'])
class TestZtp(unittest.TestCase):
def setUp(self):
'''Creates with in-memory DB'''
self.conf = {}
self.conf['dbUrl'] = 'sqlite:///'
self.conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.conf)
self.session = self.ztpServer.dao.Session()
def tearDown(self):
pass
def testGenerateDhcpConfWithNoPodDevice(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
dhcpConf = self.ztpServer.generateSingleDhcpConf()
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(1, dhcpConf.count('host-name')) # 1 global + 0 device
def testGenerateSingleDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
createDevice(self.session, 'dev1')
createDevice(self.session, 'dev2')
dhcpConf = self.ztpServer.generateSingleDhcpConf()
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(3, dhcpConf.count('host-name')) # 1 global + 2 device
def testGeneratePodSpecificDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu = True)
pod = createPod('pod1', self.session)
pod.spineJunosImage = 'testSpineImage'
pod.leafJunosImage = 'testLeafImage'
createPodDevice(self.session, 'dev1', pod)
dev2 = createPodDevice(self.session, 'dev2', pod)
dev3 = createPodDevice(self.session, 'dev3', pod)
dev3.role = 'leaf'
dev4 = createPodDevice(self.session, 'dev4', pod)
dev4.role = 'unknown'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf('pod1')
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertEquals(1, dhcpConf.count('testLeafImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(5, dhcpConf.count('host-name')) # 1 global + 4 device
def testPopulateDhcpGlobalSettings(self):
from jnpr.openclos.l3Clos import util
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25', 'dhcpOptionRoute': '10.20.30.254', 'dhcpOptionRangeStart': '10.20.30.15','dhcpOptionRangeEnd': '10.20.30.20'}}
flexmock(util, loadClosDefinition = globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.0', globalSetting['network'])
self.assertEquals('255.255.255.128', globalSetting['netmask'])
self.assertEquals('10.20.30.254', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.15', globalSetting['rangeStart'])
self.assertEquals('10.20.30.20', globalSetting['rangeEnd'])
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25'}}
flexmock(util, loadClosDefinition = globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.1', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.2', globalSetting['rangeStart'])
self.assertEquals('10.20.30.126', globalSetting['rangeEnd'])
class TestZtp(unittest.TestCase):
def setUp(self):
'''Creates with in-memory DB'''
self.conf = {}
self.conf['dbUrl'] = 'sqlite:///'
self.conf['httpServer'] = {'ipAddr': '127.0.0.1'}
self.ztpServer = ZtpServer(self.conf)
self.session = self.ztpServer.dao.Session()
def tearDown(self):
pass
def testGenerateDhcpConfWithNoPodDevice(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
dhcpConf = self.ztpServer.generateSingleDhcpConf()
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(1,
dhcpConf.count('host-name')) # 1 global + 0 device
def testGenerateSingleDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
createDevice(self.session, 'dev1')
createDevice(self.session, 'dev2')
dhcpConf = self.ztpServer.generateSingleDhcpConf()
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(3,
dhcpConf.count('host-name')) # 1 global + 2 device
def testGeneratePodSpecificDhcpConf(self):
from jnpr.openclos.l3Clos import util
flexmock(util, isPlatformUbuntu=True)
pod = createPod('pod1', self.session)
pod.spineJunosImage = 'testSpineImage'
pod.leafJunosImage = 'testLeafImage'
createPodDevice(self.session, 'dev1', pod)
dev2 = createPodDevice(self.session, 'dev2', pod)
dev3 = createPodDevice(self.session, 'dev3', pod)
dev3.role = 'leaf'
dev4 = createPodDevice(self.session, 'dev4', pod)
dev4.role = 'unknown'
dhcpConf = self.ztpServer.generatePodSpecificDhcpConf('pod1')
self.assertEquals(2, dhcpConf.count('testSpineImage'))
self.assertEquals(1, dhcpConf.count('testLeafImage'))
self.assertFalse('{{' in dhcpConf)
self.assertFalse('}}' in dhcpConf)
self.assertEquals(5,
dhcpConf.count('host-name')) # 1 global + 4 device
def testPopulateDhcpGlobalSettings(self):
from jnpr.openclos.l3Clos import util
globalZtpConf = {
'ztp': {
'dhcpSubnet': '10.20.30.0/25',
'dhcpOptionRoute': '10.20.30.254',
'dhcpOptionRangeStart': '10.20.30.15',
'dhcpOptionRangeEnd': '10.20.30.20'
}
}
flexmock(util, loadClosDefinition=globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.0', globalSetting['network'])
self.assertEquals('255.255.255.128', globalSetting['netmask'])
self.assertEquals('10.20.30.254', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.15', globalSetting['rangeStart'])
self.assertEquals('10.20.30.20', globalSetting['rangeEnd'])
globalZtpConf = {'ztp': {'dhcpSubnet': '10.20.30.0/25'}}
flexmock(util, loadClosDefinition=globalZtpConf)
globalSetting = self.ztpServer.populateDhcpGlobalSettings()
self.assertEquals('10.20.30.1', globalSetting['defaultRoute'])
self.assertEquals('10.20.30.2', globalSetting['rangeStart'])
self.assertEquals('10.20.30.126', globalSetting['rangeEnd'])