Add cache_temperature state "HOT"

telemetry/telemetry/page/cache_temperature_unittest.py

Index: telemetry/telemetry/page/cache_temperature_unittest.py
diff --git a/telemetry/telemetry/page/cache_temperature_unittest.py b/telemetry/telemetry/page/cache_temperature_unittest.py
index ae73bbd779b250f8bad3c08615e0cb4df9436d26..37bd2acb8069e0d2e842586c33cdef6b922a5af5 100644
--- a/telemetry/telemetry/page/cache_temperature_unittest.py
+++ b/telemetry/telemetry/page/cache_temperature_unittest.py
@@ -98,3 +98,43 @@ class CacheTempeartureTests(browser_test_case.BrowserTestCase):
     markers = self.traceMarkers()
     self.assertIn('telemetry.internal.warm_cache.start', markers)
     self.assertIn('telemetry.internal.warm_cache.end', markers)
+
+  @decorators.Enabled('has tabs')
+  def testEnsureHotAfterColdAndWarmRun(self):
+    with self.captureTrace():
+      story_set = story.StorySet()
+      page = page_module.Page(
+          'http://google.com', page_set=story_set,
+          cache_temperature=cache_temperature.COLD, name='http://google.com')
+      cache_temperature.EnsurePageCacheTemperature(page, self._browser)
+
+      previous_page = page
+      page = page_module.Page(
+          'http://google.com', page_set=story_set,
+          cache_temperature=cache_temperature.WARM, name='http://google.com')
+      cache_temperature.EnsurePageCacheTemperature(
+          page, self._browser, previous_page)
+
+      previous_page = page
+      page = page_module.Page(
+          'http://google.com', page_set=story_set,
+          cache_temperature=cache_temperature.HOT, name='http://google.com')
+      cache_temperature.EnsurePageCacheTemperature(
+          page, self._browser, previous_page)
+
+    markers = self.traceMarkers()
+    self.assertNotIn('telemetry.internal.warm_cache.start', markers)
+
+  @decorators.Enabled('has tabs')
+  @decorators.Disabled('chromeos')
+  def testEnsureHotFromScratch(self):
+    with self.captureTrace():
+      story_set = story.StorySet()
+      page = page_module.Page(
+          'http://google.com', page_set=story_set,
+          cache_temperature=cache_temperature.HOT, name='http://google.com')
+      cache_temperature.EnsurePageCacheTemperature(page, self._browser)
+
+    markers = self.traceMarkers()
+    self.assertIn('telemetry.internal.warm_cache.start', markers)
+    self.assertIn('telemetry.internal.warm_cache.end', markers)