test fallback tracker store

RasaHQ · Nov 25, 2019 · 3adbdf8 · 3adbdf8
1 parent e2a2ef7
commit 3adbdf8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/core/test_tracker_stores.py b/tests/core/test_tracker_stores.py
@@ -370,10 +370,12 @@ def test_fail_safe_tracker_store_with_retrieve_error():
     on_error_callback.assert_called_once()
 
 
-def test_set_fail_safe_tracker_store_domain(default_domain):
+def test_set_fail_safe_tracker_store_domain(default_domain: Domain):
     tracker_store = InMemoryTrackerStore(domain)
-    failsafe_store = FailSafeTrackerStore(tracker_store, None)
+    fallback_tracker_store = InMemoryTrackerStore(None)
+    failsafe_store = FailSafeTrackerStore(tracker_store, None, fallback_tracker_store)
 
     failsafe_store.domain = default_domain
+    assert failsafe_store.domain == default_domain
     assert tracker_store.domain is failsafe_store.domain
-    assert tracker_store.domain == default_domain
+    assert fallback_tracker_store.domain is failsafe_store.domain