Make the MockDRT implementation work again and add support for --actuals.

Tools/Scripts/webkitpy/layout_tests/port/mock_drt_unittest.py

 # Copyright (C) 2011 Google Inc. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met:
 #
 #     * Redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer.
 #     * Redistributions in binary form must reproduce the above
 # copyright notice, this list of conditions and the following disclaimer
@@ skipping 58 matching lines @@
         pass
 
     def test_get_crash_log(self):
         pass
 
     def test_check_build(self):
         pass
 
 
 class MockDRTTest(unittest.TestCase):
-    def input_line(self, port, test_name, checksum=None):
+    def input_line(self, port, test_name, pixel_tests, checksum=None):
         url = port.create_driver(0).test_to_uri(test_name)
         if url.startswith('file://'):
             url = url[len('file://'):]
-
+        if pixel_tests:
+            url += "'--pixel-test"
         if checksum:
-            return url + "'" + checksum + '\n'
+            url += "'" + checksum
         return url + '\n'
 
-    def extra_args(self, pixel_tests):
-        if pixel_tests:
-            return ['--pixel-tests', '-']
-        return ['-']
-
     def make_drt(self, options, args, host, stdin, stdout, stderr):
         return mock_drt.MockDRT(options, args, host, stdin, stdout, stderr)
 
     def make_input_output(self, port, test_name, pixel_tests,
                           expected_checksum, drt_output, drt_input=None, expected_text=None):
         if pixel_tests:
             if not expected_checksum:
                 expected_checksum = port.expected_checksum(test_name)
         if not drt_input:
-            drt_input = self.input_line(port, test_name, expected_checksum)
+            drt_input = self.input_line(port, test_name, pixel_tests, expected_checksum)
         text_output = expected_text or port.expected_text(test_name) or ''
 
         if not drt_output:
             drt_output = self.expected_output(port, test_name, pixel_tests,
                                               text_output, expected_checksum)
         return (drt_input, drt_output)
 
     def expected_output(self, port, test_name, pixel_tests, text_output, expected_checksum):
         output = ['Content-Type: text/plain\n']
         if text_output:
             output.append(text_output)
         output.append('#EOF\n')
         if pixel_tests and expected_checksum:
             output.extend(['\n',
                            'ActualHash: %s\n' % expected_checksum,
                            'ExpectedHash: %s\n' % expected_checksum])
         output.append('#EOF\n')
         return output
 
     def assertTest(self, test_name, pixel_tests, expected_checksum=None, drt_output=None, host=None, expected_text=None):
         port_name = 'test'
         host = host or MockSystemHost()
         test.add_unit_tests_to_mock_filesystem(host.filesystem)
         port = PortFactory(host).get(port_name)
         drt_input, drt_output = self.make_input_output(port, test_name,
             pixel_tests, expected_checksum, drt_output, drt_input=None, expected_text=expected_text)
 
-        args = ['--platform', port_name] + self.extra_args(pixel_tests)
+        args = ['--dump-render-tree', '--platform', port_name, '-']
         stdin = newstringio.StringIO(drt_input)
         stdout = newstringio.StringIO()
         stderr = newstringio.StringIO()
         options, args = mock_drt.parse_options(args)
 
         drt = self.make_drt(options, args, host, stdin, stdout, stderr)
         res = drt.run()
 
         self.assertEqual(res, 0)
 
         # We use the StringIO.buflist here instead of getvalue() because
         # the StringIO might be a mix of unicode/ascii and 8-bit strings.
         self.assertEqual(stdout.buflist, drt_output)
-        self.assertEqual(stderr.getvalue(), '' if options.test_shell else '#EOF\n')
+        self.assertEqual(stderr.getvalue(), '#EOF\n')
 
     def test_main(self):
         host = MockSystemHost()
         test.add_unit_tests_to_mock_filesystem(host.filesystem)
         stdin = newstringio.StringIO()
         stdout = newstringio.StringIO()
         stderr = newstringio.StringIO()
