def test_recv_return_unavailable(self): ''' Test if recv returns unavailable. ''' files = {'saltines': '/srv/salt/saltines', 'biscuits': '/srv/salt/biscuits'} dest = '/srv/salt/cheese' self.assertEqual(cp.recv(files, dest), 'Destination unavailable')
def test_recv_return_success(self): ''' Test if recv returns success. ''' files = {'saltines': 'salt://saltines', 'biscuits': 'salt://biscuits'} ret = {'/srv/salt/cheese/saltines': True, '/srv/salt/cheese/biscuits': True} dest = '/srv/salt/cheese' file_data = 'Remember to keep your files well salted.' with patch('salt.utils.fopen', mock_open(read_data=file_data)): self.assertEqual(cp.recv(files, dest), ret)