OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/password_manager/core/browser/password_manager.h" | 5 #include "components/password_manager/core/browser/password_manager.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <utility> | 8 #include <utility> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
55 MockPasswordManagerClient() { | 55 MockPasswordManagerClient() { |
56 EXPECT_CALL(*this, GetStoreResultFilter()) | 56 EXPECT_CALL(*this, GetStoreResultFilter()) |
57 .Times(AnyNumber()) | 57 .Times(AnyNumber()) |
58 .WillRepeatedly(Return(&filter_)); | 58 .WillRepeatedly(Return(&filter_)); |
59 ON_CALL(filter_, ShouldSave(_)).WillByDefault(Return(true)); | 59 ON_CALL(filter_, ShouldSave(_)).WillByDefault(Return(true)); |
60 } | 60 } |
61 | 61 |
62 MOCK_CONST_METHOD0(IsSavingAndFillingEnabledForCurrentPage, bool()); | 62 MOCK_CONST_METHOD0(IsSavingAndFillingEnabledForCurrentPage, bool()); |
63 MOCK_CONST_METHOD0(DidLastPageLoadEncounterSSLErrors, bool()); | 63 MOCK_CONST_METHOD0(DidLastPageLoadEncounterSSLErrors, bool()); |
64 MOCK_CONST_METHOD0(GetPasswordStore, PasswordStore*()); | 64 MOCK_CONST_METHOD0(GetPasswordStore, PasswordStore*()); |
65 // The code inside EXPECT_CALL for PromptUserToSaveOrUpdatePasswordPtr owns | 65 // The code inside EXPECT_CALL for PromptUserToSaveOrUpdatePasswordPtr and |
66 // the PasswordFormManager* argument. | 66 // ShowManualFallbackForSavingPtr owns the PasswordFormManager* argument. |
67 MOCK_METHOD1(PromptUserToSaveOrUpdatePasswordPtr, void(PasswordFormManager*)); | 67 MOCK_METHOD1(PromptUserToSaveOrUpdatePasswordPtr, void(PasswordFormManager*)); |
| 68 MOCK_METHOD3(ShowManualFallbackForSavingPtr, |
| 69 void(PasswordFormManager*, bool, bool)); |
| 70 MOCK_METHOD0(HideManualFallbackForSaving, void()); |
68 MOCK_METHOD1(NotifySuccessfulLoginWithExistingPassword, | 71 MOCK_METHOD1(NotifySuccessfulLoginWithExistingPassword, |
69 void(const autofill::PasswordForm&)); | 72 void(const autofill::PasswordForm&)); |
70 MOCK_METHOD0(AutomaticPasswordSaveIndicator, void()); | 73 MOCK_METHOD0(AutomaticPasswordSaveIndicator, void()); |
71 MOCK_METHOD0(GetPrefs, PrefService*()); | 74 MOCK_METHOD0(GetPrefs, PrefService*()); |
72 MOCK_CONST_METHOD0(GetMainFrameURL, const GURL&()); | 75 MOCK_CONST_METHOD0(GetMainFrameURL, const GURL&()); |
73 MOCK_METHOD0(GetDriver, PasswordManagerDriver*()); | 76 MOCK_METHOD0(GetDriver, PasswordManagerDriver*()); |
74 MOCK_CONST_METHOD0(GetStoreResultFilter, const MockStoreResultFilter*()); | 77 MOCK_CONST_METHOD0(GetStoreResultFilter, const MockStoreResultFilter*()); |
75 | 78 |
76 // Workaround for std::unique_ptr<> lacking a copy constructor. | 79 // Workaround for std::unique_ptr<> lacking a copy constructor. |
77 bool PromptUserToSaveOrUpdatePassword( | 80 bool PromptUserToSaveOrUpdatePassword( |
78 std::unique_ptr<PasswordFormManager> manager, | 81 std::unique_ptr<PasswordFormManager> manager, |
79 bool update_password) override { | 82 bool update_password) override { |
80 PromptUserToSaveOrUpdatePasswordPtr(manager.release()); | 83 PromptUserToSaveOrUpdatePasswordPtr(manager.release()); |
81 return false; | 84 return false; |
82 } | 85 } |
| 86 void ShowManualFallbackForSaving(std::unique_ptr<PasswordFormManager> manager, |
| 87 bool has_generated_password, |
| 88 bool is_update) override { |
| 89 ShowManualFallbackForSavingPtr(manager.release(), has_generated_password, |
| 90 is_update); |
| 91 } |
83 void AutomaticPasswordSave( | 92 void AutomaticPasswordSave( |
84 std::unique_ptr<PasswordFormManager> manager) override { | 93 std::unique_ptr<PasswordFormManager> manager) override { |
85 AutomaticPasswordSaveIndicator(); | 94 AutomaticPasswordSaveIndicator(); |
86 } | 95 } |
87 | 96 |
88 void FilterAllResultsForSaving() { | 97 void FilterAllResultsForSaving() { |
89 EXPECT_CALL(filter_, ShouldSave(_)).WillRepeatedly(Return(false)); | 98 EXPECT_CALL(filter_, ShouldSave(_)).WillRepeatedly(Return(false)); |
90 } | 99 } |
91 | 100 |
92 private: | 101 private: |
(...skipping 1709 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1802 std::unique_ptr<PasswordFormManager> form_manager_to_save; | 1811 std::unique_ptr<PasswordFormManager> form_manager_to_save; |
1803 EXPECT_CALL(client_, PromptUserToSaveOrUpdatePasswordPtr(_)) | 1812 EXPECT_CALL(client_, PromptUserToSaveOrUpdatePasswordPtr(_)) |
1804 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); | 1813 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
1805 | 1814 |
1806 OnPasswordFormSubmitted(form); | 1815 OnPasswordFormSubmitted(form); |
1807 observed.clear(); | 1816 observed.clear(); |
1808 manager()->OnPasswordFormsRendered(&driver_, observed, true); | 1817 manager()->OnPasswordFormsRendered(&driver_, observed, true); |
1809 } | 1818 } |
1810 #endif | 1819 #endif |
1811 | 1820 |
| 1821 TEST_F(PasswordManagerTest, ManualFallbackForSaving) { |
| 1822 EXPECT_CALL(client_, IsSavingAndFillingEnabledForCurrentPage()) |
| 1823 .WillRepeatedly(Return(true)); |
| 1824 |
| 1825 std::vector<PasswordForm> observed; |
| 1826 PasswordForm form(MakeSimpleForm()); |
| 1827 observed.push_back(form); |
| 1828 PasswordForm stored_form = form; |
| 1829 stored_form.password_value = ASCIIToUTF16("old_password"); |
| 1830 EXPECT_CALL(*store_, GetLogins(_, _)) |
| 1831 .WillOnce(WithArg<1>(InvokeConsumer(stored_form))); |
| 1832 EXPECT_CALL(driver_, FillPasswordForm(_)).Times(2); |
| 1833 manager()->OnPasswordFormsParsed(&driver_, observed); |
| 1834 manager()->OnPasswordFormsRendered(&driver_, observed, true); |
| 1835 |
| 1836 // The username of the stored form is the same, there should be update bubble. |
| 1837 std::unique_ptr<PasswordFormManager> form_manager_to_save; |
| 1838 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, false, true)) |
| 1839 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
| 1840 manager()->ShowManualFallbackForSaving(&driver_, form); |
| 1841 ASSERT_TRUE(form_manager_to_save); |
| 1842 EXPECT_THAT(form_manager_to_save->pending_credentials(), FormMatches(form)); |
| 1843 |
| 1844 // The username of the stored form is different, there should be save bubble. |
| 1845 PasswordForm new_form = form; |
| 1846 new_form.username_value = ASCIIToUTF16("another_username"); |
| 1847 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, false, false)) |
| 1848 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
| 1849 manager()->ShowManualFallbackForSaving(&driver_, new_form); |
| 1850 ASSERT_TRUE(form_manager_to_save); |
| 1851 EXPECT_THAT(form_manager_to_save->pending_credentials(), |
| 1852 FormMatches(new_form)); |
| 1853 |
| 1854 // Hide the manual fallback. |
| 1855 EXPECT_CALL(client_, HideManualFallbackForSaving()); |
| 1856 manager()->HideManualFallbackForSaving(); |
| 1857 } |
| 1858 |
| 1859 // Tests that the manual fallback for saving isn't shown if there is no response |
| 1860 // from the password storage. When crbug.com/741537 is fixed, change this test. |
| 1861 TEST_F(PasswordManagerTest, ManualFallbackForSaving_SlowBackend) { |
| 1862 EXPECT_CALL(client_, IsSavingAndFillingEnabledForCurrentPage()) |
| 1863 .WillRepeatedly(Return(true)); |
| 1864 |
| 1865 std::vector<PasswordForm> observed; |
| 1866 PasswordForm form(MakeSimpleForm()); |
| 1867 observed.push_back(form); |
| 1868 PasswordStoreConsumer* store_consumer = nullptr; |
| 1869 EXPECT_CALL(*store_, GetLogins(_, _)).WillOnce(SaveArg<1>(&store_consumer)); |
| 1870 manager()->OnPasswordFormsParsed(&driver_, observed); |
| 1871 manager()->OnPasswordFormsRendered(&driver_, observed, true); |
| 1872 |
| 1873 // There is no response from the store. Don't show the fallback. |
| 1874 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, _, _)).Times(0); |
| 1875 manager()->ShowManualFallbackForSaving(&driver_, form); |
| 1876 |
| 1877 // The storage responded. The fallback can be shown. |
| 1878 ASSERT_TRUE(store_consumer); |
| 1879 store_consumer->OnGetPasswordStoreResults( |
| 1880 std::vector<std::unique_ptr<PasswordForm>>()); |
| 1881 std::unique_ptr<PasswordFormManager> form_manager_to_save; |
| 1882 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, false, false)) |
| 1883 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
| 1884 manager()->ShowManualFallbackForSaving(&driver_, form); |
| 1885 } |
| 1886 |
| 1887 TEST_F(PasswordManagerTest, ManualFallbackForSaving_GeneratedPassword) { |
| 1888 EXPECT_CALL(client_, IsSavingAndFillingEnabledForCurrentPage()) |
| 1889 .WillRepeatedly(Return(true)); |
| 1890 |
| 1891 std::vector<PasswordForm> observed; |
| 1892 PasswordForm form(MakeSimpleForm()); |
| 1893 observed.push_back(form); |
| 1894 EXPECT_CALL(*store_, GetLogins(_, _)) |
| 1895 .WillOnce(WithArg<1>(InvokeEmptyConsumerWithForms())); |
| 1896 manager()->OnPasswordFormsParsed(&driver_, observed); |
| 1897 manager()->OnPasswordFormsRendered(&driver_, observed, true); |
| 1898 |
| 1899 // A user accepts a password generated by Chrome. It triggers password |
| 1900 // presaving and showing manual fallback. |
| 1901 std::unique_ptr<PasswordFormManager> form_manager_to_save; |
| 1902 EXPECT_CALL(*store_, AddLogin(_)); |
| 1903 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, true, false)) |
| 1904 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
| 1905 manager()->OnPresaveGeneratedPassword(form); |
| 1906 manager()->ShowManualFallbackForSaving(&driver_, form); |
| 1907 ASSERT_TRUE(form_manager_to_save); |
| 1908 EXPECT_THAT(form_manager_to_save->pending_credentials(), FormMatches(form)); |
| 1909 |
| 1910 // A user edits the generated password. And again it causes password presaving |
| 1911 // and showing manual fallback. |
| 1912 EXPECT_CALL(*store_, UpdateLoginWithPrimaryKey(_, _)); |
| 1913 EXPECT_CALL(client_, ShowManualFallbackForSavingPtr(_, true, false)) |
| 1914 .WillOnce(WithArg<0>(SaveToScopedPtr(&form_manager_to_save))); |
| 1915 manager()->OnPresaveGeneratedPassword(form); |
| 1916 manager()->ShowManualFallbackForSaving(&driver_, form); |
| 1917 |
| 1918 // A user removes the generated password. The presaved password is removed, |
| 1919 // the fallback is disabled. |
| 1920 EXPECT_CALL(*store_, RemoveLogin(_)); |
| 1921 EXPECT_CALL(client_, HideManualFallbackForSaving()); |
| 1922 manager()->OnPasswordNoLongerGenerated(form); |
| 1923 manager()->HideManualFallbackForSaving(); |
| 1924 } |
| 1925 |
1812 } // namespace password_manager | 1926 } // namespace password_manager |
OLD | NEW |