def test_get_next_number(self):
"""Verify the client can get the next number for a document."""
mock_response = Mock()
mock_response.status_code = 200
mock_response.json = Mock(return_value={'next': 42})
mock_post = Mock(return_value=mock_response)
# Act
with patch('requests.post', mock_post):
number = client.get_next_number('PREFIX')
# Assert
url = 'http://1.2.3.4:8080/documents/PREFIX/numbers'
headers = {'content-type': 'application/json'}
mock_post.assert_called_once_with(url, headers=headers)
self.assertEqual(42, number)
def test_get_next_number(self):
"""Verify the client can get the next number for a document."""
mock_response = Mock()
mock_response.status_code = 200
mock_response.json = Mock(return_value={'next': 42})
mock_post = Mock(return_value=mock_response)
# Act
with patch('requests.post', mock_post):
number = client.get_next_number('PREFIX')
# Assert
url = 'http://1.2.3.4:8080/documents/PREFIX/numbers'
headers = {'content-type': 'application/json'}
mock_post.assert_called_once_with(url, headers=headers)
self.assertEqual(42, number)
def test_get_next_number_no_server(self):
"""Verify the next number for a document is None with no server."""
self.assertIs(None, client.get_next_number('PREFIX'))
def test_get_next_number_no_server(self):
"""Verify the next number for a document is None with no server."""
self.assertIs(None, client.get_next_number('PREFIX'))