Adding add_setitem_queries tests to the prep template tests

Author: josenavas

qiita_db/metadata_template/test/test_prep_template.py

@@ -53,6 +53,45 @@ def setUp(self):
 
 
 class TestPrepSampleReadOnly(BaseTestPrepSample):
+    def test_add_setitem_queries_error(self):
+        conn_handler = SQLConnectionHandler()
+        queue = "test_queue"
+        conn_handler.create_queue(queue)
+
+        with self.assertRaises(QiitaDBColumnError):
+            self.tester.add_setitem_queries(
+                'COL_DOES_NOT_EXIST', 'Foo', conn_handler, queue)
+
+    def test_add_setitem_queries_required(self):
+        conn_handler = SQLConnectionHandler()
+        queue = "test_queue"
+        conn_handler.create_queue(queue)
+
+        self.tester.add_setitem_queries(
+            'center_name', 'FOO', conn_handler, queue)
+
+        obs = conn_handler.queues[queue]
+        sql = """UPDATE qiita.common_prep_info
+                     SET center_name=%s
+                     WHERE sample_id=%s"""
+        exp = [(sql, ('FOO', '1.SKB8.640193'))]
+        self.assertEqual(obs, exp)
+
+    def test_add_setitem_queries_dynamic(self):
+        conn_handler = SQLConnectionHandler()
+        queue = "test_queue"
+        conn_handler.create_queue(queue)
+
+        self.tester.add_setitem_queries(
+            'barcodesequence', 'AAAAAAAAAAAA', conn_handler, queue)
+
+        obs = conn_handler.queues[queue]
+        sql = """UPDATE qiita.prep_1
+                         SET barcodesequence=%s
+                         WHERE sample_id=%s"""
+        exp = [(sql, ('AAAAAAAAAAAA', '1.SKB8.640193'))]
+        self.assertEqual(obs, exp)
+
     def test_init_unknown_error(self):
         """Init errors if the PrepSample id is not found in the template"""
         with self.assertRaises(QiitaDBUnknownIDError):

qiita_db/metadata_template/test/test_sample_template.py

@@ -61,9 +61,8 @@ def test_add_setitem_queries_error(self):
         with self.assertRaises(QiitaDBColumnError):
             self.tester.add_setitem_queries(
                 'COL_DOES_NOT_EXIST', 0.30, conn_handler, queue)
-            self.tester['column that does not exist'] = 0.30
 
-    def test_add_setitem_queries_dynamic(self):
+    def test_add_setitem_queries_required(self):
         conn_handler = SQLConnectionHandler()
         queue = "test_queue"
         conn_handler.create_queue(queue)
@@ -78,7 +77,7 @@ def test_add_setitem_queries_dynamic(self):
         exp = [(sql, (True, '1.SKB8.640193'))]
         self.assertEqual(obs, exp)
 
-    def test_add_setitem_queries_required(self):
+    def test_add_setitem_queries_dynamic(self):
         conn_handler = SQLConnectionHandler()
         queue = "test_queue"
         conn_handler.create_queue(queue)