| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CONTENT_BROWSER_QUOTA_MOCK_QUOTA_MANAGER_PROXY_H_ |
| #define CONTENT_BROWSER_QUOTA_MOCK_QUOTA_MANAGER_PROXY_H_ |
| |
| #include <stdint.h> |
| |
| #include "base/macros.h" |
| #include "content/browser/quota/mock_quota_manager.h" |
| #include "storage/browser/quota/quota_client.h" |
| #include "storage/browser/quota/quota_manager_proxy.h" |
| #include "storage/common/quota/quota_types.h" |
| #include "url/gurl.h" |
| |
| using storage::QuotaManagerProxy; |
| |
| namespace content { |
| |
| class MockQuotaManager; |
| |
| class MockQuotaManagerProxy : public QuotaManagerProxy { |
| public: |
| // It is ok to give NULL to |quota_manager|. |
| MockQuotaManagerProxy(MockQuotaManager* quota_manager, |
| base::SingleThreadTaskRunner* task_runner); |
| |
| void RegisterClient(QuotaClient* client) override; |
| |
| void SimulateQuotaManagerDestroyed(); |
| |
| // We don't mock them. |
| void NotifyOriginInUse(const GURL& origin) override {} |
| void NotifyOriginNoLongerInUse(const GURL& origin) override {} |
| void SetUsageCacheEnabled(QuotaClient::ID client_id, |
| const GURL& origin, |
| StorageType type, |
| bool enabled) override {} |
| void GetUsageAndQuota( |
| base::SequencedTaskRunner* original_task_runner, |
| const GURL& origin, |
| StorageType type, |
| const QuotaManager::UsageAndQuotaCallback& callback) override; |
| |
| // Validates the |client_id| and updates the internal access count |
| // which can be accessed via notify_storage_accessed_count(). |
| // The also records the |origin| and |type| in last_notified_origin_ and |
| // last_notified_type_. |
| void NotifyStorageAccessed(QuotaClient::ID client_id, |
| const GURL& origin, |
| StorageType type) override; |
| |
| // Records the |origin|, |type| and |delta| as last_notified_origin_, |
| // last_notified_type_ and last_notified_delta_ respecitvely. |
| // If non-null MockQuotaManager is given to the constructor this also |
| // updates the manager's internal usage information. |
| void NotifyStorageModified(QuotaClient::ID client_id, |
| const GURL& origin, |
| StorageType type, |
| int64_t delta) override; |
| |
| int notify_storage_accessed_count() const { return storage_accessed_count_; } |
| int notify_storage_modified_count() const { return storage_modified_count_; } |
| GURL last_notified_origin() const { return last_notified_origin_; } |
| StorageType last_notified_type() const { return last_notified_type_; } |
| int64_t last_notified_delta() const { return last_notified_delta_; } |
| |
| protected: |
| ~MockQuotaManagerProxy() override; |
| |
| private: |
| MockQuotaManager* mock_manager() const { |
| return static_cast<MockQuotaManager*>(quota_manager()); |
| } |
| |
| int storage_accessed_count_; |
| int storage_modified_count_; |
| GURL last_notified_origin_; |
| StorageType last_notified_type_; |
| int64_t last_notified_delta_; |
| |
| QuotaClient* registered_client_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MockQuotaManagerProxy); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_BROWSER_QUOTA_MOCK_QUOTA_MANAGER_PROXY_H_ |