"""Tests for ows-features service.""" from unittest.mock import patch from sound_recordings.services import ows_features class TestOWSFeaturesGetMobileFeaturesParallelResponse: """Test call to ows-features for features parallel response.""" user_ids = ["alw:1", "alw:2"] @patch("sound_recordings.services.ows_features.parallel") def test_parallel_is_called_properly(self, mock_parallel): """Test parallel is called.""" ows_features.get_mobile_features_parallel(self.user_ids) mock_parallel.assert_called_with( { self.user_ids[0]: ows_features._features_parallel_partial( self.user_ids[0] ), self.user_ids[1]: ows_features._features_parallel_partial( self.user_ids[1] ), } )