old_contents: #
  # Project:   retdec-python
  # Copyright: (c) 2015 by Petr Zemek <s3rvac@gmail.com> and contributors
  # License:   MIT, see the LICENSE file for more details
  #
  
  """Tests for the :mod:`retdec.resource` module."""
  
  import unittest
  from unittest import mock
  
  from retdec.conn import APIConnection
  from retdec.resource import Resource
  
  
  class ResourceTests(unittest.TestCase):
      """Tests for :class:`retdec.resource.Resource`."""
  
      def test_id_returns_passed_id(self):
          r = Resource('ID', mock.Mock(spec_set=APIConnection))
          self.assertEqual(r.id, 'ID')
  
      def test_wait_until_finished_returns_when_resource_is_finished(self):
          conn_mock = mock.Mock(spec_set=APIConnection)
          conn_mock.send_get_request.return_value = {'finished': True}
  
          r = Resource('ID', conn_mock)
  
          r.wait_until_finished()
  
          conn_mock.send_get_request.assert_called_once_with('/ID/status')
  
new_contents: #
  # Project:   retdec-python
  # Copyright: (c) 2015 by Petr Zemek <s3rvac@gmail.com> and contributors
  # License:   MIT, see the LICENSE file for more details
  #
  
  """Tests for the :mod:`retdec.resource` module."""
  
  import unittest
  from unittest import mock
  
  from retdec.conn import APIConnection
  from retdec.resource import Resource
  
  
  class ResourceTests(unittest.TestCase):
      """Tests for :class:`retdec.resource.Resource`."""
  
      def test_id_returns_passed_id(self):
          r = Resource('ID', mock.Mock(spec_set=APIConnection))
          self.assertEqual(r.id, 'ID')
  
  
  class ResourceWaitUntilFinishedTests(unittest.TestCase):
      """Tests for :func:`retdec.resource.Resource.wait_until_finished()`."""
  
      def test_returns_when_resource_is_finished(self):
          conn_mock = mock.Mock(spec_set=APIConnection)
          conn_mock.send_get_request.return_value = {'finished': True}
  
          r = Resource('ID', conn_mock)
  
          r.wait_until_finished()
  
          conn_mock.send_get_request.assert_called_once_with('/ID/status')
  
subject: Move tests for Resource.wait_until_finished() into a separate class.