Move misplaced test functions to test/test_source_osv.py

Author: Neil

test/test_osv_source.py

@@ -17,4 +17,20 @@ async def test_update_ecosystems_mocked():
 
         await osv.update_ecosystems()
 
-        assert osv.ecosystems == ["ecosystem1"]
+        assert osv.ecosystems == ["ecosystem1"]
+
+@pytest.mark.asyncio
+@patch("cve_bin_tool.data_sources.osv_source.aio_run_command", new_callable=AsyncMock)
+async def test_update_ecosystems(mock_aio_run_command):
+    # Mock gsutil output: two folders, one of them contains all.zip
+    mock_aio_run_command.side_effect = [
+        b"https://osv-vulnerabilities/python/\nhttps://osv-vulnerabilities/rust/\n",
+        b"https://osv-vulnerabilities/python/all.zip\n",
+        b"",  # rust does not contain all.zip
+    ]
+
+    osv = OSV_Source()
+    await osv.update_ecosystems()
+
+    assert "python" in osv.ecosystems
+    assert "rust" not in osv.ecosystems