Skip to content
Snippets Groups Projects
test_finders.py 2.94 KiB
from django.core.files.storage import FileSystemStorage
from django.test import TestCase

from ..yarn import yarn_adapter
from ..finders import NodeModulesFinder
from .. import conf
from .base import BaseYarnCase
import os
import os.path
import shutil


class _MakeDirsTestCase(TestCase):
    """
    Helper to create and clean up test directories.
    """

    def setUp(self):
        super(_MakeDirsTestCase, self).setUp()
        self.created = []

    def tearDown(self):
        super(_MakeDirsTestCase, self).tearDown()
        for name in self.created:
            if os.path.exists(name):
                shutil.rmtree(name)

    def makedirs(self, name):
        """
        Wrap os.makedirs() to delete the created directory on teardown.
        """
        os.makedirs(name)
        self.created.append(name)


class SimpleNodeModulesFinderCase(_MakeDirsTestCase):
    """
    Simple YarnFinder tests, without any packages installed.
    """

    def test_list_nonexistent(self):
        """
        When no Yarn folder exists, just gracefully find nothing.
        """
        finder = NodeModulesFinder()
        self.assertEqual(finder.locations, [])
        self.assertEqual(finder.storages, {})

    def test_list_existent(self, leaf_name='node_modules'):
        """
        If 'node_modules' exists, use it to to find files.
        """
        root = os.path.join(conf.NODE_MODULES_ROOT, leaf_name)
        self.makedirs(root)
        finder = NodeModulesFinder()

        self.assertEqual(finder.locations, [('', root)])

        self.assertEqual(set(finder.storages.keys()), set([root]))
        storage = finder.storages[root]
        self.assertIsInstance(storage, FileSystemStorage)
        self.assertEqual(storage.prefix, '')
        self.assertEqual(storage.location, root)

    def test_list_old_path(self):
        """
        If only the old 'node_modules' folder exists, use it instead.
        """
        self.test_list_existent(leaf_name='node_modules')

    def test_list_both(self):
        """
        If both folders exist, only 'yarn_node_modules' should be used.
        """
        self.test_list_existent(leaf_name='node_modules')


class NodeModulesFinderCase(BaseYarnCase):
    """Test finding installed with yarn files"""

    def setUp(self):
        super(NodeModulesFinderCase, self).setUp()
        yarn_adapter.install(['jquery@1.9'])
        self.finder = NodeModulesFinder()

    def test_find(self):
        """Test staticfinder find"""
        test_path = os.path.join('jquery', 'jquery.min.js')
        path = self.finder.find(test_path)
        self.assertEqual(path, os.path.join(
            conf.NODE_MODULES_ROOT, 'node_modules', test_path,
        ))

    def test_list(self):
        """Test staticfinder list"""
        test_path = os.path.join('jquery', 'jquery.min.js')
        result = self.finder.list([])
        matched = [
            part for part in result if part[0] == test_path
        ]
        self.assertEqual(len(matched), 1)