push-mirage-db-into-store.js

// tests/helpers/push-mirage-db-into-store.js
import { registerAsyncHelper } from '@ember/test';
import { run } from '@ember/runloop';

let pushMirageDbIntoStore = function(server, store) {
  let tables = Object.keys(server.schema);

  tables.forEach(table => {
    if (server.schema[table].all) {
      let all = server.schema[table].all();
      let modelName = all.modelName;

      let serializer = server.serializerOrRegistry
        .serializerFor(modelName);

      let originalAlwaysIncludeLinkageData = serializer.alwaysIncludeLinkageData;
      serializer.alwaysIncludeLinkageData = true;

      let json = serializer.serialize(all);

      serializer.alwaysIncludeLinkageData = originalAlwaysIncludeLinkageData;

      run(() => {
        store.pushPayload(json);
      });
    }
  });
}

export  { pushMirageDbIntoStore };

export default registerAsyncHelper('pushMirageDbIntoStore', function(app) {
  let store = app.__container__.lookup('service:store');
  return pushMirageDbIntoStore(server, store);
});