From fcebe288395dec07a56a6d807b2cd0870f90c2c1 Mon Sep 17 00:00:00 2001
From: rdb <git@rdb.name>
Date: Tue, 26 May 2020 11:38:14 +0200
Subject: [PATCH] Fix current component not being cleared after component
 update

---
 src/runtime/internal/scheduler.ts                |  1 +
 .../onmount-get-current-component/_config.js     |  4 ++++
 .../onmount-get-current-component/main.svelte    | 16 ++++++++++++++++
 3 files changed, 21 insertions(+)
 create mode 100644 test/runtime/samples/onmount-get-current-component/_config.js
 create mode 100644 test/runtime/samples/onmount-get-current-component/main.svelte

diff --git a/src/runtime/internal/scheduler.ts b/src/runtime/internal/scheduler.ts
index b0db71035a..568739e4f8 100644
--- a/src/runtime/internal/scheduler.ts
+++ b/src/runtime/internal/scheduler.ts
@@ -45,6 +45,7 @@ export function flush() {
 			set_current_component(component);
 			update(component.$$);
 		}
+		set_current_component(null);
 
 		dirty_components.length = 0;
 
diff --git a/test/runtime/samples/onmount-get-current-component/_config.js b/test/runtime/samples/onmount-get-current-component/_config.js
new file mode 100644
index 0000000000..727f8a2d8e
--- /dev/null
+++ b/test/runtime/samples/onmount-get-current-component/_config.js
@@ -0,0 +1,4 @@
+export default {
+	skip_if_ssr: true,
+	html: `1`,
+};
diff --git a/test/runtime/samples/onmount-get-current-component/main.svelte b/test/runtime/samples/onmount-get-current-component/main.svelte
new file mode 100644
index 0000000000..de64bc65fa
--- /dev/null
+++ b/test/runtime/samples/onmount-get-current-component/main.svelte
@@ -0,0 +1,16 @@
+<script>
+	import { onMount } from 'svelte';
+	import { get_current_component } from 'svelte/internal';
+
+	let gotException = 0;
+
+	onMount(() => {
+		try {
+			get_current_component();
+		} catch (error) {
+			gotException++;
+		}
+	});
+</script>
+
+{gotException}