mirror of https://github.com/deadc0de6/catcli
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
984 B
Python
46 lines
984 B
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2017, deadc0de6
|
|
|
|
Basic unittest for graph
|
|
"""
|
|
|
|
import unittest
|
|
import tempfile
|
|
import os
|
|
|
|
from catcli.catcli import cmd_graph
|
|
from catcli.noder import Noder
|
|
from catcli.catalog import Catalog
|
|
from tests.helpers import clean, get_fakecatalog
|
|
|
|
|
|
class TestGraph(unittest.TestCase):
|
|
|
|
def test_graph(self):
|
|
# init
|
|
path = 'fake'
|
|
gpath = tempfile.gettempdir() + os.sep + 'graph.dot'
|
|
self.addCleanup(clean, path)
|
|
self.addCleanup(clean, gpath)
|
|
catalog = Catalog(path, force=True, verbose=False)
|
|
top = catalog._restore_json(get_fakecatalog())
|
|
noder = Noder()
|
|
|
|
# create fake args dict
|
|
args = {'<path>': gpath, '--verbose': True}
|
|
|
|
# create the graph
|
|
cmd_graph(args, noder, top)
|
|
|
|
# ensure file exists
|
|
self.assertTrue(os.path.exists(gpath))
|
|
|
|
|
|
def main():
|
|
unittest.main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|