def setUp(self):
self.c = Command()
self.out = StringIO()
self.c.stdout = self.c.stderr = self.out
self.test_dir = '%s/test_folder' % os.path.dirname(os.path.realpath(__file__))
os.mkdir(self.test_dir)
os.chdir(self.test_dir)
class LoadCountyLimitsTestCase(TestCase):
def prepare_sample_data(self, filename):
with open(filename, 'w') as data:
data.write("Header to be skipped\n")
data.write('DC,01,001,01001,DC County 1,417000,271050,417000\n')
data.write('DC,01,002,01002,DC County 2,417000,271050,417000\n')
data.close()
def setUp(self):
self.c = Command()
self.out = StringIO()
self.c.stdout = self.c.stderr = self.out
self.test_dir = '%s/test_folder' % os.path.dirname(
os.path.realpath(__file__))
os.mkdir(self.test_dir)
os.chdir(self.test_dir)
def tearDown(self):
shutil.rmtree(self.test_dir)
def test_handle__no_confirm(self):
""" .. check that nothing happens when confirm is not y|Y."""
self.c.handle(stdout=self.out)
self.assertNotIn(self.out.getvalue(), 'Successfully loaded data from')
def test_handle__no_arguments(self):
""" .. check that CommandError is raised."""
self.assertRaises(CommandError,
self.c.handle,
confirmed='y',
stdout=self.out)
def test_handle__bad_file(self):
""" .. check that CommandError is raised when path to file is wrong."""
self.assertRaises(CommandError,
self.c.handle,
'inexistent.file.csv',
confirmed='Y',
stdout=self.out)
def test_handle__success(self):
""" .. check that all countylimits are loaded."""
fname = 'county-limits.csv'
self.prepare_sample_data(fname)
self.c.handle('%s/%s' % (self.test_dir, fname), confirmed='Y')
self.assertIn(
'Successfully loaded data from %s/%s' % (self.test_dir, fname),
self.out.getvalue())
county_limits = CountyLimit.objects.all()
self.assertEqual(len(county_limits), 2)
class LoadCountyLimitsTestCase(TestCase):
def prepare_sample_data(self, filename):
with open(filename, 'w') as data:
data.write("Header to be skipped\n")
data.write('DC,01,001,01001,DC County 1,417000,271050,417000\n')
data.write('DC,01,002,01002,DC County 2,417000,271050,417000\n')
data.close()
def setUp(self):
self.c = Command()
self.out = StringIO()
self.c.stdout = self.c.stderr = self.out
self.test_dir = '%s/test_folder' % os.path.dirname(os.path.realpath(__file__))
os.mkdir(self.test_dir)
os.chdir(self.test_dir)
def tearDown(self):
shutil.rmtree(self.test_dir)
def test_handle__no_confirm(self):
""" .. check that nothing happens when confirm is not y|Y."""
self.c.handle(stdout=self.out)
self.assertNotIn(self.out.getvalue(), 'Successfully loaded data from')
def test_handle__no_arguments(self):
""" .. check that CommandError is raised."""
self.assertRaises(CommandError, self.c.handle, confirmed='y', stdout=self.out)
def test_handle__bad_file(self):
""" .. check that CommandError is raised when path to file is wrong."""
self.assertRaises(CommandError, self.c.handle, 'inexistent.file.csv', confirmed='Y', stdout=self.out)
def test_handle__success(self):
""" .. check that all countylimits are loaded."""
fname = 'county-limits.csv'
self.prepare_sample_data(fname)
self.c.handle('%s/%s' % (self.test_dir, fname), confirmed='Y')
self.assertIn('Successfully loaded data from %s/%s' % (self.test_dir, fname), self.out.getvalue())
county_limits = CountyLimit.objects.all()
self.assertEqual(len(county_limits), 2)
