old_contents: # -*- coding: utf-8 -*-
  '''
      :codeauthor: :email:`Daniel Mizyrycki (mzdaniel@glidelink.net)`
  
  
      tests.integration.cli.custom_module
      ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  
      Test salt-ssh sls with a custom module work.
  
      $ cat srv/custom_module.sls
      custom-module:
        module.run:
          - name: test.recho
          - text: hello
  
  
      $ cat srv/_modules/override_test.py
      __virtualname__ = 'test'
  
      def __virtual__():
          return __virtualname__
  
      def recho(text):
          return text[::-1]
  
  
      $ salt-ssh localhost state.sls custom_module
      localhost:
          olleh
  
  '''
  
  # Import Salt Libs
  import integration
  
  
  class SSHCustomModuleTest(integration.SSHCase):
      '''
      Test sls with custom module functionality using ssh
      '''
  
      def test_ssh_regular_module(self):
          '''
          Test regular module work using SSHCase environment
          '''
          expected = 'hello'
          cmd = self.run_function('test.echo', arg=['hello'])
          self.assertEqual(expected, cmd)
  
  
      def test_ssh_custom_module(self):
          '''
          Test custom module work using SSHCase environment
          '''
          expected = 'hello'[::-1]
          cmd = self.run_function('test.recho', arg=['hello'])
          self.assertEqual(expected, cmd)
  
  
      def test_ssh_sls_with_custom_module(self):
          '''
          Test sls with custom module work using SSHCase environment
          '''
          expected = {
              "module_|-regular-module_|-test.echo_|-run": 'hello',
              "module_|-custom-module_|-test.recho_|-run": 'olleh'}
          cmd = self.run_function('state.sls', arg=['custom_module'])
          for key in cmd:
              if not cmd[key]['result']:
                  raise AssertionError(cmd[key]['comment'])
              cmd_ret = cmd[key]['changes'].get('ret', None)
              self.assertEqual(cmd_ret, expected[key])
  
  
  if __name__ == '__main__':
      '''
      This test can be run in a small test suite with:
  
      python tests/runtests.py -C --ssh
      '''
      from integration import run_tests
      run_tests(SSHCustomModuleTest)
  
new_contents: # -*- coding: utf-8 -*-
  '''
      :codeauthor: :email:`Daniel Mizyrycki (mzdaniel@glidelink.net)`
  
  
      tests.integration.cli.custom_module
      ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  
      Test salt-ssh sls with a custom module work.
  
      $ cat srv/custom_module.sls
      custom-module:
        module.run:
          - name: test.recho
          - text: hello
  
  
      $ cat srv/_modules/override_test.py
      __virtualname__ = 'test'
  
      def __virtual__():
          return __virtualname__
  
      def recho(text):
          return text[::-1]
  
  
      $ salt-ssh localhost state.sls custom_module
      localhost:
          olleh
  
  
      This test can be run in a small test suite with:
  
      $ python tests/runtests.py -C --ssh
  '''
  
  # Import Salt Libs
  import integration
  
  
  class SSHCustomModuleTest(integration.SSHCase):
      '''
      Test sls with custom module functionality using ssh
      '''
  
      def test_ssh_regular_module(self):
          '''
          Test regular module work using SSHCase environment
          '''
          expected = 'hello'
          cmd = self.run_function('test.echo', arg=['hello'])
          self.assertEqual(expected, cmd)
  
      def test_ssh_custom_module(self):
          '''
          Test custom module work using SSHCase environment
          '''
          expected = 'hello'[::-1]
          cmd = self.run_function('test.recho', arg=['hello'])
          self.assertEqual(expected, cmd)
  
      def test_ssh_sls_with_custom_module(self):
          '''
          Test sls with custom module work using SSHCase environment
          '''
          expected = {
              "module_|-regular-module_|-test.echo_|-run": 'hello',
              "module_|-custom-module_|-test.recho_|-run": 'olleh'}
          cmd = self.run_function('state.sls', arg=['custom_module'])
          for key in cmd:
              if not cmd[key]['result']:
                  raise AssertionError(cmd[key]['comment'])
              cmd_ret = cmd[key]['changes'].get('ret', None)
              self.assertEqual(cmd_ret, expected[key])
  
  
  if __name__ == '__main__':
      from integration import run_tests
      run_tests(SSHCustomModuleTest)
  
subject: Fix lint