-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Expand file tree
/
Copy pathtest_import_restrict.py
More file actions
30 lines (22 loc) · 1004 Bytes
/
test_import_restrict.py
File metadata and controls
30 lines (22 loc) · 1004 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import sys
import os.path
import shutil
import unittest
import semmle.populator
from tests import test_utils
class ExtractorImportRestrictTest(test_utils.ExtractorTest):
def __init__(self, name):
super(ExtractorImportRestrictTest, self).__init__(name)
self.module_path = os.path.abspath(os.path.join(self.here, "data-imports"))
def test_import_unrestricted(self):
self.run_extractor("mod1")
self.check_only_traps_exists_and_clear("mod1", "mod2", "mod3", "mod4", "mod5")
def test_import_unrestricted_2(self):
self.run_extractor("mod2")
self.check_only_traps_exists_and_clear("mod2", "mod3", "mod4", "mod5")
def test_import_depth(self):
self.run_extractor("--max-import-depth", "1", "mod1")
self.check_only_traps_exists_and_clear("mod1", "mod2")
def test_import_depth_2(self):
self.run_extractor("--max-import-depth", "2", "mod1")
self.check_only_traps_exists_and_clear("mod1", "mod2", "mod3", "mod4")