chrome/browser/ash/fileapi/test/fake_recent_source.cc

// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ash/fileapi/test/fake_recent_source.h"

#include <string>
#include <utility>

#include "chrome/browser/ash/fileapi/recent_file.h"
#include "net/base/mime_util.h"

namespace ash {

FakeRecentSource::FakeRecentSource() : lag_(base::Seconds(0)) {}

FakeRecentSource::~FakeRecentSource() = default;

void FakeRecentSource::AddFile(const RecentFile& file) {
  canned_files_.emplace_back(file);
}

void FakeRecentSource::GetRecentFiles(Params params) {
  timer_.Start(FROM_HERE, lag_,
               base::BindOnce(&FakeRecentSource::OnFilesReady,
                              base::Unretained(this), std::move(params)));
}

void FakeRecentSource::OnFilesReady(Params params) {
  std::move(params.callback()).Run(GetMatchingFiles(params));
}

void FakeRecentSource::SetLag(const base::TimeDelta& lag) {
  lag_ = lag;
}

bool FakeRecentSource::MatchesFileType(const RecentFile& file,
                                       RecentSource::FileType file_type) const {
  if (file_type == FileType::kAll) {
    return true;
  }

  std::string mime_type;
  if (!net::GetMimeTypeFromFile(file.url().path(), &mime_type)) {
    return false;
  }

  switch (file_type) {
    case FileType::kAudio:
      return net::MatchesMimeType("audio/*", mime_type);
    case FileType::kImage:
      return net::MatchesMimeType("image/*", mime_type);
    case FileType::kVideo:
      return net::MatchesMimeType("video/*", mime_type);
    default:
      return false;
  }
}

std::vector<RecentFile> FakeRecentSource::GetMatchingFiles(
    const Params& params) {
  std::vector<RecentFile> result;
  for (const auto& file : canned_files_) {
    if (MatchesFileType(file, params.file_type())) {
      result.push_back(file);
    }
  }
  return result;
}

}  // namespace ash