20b0467f89
- extract, isolate and package it in a completely independent Python module, versioned and in a way that allows releases on PyPI.org - fixed error in placeholder for secondary school (data registry defaults) - added restriction in pytest version to install - expected number of new cases fix - data registry update (schema v2.1.1) - github update - deprecate ExpertApplication and CO2Application - changes to reflect schema update 2.0.2 - version update - Fixed error with f_inf (short-range) - new folder layout - Conditional probability data update - General fixes - Fitting results in L/S/person - CO2 fitting algorithm refinement
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
import unittest
|
|
from unittest.mock import Mock, patch
|
|
|
|
from caimira.calculator.store.data_service import DataService
|
|
|
|
|
|
class DataServiceTests(unittest.TestCase):
|
|
def setUp(self):
|
|
# Set up any necessary test data or configurations
|
|
self.data_service = DataService.create(host="https://dataservice.example.com")
|
|
|
|
@patch("requests.get")
|
|
def test_fetch_successful(self, mock_get):
|
|
# Mock successful fetch response
|
|
mock_get.return_value = Mock()
|
|
mock_get.return_value.status_code = 200
|
|
mock_get.return_value.json.return_value = {"data": "dummy_data"}
|
|
data = self.data_service._fetch()
|
|
|
|
# Assert that the data is returned correctly
|
|
self.assertEqual(data, {"data": "dummy_data"})
|
|
|
|
# Verify that the fetch method was called with the expected arguments
|
|
mock_get.assert_called_once_with(
|
|
"https://dataservice.example.com/data",
|
|
headers={
|
|
"Content-Type": "application/json",
|
|
},
|
|
)
|
|
|
|
@patch("requests.get")
|
|
def test_fetch_error(self, mock_get):
|
|
# Mock fetch error response
|
|
mock_get.return_value = Mock()
|
|
mock_get.return_value.status_code = 500
|
|
|
|
# Call the fetch method
|
|
data = self.data_service._fetch()
|
|
|
|
# Assert that the fetch method returns None in case of an error
|
|
self.assertIsNone(data)
|