[Fix] Remove all Scenarios, catch empty SMILES, prevent default Package delete (#134)

Co-authored-by: Tim Lorsbach <tim@lorsba.ch> Reviewed-on: enviPath/enviPy#134
2025-09-30 19:10:57 +13:00
parent b757a07f91
commit 3f5bb76633
9 changed files with 215 additions and 53 deletions
--- a/tests/views/test_model_views.py
+++ b/tests/views/test_model_views.py
@ -0,0 +1,113 @@
+from django.test import TestCase, override_settings
+from django.urls import reverse
+from django.conf import settings as s
+
+from epdb.logic import UserManager, PackageManager
+from epdb.models import Pathway, Edge, Package, User
+
+
+@override_settings(MODEL_DIR=s.FIXTURE_DIRS[0] / "models")
+class PathwayViewTest(TestCase):
+    fixtures = ["test_fixtures_incl_model.jsonl.gz"]
+
+    @classmethod
+    def setUpClass(cls):
+        super(PathwayViewTest, cls).setUpClass()
+        cls.user1 = UserManager.create_user("user1", "user1@envipath.com", "SuperSafe",
+                                            set_setting=True, add_to_group=True, is_active=True)
+        cls.user1_default_package = cls.user1.default_package
+        cls.model_package = Package.objects.get(name='Fixtures')
+
+    def setUp(self):
+        self.client.force_login(self.user1)
+
+    def test_predict(self):
+        self.client.force_login(User.objects.get(username="admin"))
+        response = self.client.get(
+            reverse("package model detail", kwargs={
+                'package_uuid': str(self.model_package.uuid),
+                'model_uuid': str(self.model_package.models.first().uuid)
+            }), {
+                'classify': 'ILikeCats!',
+                'smiles': 'CCN(CC)C(=O)C1=CC(=CC=C1)CO',
+            }
+        )
+
+        expected = [
+            {
+                'products': [
+                    [
+                        'O=C(O)C1=CC(CO)=CC=C1',
+                        'CCNCC'
+                    ]
+                ],
+                'probability': 0.25,
+                'btrule': {
+                    'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/0e6e9290-b658-4450-b291-3ec19fa19206',
+                    'name': 'bt0430-4011'
+                }
+            }, {
+                'products': [
+                    [
+                        'CCNC(=O)C1=CC(CO)=CC=C1',
+                        'CC=O'
+                    ]
+                ], 'probability': 0.0,
+                'btrule': {
+                    'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/27a3a353-0b66-4228-bd16-e407949e90df',
+                    'name': 'bt0243-4301'
+                }
+            }, {
+                'products': [
+                    [
+                        'CCN(CC)C(=O)C1=CC(C=O)=CC=C1'
+                    ]
+                ], 'probability': 0.75,
+                'btrule': {
+                    'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/2f2e0c39-e109-4836-959f-2bda2524f022',
+                    'name': 'bt0001-3568'
+                }
+            }
+        ]
+
+        actual = response.json()
+        self.assertEqual(actual, expected)
+
+        response = self.client.get(
+            reverse("package model detail", kwargs={
+                'package_uuid': str(self.model_package.uuid),
+                'model_uuid': str(self.model_package.models.first().uuid)
+            }), {
+                'classify': 'ILikeCats!',
+                'smiles': '',
+            }
+        )
+
+        self.assertEqual(response.status_code, 400)
+        self.assertEqual(response.json()['error'], 'Received empty SMILES')
+
+        response = self.client.get(
+            reverse("package model detail", kwargs={
+                'package_uuid': str(self.model_package.uuid),
+                'model_uuid': str(self.model_package.models.first().uuid)
+            }), {
+                'classify': 'ILikeCats!',
+                'smiles': '  ',  # Input should be stripped
+            }
+        )
+
+        self.assertEqual(response.status_code, 400)
+        self.assertEqual(response.json()['error'], 'Received empty SMILES')
+
+        response = self.client.get(
+            reverse("package model detail", kwargs={
+                'package_uuid': str(self.model_package.uuid),
+                'model_uuid': str(self.model_package.models.first().uuid)
+            }), {
+                'classify': 'ILikeCats!',
+                'smiles': 'RandomInput',
+            }
+        )
+
+        self.assertEqual(response.status_code, 400)
+        self.assertEqual(response.json()['error'], '"RandomInput" is not a valid SMILES')