Add test for checking invalid file format

openwisp · Nov 20, 2019 · 2323465 · 2323465
1 parent 0c4baf3
commit 2323465
Showing 1 changed file with 17 additions and 10 deletions.
diff --git a/django_freeradius/tests/base/test_utils.py b/django_freeradius/tests/base/test_utils.py
@@ -7,17 +7,24 @@
 class BaseTestUtils(object):
     def test_find_available_username(self):
         User = get_user_model()
-        User.objects.create(username='rohith', password='password')
-        self.assertEqual(find_available_username('rohith', []), 'rohith1')
-        User.objects.create(username='rohith1', password='password')
-        self.assertEqual(find_available_username('rohith', []), 'rohith2')
+        User.objects.create(username="rohith", password="password")
+        self.assertEqual(find_available_username("rohith", []), "rohith1")
+        User.objects.create(username="rohith1", password="password")
+        self.assertEqual(find_available_username("rohith", []), "rohith2")
 
     def test_validate_csvfile(self):
-        invalid_csv_path = self._get_path('static/test_batch_invalid.csv')
-        improper_csv_path = self._get_path('static/test_batch_improper.csv')
+        invalid_csv_path = self._get_path("static/test_batch_invalid.csv")
+        improper_csv_path = self._get_path("static/test_batch_improper.csv")
+        invalid_format_path = self._get_path("static/test_batch_invalid_format.pdf")
         with self.assertRaises(ValidationError) as error:
-            validate_csvfile(open(invalid_csv_path, 'rt'))
-        self.assertTrue('Enter a valid email address' in error.exception.message)
+            validate_csvfile(open(invalid_format_path, "rb"))
+        self.assertTrue(
+            "Incorrect file format has been uploaded. Please Try Again!"
+            in error.exception.message
+        )
         with self.assertRaises(ValidationError) as error:
-            validate_csvfile(open(improper_csv_path, 'rt'))
-        self.assertTrue('Improper CSV format' in error.exception.message)
+            validate_csvfile(open(invalid_csv_path, "rt"))
+        self.assertTrue("Enter a valid email address" in error.exception.message)
+        with self.assertRaises(ValidationError) as error:
+            validate_csvfile(open(improper_csv_path, "rt"))
+        self.assertTrue("Improper CSV format" in error.exception.message)