Added test for GetOrphanTasks

uber · May 23, 2024 · 359cb46 · 359cb46
1 parent 9363cfe
commit 359cb46
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/common/persistence/nosql/nosql_task_store_test.go b/common/persistence/nosql/nosql_task_store_test.go
@@ -75,6 +75,17 @@ func setupNoSQLStoreMocks(t *testing.T) (*nosqlTaskStore, *nosqlplugin.MockDB) {
  return store, dbMock
 }
 
+func TestGetOrphanTasks(t *testing.T) {
+ store, _ := setupNoSQLStoreMocks(t)
+
+ // We just expect the function to return an error so we don't need to check the result
+ _, err := store.GetOrphanTasks(ctx.Background(), nil)
+
+ var expectedErr *types.InternalServiceError
+ assert.ErrorAs(t, err, &expectedErr)
+ assert.ErrorContains(t, err, "Unimplemented call to GetOrphanTasks for NoSQL")
+}
+
 func TestGetTaskListSize(t *testing.T) {
  store, db := setupNoSQLStoreMocks(t)