Merge pull request #6 from CIAT-DAPA/develop

GAP-31 new tests have been added for models

Author: victor-993

src/tests/test_accession.py

@@ -44,6 +44,61 @@ def test_create_accession(self):
         self.assertEqual(accession.latitude, 40.7128)
         self.assertEqual(accession.longitude, -74.0060)
 
+    
+    def test_get_accession_by_crop(self):
+        self.accession.save()
+        accessions = Accession.objects(crop=self.crop)
+        self.assertGreaterEqual(len(accessions), 1)
+        accession = accessions.first()
+        self.assertEqual(self.crop.id, accession.crop.id)
+
+    def test_get_accession_by_group(self):
+        self.accession.save()
+        accessions =Accession.objects(landrace_group=self.group)
+        self.assertGreaterEqual(len(accessions), 1)
+        accession = accessions.first()
+        self.assertEqual(self.group.id, accession.landrace_group.id)
+    
+    def test_unique_ext_id(self):
+        self.accession.save()
+        accession = Accession(
+            species_name='Test Species',
+            crop=self.crop,
+            landrace_group=self.group,
+            institution_name='ICARDA',
+            source_database='GENESYS',
+            latitude=91, # Latitud inválida
+            longitude=-181, # Longitud inválida
+            accession_id='12345',
+            ext_id='123'
+        )
+        with self.assertRaises(Exception) as context:
+            accession.save()
+        print(str(context.exception))
+        self.assertTrue('Tried to save duplicate unique keys (E11000 Duplicate Key Error)' in str(context.exception))
+
+    def test_get_accession_by_invalid_id(self):
+        # Busca un country por un ID que no existe
+        found_country = Accession.objects(crop='609e996db1751152b09f5555').first()
+        self.assertIsNone(found_country)
+
+    def test_update_accession(self):
+        self.accession.save()
+
+        self.accession.species_name = 'New Species Name'
+        self.accession.save()
+
+        accession = Accession.objects(id=self.accession.id).first()
+        self.assertEqual(accession.species_name, 'New Species Name')
+    
+    def test_delete_accession(self):
+        self.accession.save()
+
+        self.accession.delete()
+
+        accession = Accession.objects(id=self.accession.id).first()
+        self.assertIsNone(accession)
+
     def tearDown(self):
         self.crop.delete()
         self.group.delete()

src/tests/test_country.py

@@ -30,6 +30,58 @@ def test_create_country(self):
         self.assertEqual(country.iso_2, 'US')
         self.assertEqual(country.name, 'United States of America')
 
+    def test_get_country_by_invalid_id(self):
+        # Busca un country por un ID que no existe
+        found_country = Country.objects(id='609e996db1751152b09f5555').first()
+        self.assertIsNone(found_country)
+
+    def test_get_country_by_iso_2(self):
+        self.country.save()
+
+        # Busca el country por su ISO 2
+        found_country = Country.objects(iso_2=self.country.iso_2).first()
+        self.assertIsNotNone(found_country)
+
+        # Verifica que los datos del country encontrado sean los mismos que los creados previamente
+        self.assertEqual(found_country.iso_2, 'US')
+        self.assertEqual(found_country.name, 'United States of America')
+
+    def test_get_country_by_invalid_iso_2(self):
+        # Busca un country por un ISO 2 que no existe
+        found_country = Country.objects(iso_2='XX').first()
+        self.assertIsNone(found_country)
+    
+    def test_update_country(self):
+        # Actualiza los datos de un país existente
+
+        self.country.save()
+
+        self.country.name = 'USA'
+        self.country.save()
+
+        updated_country = Country.objects(id=self.country.id).first()
+        self.assertEqual(updated_country.name, 'USA')
+
+    def test_unique_iso_2(self):
+        # Verifica que no se puedan crear dos países con el mismo ISO 2
+        self.country.save()
+        duplicate_country = Country(
+            iso_2='US',
+            name='USA',
+        )
+
+        with self.assertRaises(Exception) as context:       
+            duplicate_country.save()
+
+        self.assertTrue('Tried to save duplicate unique keys (E11000 Duplicate Key Error)' in str(context.exception))
+
+    def test_delete_country(self):
+        self.country.save()
+
+        self.country.delete()
+
+        country = Country.objects(id=self.country.id).first()
+        self.assertIsNone(country)  
 
     def tearDown(self):
         Country.objects.delete()

src/tests/test_crop.py

@@ -35,6 +35,48 @@ def test_create_crop(self):
         self.assertEqual(crop.base_name, 'zea_mays')
         self.assertEqual(crop.app_name, 'Corn')
 
+    def test_create_duplicate_crop(self):
+        self.crop.save()
+        crop2 = Crop(
+            ext_id='1234',
+            name='wheat',
+            base_name='triticum_aestivum',
+            app_name='Wheat'
+        )
+
+        with self.assertRaises(Exception) as context:
+            crop2.save()
+        
+        print(str(context.exception))
+        self.assertTrue('Tried to save duplicate unique keys (E11000 Duplicate Key Error)' in str(context.exception))
+
+    def test_get_crop_by_invalid_id(self):
+        # Busca un country por un ID que no existe
+        found_country = Crop.objects(id='609e996db1751152b09f5555').first()
+        self.assertIsNone(found_country)
+    
+    def test_update_crop(self):
+        self.crop.save()
+
+        self.crop.name = 'maize'
+        self.crop.base_name = 'zea_mays_subsp_mays'
+        self.crop.app_name = 'Maize'
+        self.crop.save()
+
+        crop = Crop.objects(id=self.crop.id).first()
+        self.assertEqual(crop.ext_id, '1234')
+        self.assertEqual(crop.name, 'maize')
+        self.assertEqual(crop.base_name, 'zea_mays_subsp_mays')
+        self.assertEqual(crop.app_name, 'Maize')
+    
+    def test_delete_crop(self):
+        self.crop.save()
+
+        self.crop.delete()
+        crop = Crop.objects(id=self.crop.id).first()
+
+        self.assertIsNone(crop)
+
     def tearDown(self):
         Crop.objects.delete()
 

src/tests/test_group.py

@@ -31,6 +31,45 @@ def test_create_group(self):
         self.assertEqual(group.crop, self.crop)        
         self.assertEqual(group.ext_id, '1234')
 
+    def test_create_duplicate_group(self):
+        # Intenta crear un grupo con el mismo ext_id que otro grupo existente
+        self.group.save()
+        group2 = Group(
+            group_name='Modern corn hybrids',
+            crop=self.crop,
+            ext_id='1234'
+        )
+        with self.assertRaises(Exception) as context:
+            group2.save()
+        
+        print(str(context.exception))
+        self.assertTrue('Tried to save duplicate unique keys (E11000 Duplicate Key Error)' in str(context.exception))
+
+    def test_create_group_without_crop(self):
+        # Intenta crear un grupo sin el objeto Crop
+        group2 = Group(
+            group_name='Landraces of corn',
+            ext_id='5678'
+        )
+        with self.assertRaises(Exception) as context:
+            group2.save()
+            
+        print(str(context.exception))
+        self.assertTrue("ValidationError (Group:None) (Field is required: ['crop'])" in str(context.exception))
+
+    def test_delete_group(self):
+        self.group.save()
+
+        self.group.delete()
+        group = Group.objects(id=self.group.id).first()
+
+        self.assertIsNone(group)
+
+    def test_get_country_by_invalid_id(self):
+        # Busca un country por un ID que no existe
+        found_country = Group.objects(crop='609e996db1751152b09f5555').first()
+        self.assertIsNone(found_country)
+
     def tearDown(self):
         self.crop.delete()
         Group.objects.delete()