From d9979861cb196567fe3e6399720bdd845995f667 Mon Sep 17 00:00:00 2001
From: drunest <venus1004@proton.me>
Date: Tue, 10 Sep 2024 00:15:33 -0500
Subject: [PATCH] feat: add signautre test

---
 app/src/auto_validator/core/tests/test_api.py | 25 +++++++++++-
 .../core/tests/test_decorators.py             | 39 +++++++++++++++++++
 2 files changed, 63 insertions(+), 1 deletion(-)
 create mode 100644 app/src/auto_validator/core/tests/test_decorators.py

diff --git a/app/src/auto_validator/core/tests/test_api.py b/app/src/auto_validator/core/tests/test_api.py
index 642b3aa..6719b91 100644
--- a/app/src/auto_validator/core/tests/test_api.py
+++ b/app/src/auto_validator/core/tests/test_api.py
@@ -1,15 +1,19 @@
 import io
 from datetime import datetime
+from unittest import mock
 
 import pytest
 from rest_framework import status
+from rest_framework.exceptions import PermissionDenied
 
 from auto_validator.core.models import UploadedFile
 
 V1_FILES_URL = "/api/v1/files/"
 
 
-def test_file_upload(api_client, user, eq):
+@mock.patch("auto_validator.core.utils.decorators.verify_signature_and_route_subnet", lambda x: x)
+@pytest.mark.django_db
+def test_file_upload_with_valid_signature(api_client, user, eq):
     file_content = io.BytesIO(b"file content")
     file_content.name = "testfile.txt"
 
@@ -37,6 +41,25 @@ def test_file_upload(api_client, user, eq):
     assert uploaded_file.file_size == 12
 
 
+@pytest.mark.django_db
+def test_file_upload_with_invalid_signature(api_client, user):
+    file_content = io.BytesIO(b"file content")
+    file_content.name = "testfile.txt"
+
+    file_data = {
+        "file": file_content,
+    }
+
+    with mock.patch(
+        "auto_validator.core.utils.decorators.verify_signature_and_route_subnet",
+        side_effect=PermissionDenied("Invalid signature"),
+    ):
+        with pytest.raises(PermissionDenied):
+            api_client.post(V1_FILES_URL, file_data, format="multipart")
+
+    assert UploadedFile.objects.count() == 0
+
+
 @pytest.mark.django_db
 def test_list_files(api_client, user, django_user_model, eq):
     UploadedFile.objects.create(file_name="file1.txt", file_size=1, user=user, storage_file_name="file1.txt")
diff --git a/app/src/auto_validator/core/tests/test_decorators.py b/app/src/auto_validator/core/tests/test_decorators.py
new file mode 100644
index 0000000..d855da7
--- /dev/null
+++ b/app/src/auto_validator/core/tests/test_decorators.py
@@ -0,0 +1,39 @@
+import time
+from unittest.mock import Mock, patch
+
+import pytest
+from rest_framework.response import Response
+
+from auto_validator.core.utils.decorators import verify_signature_and_route_subnet
+
+
+@patch("auto_validator.core.utils.decorators.bt.Keypair")
+@patch("auto_validator.core.utils.decorators.get_subnets_by_hotkeys")
+@patch("auto_validator.core.utils.decorators.send_messages")
+@pytest.mark.django_db
+def test_valid_signature(mock_send_messages, mock_get_subnets_by_hotkeys, mock_keypair):
+    mock_get_subnets_by_hotkeys.return_value = ["subnet1"]
+    mock_keypair.return_value.verify.return_value = True
+
+    @verify_signature_and_route_subnet
+    def mock_view(view, *args, **kwargs):
+        return Response({"detail": "Success"}, status=200)
+
+    request = Mock()
+    request.headers = {
+        "Nonce": str(time.time()),
+        "Hotkey": "5GHbqzeqBBVkBjA94N4Hq2ar4WSRKFGyWWBzkWuWpn8v1vZu",
+        "Signature": "d470af4f62165069982c8fc6ef4606c962aa0ec8e39ed031d1faee3aabc26467cb4c115318040fbb066129691e1ef5eb2a480367640ff888149bf7653f790280",
+        "Authorization": "Token 2920141af1f81b1d41b0ca3d6a170f3122e47909",
+    }
+    request.method = "POST"
+    request.build_absolute_uri.return_value = "http://testserver/test"
+    request.FILES = {"file": Mock()}
+    request.FILES["file"].read.return_value = b"file_content"
+
+    view = Mock()
+    view.request = request
+
+    response = mock_view(view)
+    assert response.status_code == 200
+    assert response.data == {"detail": "Success"}