Ensure uniq project name

hitier
1 parent 15a97bc4
Showing 4 changed files with 46 additions and 6 deletions Show diff stats
app/main/routes.py
app/models.py
resources/lesia-btp.sqlite
tests/frontend_tests.py
@@ -268,6 +268,11 @@ def project_edit(project_id=None):
             this_project = Project.query.get(int(project_id))
             done_string = "mis à jour."
         else:
+            # check name doesnt exist yet
+            existing_project = Project.query.filter(Project.name == request.form.get('name')).one_or_none()
+            if existing_project:
+                flash(f"Le projet >{existing_project.name}< existe déjà.", 'warning')
+                return redirect(url_for('main.project_edit'))
             # or create from scratch
             this_project = Project()
             done_string = "ajouté."
@@ -54,7 +54,7 @@ class Formable:
  
 class Project(db.Model, Formable):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String)
+    name = db.Column(db.String, unique=True)
     labels = relationship("ProjectLabel", back_populates="project")
  
     # add keys to import/export
@@ -161,7 +161,7 @@ class CategoryLabel(db.Model):
  
 class Category(db.Model, Formable):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String)
+    name = db.Column(db.String, unique=True)
     labels = relationship("CategoryLabel", back_populates="category")
  
     export_keys = ['name']
@@ -198,25 +198,25 @@ class Category(db.Model, Formable):
  
 class AgentBap(db.Model):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(16))
+    name = db.Column(db.String(16), unique=True)
     agents = relationship("Agent", back_populates="bap")
  
  
 class AgentGrade(db.Model):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(16))
+    name = db.Column(db.String(16), unique=True)
     agents = relationship("Agent", back_populates="grade")
  
  
 class AgentStatus(db.Model):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(16))
+    name = db.Column(db.String(16), unique=True)
     agents = relationship("Agent", back_populates="status")
  
  
 class Company(db.Model):
     id = db.Column(db.Integer, primary_key=True)
-    name = db.Column(db.String(16))
+    name = db.Column(db.String(16), unique=True)
     agents = relationship("Agent", back_populates="company")
  
  
@@ -199,6 +199,41 @@ class FormsTestCase(BaseFrontTestCase):
         latest_agent = Project.query.filter(Project.id == 1).one()
         self.assertEqual(project_name, latest_agent.name)
  
+    # Test project form
+    def test_project_add_twice(self):
+        project_name = "My Project Name"
+        # load the form
+        target_url = self.get_server_url() + url_for('main.project_edit')
+        self.driver.get(target_url)
+        # fill it in
+        name_input = self.driver.find_elements_by_xpath("//input[@id='name']")[0]
+        name_input.send_keys(project_name)
+        # submit
+        submit_button = self.driver.find_elements_by_xpath("//input[@type='submit']")[0]
+        submit_button.send_keys(Keys.ENTER)
+        #
+        # check page return
+        message = self.driver.find_element_by_xpath("//div[@role='alert']")
+        self.assertTrue(project_name in message.text)
+        #
+        # check on database
+        latest_agent = Project.query.order_by(Project.id.desc()).first()
+        self.assertEqual(project_name, latest_agent.name)
+        #
+        # Submit again
+        target_url = self.get_server_url() + url_for('main.project_edit')
+        self.driver.get(target_url)
+        # fill it in
+        name_input = self.driver.find_elements_by_xpath("//input[@id='name']")[0]
+        name_input.send_keys(project_name)
+        # submit
+        submit_button = self.driver.find_elements_by_xpath("//input[@type='submit']")[0]
+        submit_button.send_keys(Keys.ENTER)
+        #
+        # check warning message
+        message = self.driver.find_element_by_xpath("//div[@role='alert']")
+        self.assertTrue(project_name + "< existe déjà" in message.text)
+
     # Test charge form add new
     def test_charge_add(self):
         # load the form