test_aliases.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. # -*- coding: utf-8 -*-
  2. # Import Python libs
  3. from __future__ import absolute_import, print_function, unicode_literals
  4. # Import Salt Testing libs
  5. from tests.support.case import ModuleCase
  6. class AliasesTest(ModuleCase):
  7. '''
  8. Validate aliases module
  9. '''
  10. def test_set_target(self):
  11. '''
  12. aliases.set_target and aliases.get_target
  13. '''
  14. set_ret = self.run_function(
  15. 'aliases.set_target',
  16. alias='fred',
  17. target='bob')
  18. self.assertTrue(set_ret)
  19. tgt_ret = self.run_function(
  20. 'aliases.get_target',
  21. alias='fred')
  22. self.assertEqual(tgt_ret, 'bob')
  23. def test_has_target(self):
  24. '''
  25. aliases.set_target and aliases.has_target
  26. '''
  27. set_ret = self.run_function(
  28. 'aliases.set_target',
  29. alias='fred',
  30. target='bob')
  31. self.assertTrue(set_ret)
  32. tgt_ret = self.run_function(
  33. 'aliases.has_target',
  34. alias='fred',
  35. target='bob')
  36. self.assertTrue(tgt_ret)
  37. def test_list_aliases(self):
  38. '''
  39. aliases.list_aliases
  40. '''
  41. set_ret = self.run_function(
  42. 'aliases.set_target',
  43. alias='fred',
  44. target='bob')
  45. self.assertTrue(set_ret)
  46. tgt_ret = self.run_function(
  47. 'aliases.list_aliases')
  48. self.assertIsInstance(tgt_ret, dict)
  49. self.assertIn('fred', tgt_ret)
  50. def test_rm_alias(self):
  51. '''
  52. aliases.rm_alias
  53. '''
  54. set_ret = self.run_function(
  55. 'aliases.set_target',
  56. alias='frank',
  57. target='greg')
  58. self.assertTrue(set_ret)
  59. self.run_function(
  60. 'aliases.rm_alias',
  61. alias='frank')
  62. tgt_ret = self.run_function(
  63. 'aliases.list_aliases')
  64. self.assertIsInstance(tgt_ret, dict)
  65. self.assertNotIn('alias=frank', tgt_ret)