import os

from karton.android import Android

from karton.core import Task
from karton.core.test import KartonTestCase, TestResource


class AndroidMagicTestCase(KartonTestCase):
    karton_class = Android

    def test_android(self):
        testcase = os.path.join(os.path.dirname(__file__), "testsdata", "example.apk")
        with self.subTest(testcase):
            with open(testcase, "rb") as f:
                content = f.read()
            sample = TestResource(testcase, content)
            expected = Task(
                {
                    "type": "sample",
                    "stage": "analyzed",
                    "origin": "karton.android",
                },
                payload={
                    "sample": sample,
                    "attributes": {
                        "certificate": ["61ED377E85D386A8DFEE6B864BD85B0BFAA5AF81"],
                        "main_activity": [
                            "com.example.android.contactmanager..ContactManager"
                        ],
                        "package": ["com.example.android.contactmanager"],
                        "activities": [
                            "com.example.android.contactmanager..ContactManager",
                            "com.example.android.contactmanager.ContactAdder",
                        ],
                        "permissions": [
                            "android.permission.GET_ACCOUNTS",
                            "android.permission.READ_CONTACTS",
                            "android.permission.WRITE_CONTACTS",
                        ],
                        "certificate_not_after": ["Jul 17 01:33:46 2035 UTC"],
                        "certificate_serial": ["10623618503190643167"],
                        "certificate_subject": [
                            "Email Address: android@android.com,"
                            " Common Name: Android,"
                            " Organizational Unit: Android,"
                            " Organization: Android,"
                            " Locality: Mountain View,"
                            " State/Province: California,"
                            " Country: US"
                        ],
                        "certificate_not_before": ["Feb 29 01:33:46 2008 UTC"],
                        "certificate_issuer": [
                            "Email Address: android@android.com,"
                            " Common Name: Android,"
                            " Organizational Unit: Android,"
                            " Organization: Android,"
                            " Locality: Mountain View,"
                            " State/Province: California,"
                            " Country: US"
                        ],
                        "app_name": ["Contact Manager"],
                    },
                },
            )
            task = Task(
                {
                    "type": "sample",
                    "extension": "apk",
                },
                payload={"sample": sample},
            )
            results = self.run_task(task)

            self.assertTasksEqual(results, [expected])