-        res = mock_drt.main(['--platform', 'test'] + self.extra_args(False),
+        res = mock_drt.main(['--dump-render-tree', '--platform', 'test', '-'],
                             host, stdin, stdout, stderr)
         self.assertEqual(res, 0)
         self.assertEqual(stdout.getvalue(), '')
         self.assertEqual(stderr.getvalue(), '')
         self.assertEqual(host.filesystem.written_files, {})
 
     def test_pixeltest_passes(self):
         # This also tests that we handle HTTP: test URLs properly.
         self.assertTest('http/tests/passes/text.html', True)
 
@@ skipping 14 matching lines @@
     def test_checksum_in_png(self):
         self.assertTest('passes/checksum_in_image.html', True)
 
     def test_missing_image(self):
         self.assertTest('failures/expected/missing_image.html', True)
 
     def test_missing_text(self):
         self.assertTest('failures/expected/missing_text.html', True)
 
     def test_reftest_match(self):
-        self.assertTest('passes/reftest.html', False, expected_checksum='mock-checksum', expected_text='reference text\n')
         self.assertTest('passes/reftest.html', True, expected_checksum='mock-checksum', expected_text='reference text\n')
 
     def test_reftest_mismatch(self):
-        self.assertTest('passes/mismatch.html', False, expected_checksum='mock-checksum', expected_text='reference text\n')
         self.assertTest('passes/mismatch.html', True, expected_checksum='mock-checksum', expected_text='reference text\n')
 
+    def test_audio(self):
+        self.assertTest('passes/audio.html', pixel_tests=True,
+                        drt_output=['Content-Type: audio/wav\n',
+                         'Content-Transfer-Encoding: base64\n',
+                         'YXVkaW8td2F2',
+                         '\n',
+                         '#EOF\n',
+                         '#EOF\n'])
 
-class MockTestShellTest(MockDRTTest):
-    def extra_args(self, pixel_tests):
-        if pixel_tests:
-            return ['--pixel-tests=/tmp/png_result0.png']
-        return []
-
-    def make_drt(self, options, args, host, stdin, stdout, stderr):
-        options.test_shell = True
-
-        # We have to set these by hand because --platform test won't trigger
-        # the Chromium code paths.
-        options.pixel_path = '/tmp/png_result0.png'
-        options.pixel_tests = True
-
-        return mock_drt.MockTestShell(options, args, host, stdin, stdout, stderr)
-
-    def input_line(self, port, test_name, checksum=None):
-        url = port.create_driver(0).test_to_uri(test_name)
-        if checksum:
-            return url + ' 6000 ' + checksum + '\n'
-        return url + ' 6000\n'
-
-    def expected_output(self, port, test_name, pixel_tests, text_output, expected_checksum):
-        url = port.create_driver(0).test_to_uri(test_name)
-        output = ['#URL:%s\n' % url]
-        if expected_checksum:
-            output.append('#MD5:%s\n' % expected_checksum)
-        if text_output:
-            output.append(text_output)
-            if not text_output.endswith('\n'):
-                output.append('\n')
-        output.append('#EOF\n')
-        return output
-
-    def test_pixeltest__fails(self):
-        host = MockSystemHost()
-        url = '#URL:file://'
-        url = url + '%s/failures/expected/image_checksum.html' % PortFactory(host).get('test').layout_tests_dir()
-        self.assertTest('failures/expected/image_checksum.html', pixel_tests=True,
-            expected_checksum='image_checksum',
-            drt_output=[url + '\n',
-                        '#MD5:image_checksum-checksum\n',
-                        'image_checksum-txt',
-                        '\n',
-                        '#EOF\n'],
-            host=host)
-        self.assertEqual(host.filesystem.written_files,
-            {'/tmp/png_result0.png': 'image_checksum\x8a-pngtEXtchecksum\x00image_checksum-checksum'})
-
-    def test_test_shell_parse_options(self):
-        options, args = mock_drt.parse_options(['--platform', 'mac', '--test-shell',
-            '--pixel-tests=/tmp/png_result0.png'])
-        self.assertTrue(options.test_shell)
-        self.assertTrue(options.pixel_tests)
-        self.assertEqual(options.pixel_path, '/tmp/png_result0.png')
+    def test_virtual(self):
+        self.assertTest('virtual/passes/text.html', True)