diff --git a/tests/components/netatmo/test_media_source.py b/tests/components/netatmo/test_media_source.py
index 390da95496a..01422dfd118 100644
--- a/tests/components/netatmo/test_media_source.py
+++ b/tests/components/netatmo/test_media_source.py
@@ -3,9 +3,14 @@ import ast
 
 import pytest
 
-from homeassistant.components import media_source
-from homeassistant.components.media_source import const
-from homeassistant.components.media_source.models import PlayMedia
+from homeassistant.components.media_source import (
+    DOMAIN as MS_DOMAIN,
+    URI_SCHEME,
+    BrowseError,
+    PlayMedia,
+    async_browse_media,
+    async_resolve_media,
+)
 from homeassistant.components.netatmo import DATA_CAMERAS, DATA_EVENTS, DOMAIN
 from homeassistant.setup import async_setup_component
 
@@ -27,59 +32,51 @@ async def test_async_browse_media(hass):
         "12:34:56:78:90:ac": "MyOutdoorCamera",
     }
 
-    assert await async_setup_component(hass, const.DOMAIN, {})
+    assert await async_setup_component(hass, MS_DOMAIN, {})
     await hass.async_block_till_done()
 
     # Test camera not exists
-    with pytest.raises(media_source.BrowseError) as excinfo:
-        await media_source.async_browse_media(
-            hass, f"{const.URI_SCHEME}{DOMAIN}/events/98:76:54:32:10:ff"
-        )
+    with pytest.raises(BrowseError) as excinfo:
+        await async_browse_media(hass, f"{URI_SCHEME}{DOMAIN}/events/98:76:54:32:10:ff")
     assert str(excinfo.value) == "Camera does not exist."
 
     # Test browse event
-    with pytest.raises(media_source.BrowseError) as excinfo:
-        await media_source.async_browse_media(
-            hass, f"{const.URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/12345"
+    with pytest.raises(BrowseError) as excinfo:
+        await async_browse_media(
+            hass, f"{URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/12345"
         )
     assert str(excinfo.value) == "Event does not exist."
 
     # Test invalid base
-    with pytest.raises(media_source.BrowseError) as excinfo:
-        await media_source.async_browse_media(
-            hass, f"{const.URI_SCHEME}{DOMAIN}/invalid/base"
-        )
+    with pytest.raises(BrowseError) as excinfo:
+        await async_browse_media(hass, f"{URI_SCHEME}{DOMAIN}/invalid/base")
     assert str(excinfo.value) == "Unknown source directory."
 
     # Test invalid base
-    with pytest.raises(media_source.BrowseError) as excinfo:
-        await media_source.async_browse_media(hass, f"{const.URI_SCHEME}{DOMAIN}/")
+    with pytest.raises(BrowseError) as excinfo:
+        await async_browse_media(hass, f"{URI_SCHEME}{DOMAIN}/")
     assert str(excinfo.value) == "Invalid media source URI"
 
     # Test successful listing
-    media = await media_source.async_browse_media(
-        hass, f"{const.URI_SCHEME}{DOMAIN}/events"
-    )
+    media = await async_browse_media(hass, f"{URI_SCHEME}{DOMAIN}/events")
 
     # Test successful listing
-    media = await media_source.async_browse_media(
-        hass, f"{const.URI_SCHEME}{DOMAIN}/events/"
-    )
+    media = await async_browse_media(hass, f"{URI_SCHEME}{DOMAIN}/events/")
 
     # Test successful events listing
-    media = await media_source.async_browse_media(
-        hass, f"{const.URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab"
+    media = await async_browse_media(
+        hass, f"{URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab"
     )
 
     # Test successful event listing
-    media = await media_source.async_browse_media(
-        hass, f"{const.URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/1599152672"
+    media = await async_browse_media(
+        hass, f"{URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/1599152672"
     )
     assert media
 
     # Test successful event resolve
-    media = await media_source.async_resolve_media(
-        hass, f"{const.URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/1599152672", None
+    media = await async_resolve_media(
+        hass, f"{URI_SCHEME}{DOMAIN}/events/12:34:56:78:90:ab/1599152672", None
     )
     assert media == PlayMedia(
         url="http:///files/high/index.m3u8", mime_type="application/x-mpegURL"