import time
import json
from collections import defaultdict
import pytest

from base.client_base import TestcaseBase
from common import common_func as cf
from common import common_type as ct
from deploy.common import get_all_collections
from common.common_type import CaseLabel
from utils.util_log import test_log as log


class TestGetCollections(TestcaseBase):
    """ Test case of end to end"""

    def teardown_method(self, method):
        log.info(("*" * 35) + " teardown " + ("*" * 35))
        log.info("[teardown_method] Start teardown test case %s..." %
                    method.__name__)

    @pytest.mark.tags(CaseLabel.L1)
    def test_get_collections_by_prefix(self,):
        self._connect()
        all_collections = self.utility_wrap.list_collections()[0]
        all_collections = [c_name for c_name in all_collections if "test_reinstall" in c_name or "test_upgrade" in c_name]
        log.info(f"find {len(all_collections)} collections:")
        log.info(all_collections)
        data = {
            "all": all_collections
        }
        with open("/tmp/ci_logs/all_collections.json", "w") as f:
            f.write(json.dumps(data))
        log.info(f"write {len(all_collections)} collections to /tmp/ci_logs/all_collections.json")
        collections_in_json = get_all_collections()
        assert len(all_collections) == len(collections_in_json)