GitAuto: [FEATURE] Implement SKU Service Attachment API

database_schema.sql

@@ -0,0 +1,10 @@
+-- Create table to associate SKU Services with Attachments
+CREATE TABLE sku_service_attachments (
+    sku_service_id INT NOT NULL,
+    attachment_id INT NOT NULL,
+    PRIMARY KEY (sku_service_id, attachment_id),
+    FOREIGN KEY (sku_service_id) REFERENCES sku_services(id),
+    FOREIGN KEY (attachment_id) REFERENCES attachments(id)
+);
+
+-- Add any additional constraints or indexes as needed

sku_service_attachment_api.py

@@ -0,0 +1,20 @@
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+# Endpoint to associate an attachment with a SKU Service
+@app.route('/sku_service/attachment', methods=['POST'])
+def associate_attachment():
+    data = request.json
+    # Logic to associate attachment
+    return jsonify({"message": "Attachment associated successfully"}), 201
+
+# Endpoint to disassociate an attachment from a SKU Service
+@app.route('/sku_service/attachment', methods=['DELETE'])
+def disassociate_attachment():
+    data = request.json
+    # Logic to disassociate attachment
+    return jsonify({"message": "Attachment disassociated successfully"}), 200
+
+if __name__ == '__main__':
+    app.run(debug=True)

sku_service_attachment_service.py

@@ -0,0 +1,21 @@
+class SKUServiceAttachmentService:
+    def __init__(self, db_connection):
+        self.db = db_connection
+
+    def associate_attachment(self, sku_service_id, attachment_id):
+        # Logic to associate attachment
+        query = "INSERT INTO sku_service_attachments (sku_service_id, attachment_id) VALUES (%s, %s)"
+        self.db.execute(query, (sku_service_id, attachment_id))
+        return "Attachment associated successfully"
+
+    def disassociate_attachment(self, sku_service_id, attachment_id):
+        # Logic to disassociate attachment
+        query = "DELETE FROM sku_service_attachments WHERE sku_service_id = %s AND attachment_id = %s"
+        self.db.execute(query, (sku_service_id, attachment_id))
+        return "Attachment disassociated successfully"
+
+# Example usage
+# db_connection = get_db_connection()
+# service = SKUServiceAttachmentService(db_connection)
+# service.associate_attachment(1, 2)
+# service.disassociate_attachment(1, 2)

test_sku_service_attachment.py

@@ -0,0 +1,32 @@
+import unittest
+from sku_service_attachment_service import SKUServiceAttachmentService
+
+class TestSKUServiceAttachment(unittest.TestCase):
+    def setUp(self):
+        # Mock database connection
+        self.db_connection = MockDBConnection()
+        self.service = SKUServiceAttachmentService(self.db_connection)
+
+    def test_associate_attachment(self):
+        result = self.service.associate_attachment(1, 2)
+        self.assertEqual(result, "Attachment associated successfully")
+        self.assertIn((1, 2), self.db_connection.data)
+
+    def test_disassociate_attachment(self):
+        self.db_connection.data.add((1, 2))
+        result = self.service.disassociate_attachment(1, 2)
+        self.assertEqual(result, "Attachment disassociated successfully")
+        self.assertNotIn((1, 2), self.db_connection.data)
+
+class MockDBConnection:
+    def __init__(self):
+        self.data = set()
+
+    def execute(self, query, params):
+        if "INSERT" in query:
+            self.data.add(params)
+        elif "DELETE" in query:
+            self.data.discard(params)
+
+if __name__ == '__main__':
+    unittest.main()