mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
tests.unit.utils.extend_test
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Test the salt extend script, leave templates/test alone to keep this working!
|
|
'''
|
|
|
|
# Import python libs
|
|
from __future__ import absolute_import
|
|
|
|
import os
|
|
import shutil
|
|
from datetime import date
|
|
|
|
# Import salt libs
|
|
import salt.utils.extend
|
|
from salt.utils import fopen
|
|
|
|
# Import Salt Testing libs
|
|
from salttesting import TestCase
|
|
from salttesting.helpers import ensure_in_syspath
|
|
ensure_in_syspath('../../')
|
|
|
|
|
|
class ExtendTestCase(TestCase):
|
|
def setUp(self):
|
|
self.out = None
|
|
|
|
def tearDown(self):
|
|
if self.out is not None:
|
|
if os.path.exists(self.out):
|
|
shutil.rmtree(self.out, True)
|
|
|
|
def test_run(self):
|
|
out = salt.utils.extend.run('test', 'test', 'this description', '.', False)
|
|
self.out = out
|
|
year = date.today().strftime('%Y')
|
|
self.assertTrue(os.path.exists(out))
|
|
self.assertFalse(os.path.exists(os.path.join(out, 'template.yml')))
|
|
self.assertTrue(os.path.exists(os.path.join(out, 'directory')))
|
|
self.assertTrue(os.path.exists(os.path.join(out, 'directory', 'test.py')))
|
|
with fopen(os.path.join(out, 'directory', 'test.py'), 'r') as test_f:
|
|
self.assertEqual(test_f.read(), year)
|
|
|
|
if __name__ == '__main__':
|
|
from unit import run_tests
|
|
run_tests(ExtendTestCase, needs_daemon=False)
|