import { Params } from "upload-diagnostic-plugin/types/Params";
import { MemoryTestFragmented } from "upload-diagnostic-plugin/MemoryTestFragmented";
import { transform } from "upload-plugin-sdk";
import { assertUnreachable } from "upload-diagnostic-plugin/common/TypeUtils";
import { MemoryTestContiguous } from "upload-diagnostic-plugin/MemoryTestContiguous";

const memoryTestContiguous = new MemoryTestContiguous();
const memoryTestFragmented = new MemoryTestFragmented();

export default transform<Params>({
  run: async ({ params }) => {
    switch (params.mode) {
      case "memory_test_contiguous":
        return {
          async: memoryTestContiguous.run(params)
        }
      case "memory_test_fragmented":
        return {
          async: memoryTestFragmented.run()
        }
      default:
        assertUnreachable(params);
    }
  }
});