import subprocess
import tempfile

class TestIgnoreDir(unittest.TestCase):
    '''Test --ignore-dir for --listfuncs and --trackcalls.'''

    def setUp(self):
        self.igdir = os.path.dirname(sys.modules['os'].__file__)
        fd, self.tempname = tempfile.mkstemp()
        f = os.fdopen(fd, 'w')
        f.write('''\
import os
import sys

def main():
    os.path.join('foo', 'bar')

if __name__ == '__main__':
    main()
''')
        f.close()

    def tearDown(self):
        os.unlink(self.tempname)

    def test_listfuncs(self):
        output = subprocess.check_output(
            [sys.executable, '-m', 'trace', '--listfuncs', '--ignore-dir',
             self.igdir, self.tempname]
        )
        output = output.decode()
        self.assertTrue(self.igdir not in output)

    def test_trackcalls(self):
        output = subprocess.check_output(
            [sys.executable, '-m', 'trace', '--trackcalls', '--ignore-dir',
             self.igdir, self.tempname]
        )
        output = output.decode()
        self.assertTrue(self.igdir not in output)