"""Test get_product_managers.""" import json from tests.unit import db_utils from tests.unit.factories.oa_user import OAUserFactory from users import ( constants, # noqa ) from users.app import app def _populate_values(): """Populate test data in db.""" db_utils.seed_models(OAUserFactory.build(user_id=12, active='Y', is_product_manager=1)) expected_message = { 'items': [ { 'type': 'oa', 'user_id': 12, 'full_name': 'first name last name', 'email': 'email', 'active': 'Y', } ] } return expected_message @db_utils.test_schema def test_get_product_managers(mocker): """Test fetch product_managers from DB.""" expected_response = _populate_values() navigator = app.test_client() request = navigator.get( '/users/product_managers', headers={constants.GRASS_ACCOUNT_TYPE: 'vendor', constants.GRASS_ACCOUNT_ID: '123'}, ) assert request.status_code == 200 response_data = json.loads(request.data.decode('utf-8')) assert response_data == expected_response