runtime error when get_current_component in on_mount

src/runtime/internal/scheduler.ts

@@ -46,6 +46,8 @@ export function flush() {
 			update(component.$$);
 		}
 
+		set_current_component(null);
+
 		dirty_components.length = 0;
 
 		while (binding_callbacks.length) binding_callbacks.pop()();

test/runtime/samples/onmount-get-current-component/_config.js

@@ -0,0 +1,7 @@
+export default {
+	error: `Function called outside component initialization`,
+	after_test() {
+		// clear cache for `svelte/internal`
+		delete require.cache[require.resolve(process.cwd() + '/internal')];
+	},
+};

test/runtime/samples/onmount-get-current-component/main.svelte

@@ -0,0 +1,7 @@
+<script>
+	import { getContext, onMount } from 'svelte';
+	import { get_current_component } from 'svelte/internal';
+	onMount(() => {
+		get_current_component();
+	});
+</script>