Home | History | Annotate | Download | only in util 
      1 # Copyright 2013 The Chromium Authors. All rights reserved.
      2 # Use of this source code is governed by a BSD-style license that can be
      3 # found in the LICENSE file.
      4 
      5 import tempfile
      6 import unittest
      7 
      8 import md5_check
      9 
     10 
     11 class TestMd5Check(unittest.TestCase):
     12   def testCallAndRecordIfStale(self):
     13     input_strings = ['string1', 'string2']
     14     input_file1 = tempfile.NamedTemporaryFile()
     15     input_file2 = tempfile.NamedTemporaryFile()
     16     file1_contents = 'input file 1'
     17     file2_contents = 'input file 2'
     18     input_file1.write(file1_contents)
     19     input_file1.flush()
     20     input_file2.write(file2_contents)
     21     input_file2.flush()
     22     input_files = [input_file1.name, input_file2.name]
     23 
     24     record_path = tempfile.NamedTemporaryFile(suffix='.stamp')
     25 
     26     def CheckCallAndRecord(should_call, message, force=False):
     27       self.called = False
     28       def MarkCalled():
     29         self.called = True
     30       md5_check.CallAndRecordIfStale(
     31           MarkCalled,
     32           record_path=record_path.name,
     33           input_paths=input_files,
     34           input_strings=input_strings,
     35           force=force)
     36       self.failUnlessEqual(should_call, self.called, message)
     37 
     38     CheckCallAndRecord(True, 'should call when record doesn\'t exist')
     39     CheckCallAndRecord(False, 'should not call when nothing changed')
     40     CheckCallAndRecord(True, force=True, message='should call when forced')
     41 
     42     input_file1.write('some more input')
     43     input_file1.flush()
     44     CheckCallAndRecord(True, 'changed input file should trigger call')
     45 
     46     input_files = input_files[::-1]
     47     CheckCallAndRecord(False, 'reordering of inputs shouldn\'t trigger call')
     48 
     49     input_files = input_files[:1]
     50     CheckCallAndRecord(True, 'removing file should trigger call')
     51 
     52     input_files.append(input_file2.name)
     53     CheckCallAndRecord(True, 'added input file should trigger call')
     54 
     55     input_strings[0] = input_strings[0] + ' a bit longer'
     56     CheckCallAndRecord(True, 'changed input string should trigger call')
     57 
     58     input_strings = input_strings[::-1]
     59     CheckCallAndRecord(True, 'reordering of string inputs should trigger call')
     60 
     61     input_strings = input_strings[:1]
     62     CheckCallAndRecord(True, 'removing a string should trigger call')
     63 
     64     input_strings.append('a brand new string')
     65     CheckCallAndRecord(True, 'added input string should trigger call')
     66 
     67 
     68 if __name__ == '__main__':
     69   unittest.main()
     70