InstantExtended: Bail on TemplateURLs with no espv.

chrome/browser/ui/search/search_unittest.cc

 // Copyright (c) 2013 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.
 
 #include "chrome/browser/ui/search/search.h"
+
+#include "chrome/browser/search_engines/template_url.h"
+#include "chrome/browser/search_engines/template_url_service.h"
+#include "chrome/browser/search_engines/template_url_service_factory.h"
+#include "chrome/browser/search_engines/template_url_service_test_util.h"
+#include "chrome/test/base/testing_profile.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chrome {
 namespace search {
 
 TEST(EmbeddedSearchFieldTrialTest, GetFieldTrialInfo) {
   FieldTrialFlags flags;
   uint64 group_number = 0;
   const uint64 ZERO = 0;
 
@@ skipping 39 matching lines @@
       "moose", "default", flags));
 
   group_number = 0;
   flags.clear();
   GetFieldTrialInfo(
       "Group77 bar:1 baz:7 cat:dogs DISABLED", &flags, &group_number);
   EXPECT_EQ(ZERO, group_number);
   EXPECT_EQ(ZERO, flags.size());
 }
 
+TEST(EmbeddedSearchPageVersionTest, BailsWithBadTemplateURL) {
+  EnableInstantExtendedAPIForTesting();
+
+  TemplateURLServiceTestUtil test_util;
+  test_util.SetUp();
+  test_util.VerifyLoad();
+  TemplateURLData data;
+  data.SetURL("http://google.com/?q={searchTerms}");
+  data.instant_url = "http://google.com/instant#q={searchTerms}";
+  TemplateURL* url = new TemplateURL(test_util.profile(), data);
+  test_util.model()->Add(url);
+  test_util.model()->SetDefaultSearchProvider(url);
+  ASSERT_EQ(url, test_util.model()->GetDefaultSearchProvider());
+  ASSERT_TRUE(TemplateURLServiceFactory::GetForProfile(test_util.profile())->
+      FindNewDefaultSearchProvider() != NULL);
+
+  EXPECT_EQ(0U, chrome::search::EmbeddedSearchPageVersion(
+      test_util.profile()->GetOriginalProfile()));
+  test_util.TearDown();
+}
+
+TEST(EmbeddedSearchPageVersionTest, DoesNotBailWithGoodTemplateURL) {
+  EnableInstantExtendedAPIForTesting();
+
+  TemplateURLServiceTestUtil test_util;
+  test_util.SetUp();
+  test_util.VerifyLoad();
+  TemplateURLData data;
+  data.SetURL("http://google.com/?q={searchTerms}");
+  data.instant_url = "http://google.com/instant#q={searchTerms}"
+      "{google:instantExtendedEnabledParameter}";
+  TemplateURL* url = new TemplateURL(test_util.profile(), data);
+  test_util.model()->Add(url);
+  test_util.model()->SetDefaultSearchProvider(url);
+  ASSERT_EQ(url, test_util.model()->GetDefaultSearchProvider());
+  ASSERT_TRUE(TemplateURLServiceFactory::GetForProfile(test_util.profile())->
+      FindNewDefaultSearchProvider() != NULL);
+
+  EXPECT_NE(0U, chrome::search::EmbeddedSearchPageVersion(
+      test_util.profile()->GetOriginalProfile()));
+  test_util.TearDown();
+}
+
 }  // namespace search
 }  // namespace chrome