Skip to content

Commit

Permalink
Update tests to use get_credentials
Browse files Browse the repository at this point in the history
  • Loading branch information
sameer committed Feb 15, 2019
1 parent 9e9f5a0 commit c9de53b
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions test/test_ultimaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,20 +90,20 @@ def test_printer_acquires_credentials_only_once(self):
class AlreadyHasCredentialsTest(unittest.TestCase):
def setUp(self):
printer = default_printer_mock()
printer.credentials = Mock(return_value=mock_credentials)
printer.get_credentials = Mock(return_value=mock_credentials)
self.printer = printer

def test_printer_is_authorized(self):
self.assertTrue(self.printer.is_authorized())
self.printer.credentials.assert_called_once()
self.printer.get_credentials.assert_called_once()
self.printer.post_auth_request.assert_not_called()
self.printer.get_auth_check.assert_called_once()


class UltimakerJsonTest(unittest.TestCase):
def test_expected_json_is_produced_when_idle(self):
printer = default_printer_mock()
printer.credentials = Mock(return_value=mock_credentials)
printer.get_credentials = Mock(return_value=mock_credentials)
printer.get_camera_snapshot_uri = Mock(
return_value=mock_camera_snapshot_uri
)
Expand All @@ -122,7 +122,7 @@ def test_expected_json_is_produced_when_idle(self):

def test_expected_json_is_produced_when_errored(self):
printer = default_printer_mock()
printer.credentials = Mock(return_value=mock_credentials)
printer.get_credentials = Mock(return_value=mock_credentials)
printer.get_printer_status = generic_exception_raiser
printer.get_print_job_state = Mock(return_value=mock_print_job_state)
json = printer.into_ultimaker_json()
Expand All @@ -132,7 +132,7 @@ def test_expected_json_is_produced_when_errored(self):

def test_expected_json_is_produced_when_printing(self):
printer = default_printer_mock()
printer.credentials = Mock(return_value=mock_credentials)
printer.get_credentials = Mock(return_value=mock_credentials)
printer.get_printer_status = Mock(return_value='printing')
printer.get_print_job_time_elapsed = Mock(
return_value=mock_print_job_time_elapsed)
Expand Down

0 comments on commit c9de53b

Please sign in to comment.