cache_warming/fetch_popular_test.py

#!/usr/bin/python
"""Tests for fetch_popular.py."""

import os
import fetch_popular
from wvtest import wvtest


@wvtest.wvtest
def testSortHitsLog_empty():
  try:
    file_name = 'test_log.json'
    with open(file_name, 'w') as f:
      f.write('{}')

    expected = []
    actual = fetch_popular.sort_hits_log(file_name)
    wvtest.WVPASSEQ(actual, expected)
  finally:
    os.remove(file_name)


@wvtest.wvtest
def testSortHitsLog_nonEmpty():
  try:
    file_name = 'test_log.json'
    with open(file_name, 'w') as f:
      f.write('{"www.google.com": [2, "123456789"], "www.yahoo.com":'
      	       ' [1,"987654321"], "www.espn.com": [3, "135792468"]}')

    expected = ['www.espn.com', 'www.google.com', 'www.yahoo.com']
    actual = fetch_popular.sort_hits_log(file_name)
    wvtest.WVPASSEQ(actual, expected)
  finally:
    os.remove(file_name)


if __name__ == '__main__':
  wvtest.wvtest_main()