class ExternalTftpTest(unittest.TestCase):
"""Tests the ExternalTftp class.
..note:
* For testing purposes the 'external' server points to an internally
hosted one, but it allows us to make sure the actual TFTP protocol is
working.
"""
def setUp(self):
"""Create an ExternalTftp object to test with."""
self.itftp = InternalTftp(ip_address='127.0.0.250')
self.etftp = ExternalTftp(
self.itftp.get_address(relative_host=_get_relative_host()),
self.itftp.port)
def test_put_and_get(self):
"""Test the put_file(src, dest) function. Test the get_file(src,dest)
function by movign local files around using the TFT Protocol.
"""
# Create file
filename = random_file(1024)
contents = open(filename).read()
# Upload and remove original file
basename = os.path.basename(filename)
self.etftp.put_file(src=filename, dest=basename)
os.remove(filename)
self.assertFalse(os.path.exists(filename))
# Download
self.etftp.get_file(src=basename, dest=filename)
# Verify match
self.assertEqual(open(filename).read(), contents)
os.remove(filename)
class ExternalTftpTest(unittest.TestCase):
"""Tests the ExternalTftp class.
..note:
* For testing purposes the 'external' server points to an internally
hosted one, but it allows us to make sure the actual TFTP protocol is
working.
"""
def setUp(self):
"""Create an ExternalTftp object to test with."""
self.itftp = InternalTftp(ip_address='127.0.0.250')
self.etftp = ExternalTftp(
self.itftp.get_address(relative_host=_get_relative_host()),
self.itftp.port)
def test_put_and_get(self):
"""Test the put_file(src, dest) function. Test the get_file(src,dest)
function by movign local files around using the TFT Protocol.
"""
# Create file
filename = random_file(1024)
contents = open(filename).read()
# Upload and remove original file
basename = os.path.basename(filename)
self.etftp.put_file(src=filename, dest=basename)
os.remove(filename)
self.assertFalse(os.path.exists(filename))
# Download
self.etftp.get_file(src=basename, dest=filename)
# Verify match
self.assertEqual(open(filename).read(), contents)
os.remove(filename)
class InternalTftpTest(unittest.TestCase):
""" Tests the functions of the InternalTftp class."""
def setUp(self):
"""Create local Internal TFTP objects to test with."""
self.tftp1 = InternalTftp()
self.tftp2 = InternalTftp(ip_address='127.0.0.254')
def test_put_and_get(self):
""" Test file transfers on an internal host """
# Create file
filename = random_file(1024)
contents = open(filename).read()
# Upload and remove
basename = os.path.basename(filename)
self.tftp1.put_file(filename, basename)
os.remove(filename)
self.assertFalse(os.path.exists(filename))
# Download
self.tftp1.get_file(basename, filename)
# Verify match
self.assertEqual(open(filename).read(), contents)
os.remove(filename)
def test_get_address_with_relhost(self):
"""Tests the get_address(relative_host) function with a relative_host
specified.
"""
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
# RFC863 defines port 9 as the UDP discard port, so we use it to
# find out our local ip to pass as a relative_host
sock.connect((socket.gethostname(), 9))
relative_host = sock.getsockname()[0]
except socket.error:
raise
self.assertEqual(self.tftp2.ip_address,
self.tftp2.get_address(relative_host=relative_host))
sock.close()
class InternalTftpTest(unittest.TestCase):
""" Tests the functions of the InternalTftp class."""
def setUp(self):
"""Create local Internal TFTP objects to test with."""
self.tftp1 = InternalTftp()
self.tftp2 = InternalTftp(ip_address='127.0.0.254')
def test_put_and_get(self):
""" Test file transfers on an internal host """
# Create file
filename = random_file(1024)
contents = open(filename).read()
# Upload and remove
basename = os.path.basename(filename)
self.tftp1.put_file(filename, basename)
os.remove(filename)
self.assertFalse(os.path.exists(filename))
# Download
self.tftp1.get_file(basename, filename)
# Verify match
self.assertEqual(open(filename).read(), contents)
os.remove(filename)
def test_get_address_with_relhost(self):
"""Tests the get_address(relative_host) function with a relative_host
specified.
"""
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
# RFC863 defines port 9 as the UDP discard port, so we use it to
# find out our local ip to pass as a relative_host
sock.connect((socket.gethostname(), 9))
relative_host = sock.getsockname()[0]
except socket.error:
raise
self.assertEqual(self.tftp2.ip_address,
self.tftp2.get_address(relative_host=relative_host))
sock.close()
