#!/usr/bin/env python
# Copyright 2016 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.
"""Tests for dexdiffer."""

import dexdiffer
import json
import unittest


class DexdifferTest(unittest.TestCase):

  def testReadDict(self):
    mapping_file = [
        'package.ClassName -> package.qq:\n',
        'android.support.v8.MenuPopupHelper -> android.support.v8.v:',
        '    android.view.LayoutInflater mInflater -> d\n',
        '    117:118:void setForceShowIcon(boolean) -> b',
        '    1:1:package.ClassName <init>(int,int) -> <init>',
    ]
    expected = {
        'package.qq': ['package.ClassName', {}],
        'android.support.v8.v':
        ['android.support.v8.MenuPopupHelper',
         {'android.view.LayoutInflater d':
              'android.view.LayoutInflater mInflater',
          'void b(boolean)': 'void setForceShowIcon(boolean)',
          'package.ClassName <init>(int,int)':
              'package.ClassName <init>(int,int)'}],
    }

    actual = dexdiffer._ReadMappingDict(mapping_file)
    self.assertDeepEqual(actual, expected)

  def testGetLineTokens1(self):
    line = '        public  void <init> (android.content.Context,float); // Cxr'
    expected = ['void', '<init>', 'android.content.Context', 'float']
    actual = dexdiffer._GetLineTokens(line)
    self.assertDeepEqual(actual, expected)

  def testGetLineTokens2(self):
    line = '/** asdfasdf /**/ private final static   /*asdf*/int varname$1;'
    expected = ['int', 'varname$1']
    actual = dexdiffer._GetLineTokens(line)
    self.assertDeepEqual(actual, expected)

  def testGetLineTokens3(self):
    line = 'int[] varname_1;'
    expected = ['int[]', 'varname_1']
    actual = dexdiffer._GetLineTokens(line)
    self.assertDeepEqual(actual, expected)

  def testGetLineTokensEmpty(self):
    line = '/***/   /*asdf*/ //comment;'
    expected = []
    actual = dexdiffer._GetLineTokens(line)
    self.assertDeepEqual(actual, expected)

  def testGetMemberIdentifier(self):
    line_tokens = ['void', 'b', 'boolean', 'rnmd[]']
    expected = 'void foo(boolean,renamed.type[])'
    renamed_class_name = 'renamed.class'
    mapping_dict = { renamed_class_name : [ 'actual.class.name', {
        'void b': 'void variable_name',
        'boolean a': 'boolean variable_name2',
        'void b(boolean,renamed.type)':
            'void wrong_function(boolean,renamed.type)',
        'void b(boolean,renamed.type[])': expected,
    }], 'rnmd': ['renamed.type', {}]}
    actual = dexdiffer._GetMemberIdentifier(line_tokens, mapping_dict,
                                             renamed_class_name, True)
    self.assertEqual(expected, actual)

  def testIsLineFunctionDefinition(self):
    line = 'java.lang.String  CONSOLE_ELISION= "[(0)]"'
    self.assertFalse(dexdiffer._IsLineFunctionDefinition(line))

  def testStripQuotes(self):
    string = 'abc\'123\'"456"def\'"7"\''
    self.assertEqual('abcdef', dexdiffer._StripQuotes(string))

  def testBuildMappedDexDict(self):
    dextra_output = [
'/* 3396 */ public class   org.chromium.chrome.browser.widget.a',
'           extends android.view.View    {',
'         /** 2 Instance Fields  **/',
'          private  org.chromium.chrome.browser.widget.b$1      x;',
'          private  int  mPosition;',
'         /** 1 Direct Methods  **/',
'         public  void <init> (android.content.Context, android.util.Attribute'
              'Set); // Constructor',
'         /** 2 Virtual Methods  **/',
'         public  void init (int, int);',
'         protected  void onDraw (android.graphics.Canvas);',
'        }  // end class org.chromium.chrome.browser.widget.a',
'/* 3397 */ class   org.chromium.chrome.browser.widget.b$1',
'           implements android.text.TextWatcher  {',
'         /** 1 Instance Fields  **/',
'          final  org.chromium.chrome.browser.widget.FloatLabelLayout'
              '    this$0;',
'         /** 1 Direct Methods  **/',
'          bool <init> (); // Constructor',
'         /** 1 Virtual Methods  **/',
'         public  void x (org.chromium.chrome.browser.widget.a, int, int,'
              ' int);',
'        }  // end class org.chromium.chrome.browser.widget.b$1']
    mapping_dict = {
        'org.chromium.chrome.browser.widget.a': ['class_name_a', {
            'class_name_b$1 x': 'class_name_b$1 full_member_variable',
            'int mPosition': 'int mPosition',
            'void <init>(android.content.Context,android.util.AttributeSet)':
              'void <init>(android.content.Context,android.util.AttributeSet)',
            'void init(int,int)': 'void init(int,int)',
            'void onDraw(android.graphics.Canvas)':
                'void onDraw(android.graphics.Canvas)',
        }],
        'org.chromium.chrome.browser.widget.b$1': ['class_name_b$1', {
            'org.chromium.chrome.browser.widget.FloatLabelLayout this$0':
              'org.chromium.chrome.browser.widget.FloatLabelLayout this$0',
            'boolean <init>()': 'boolean <init>()',
            'void x(class_name_a,int,int,int)':
              'void full_function_name(class_name_a,int,int,int)'
          }],
    }
    actual = dexdiffer._BuildMappedDexDict(dextra_output, mapping_dict)
    expected = { 'class_name_a': [
        'class_name_b$1 full_member_variable',
        'int mPosition',
        'void <init>(android.content.Context,android.util.AttributeSet)',
        'void init(int,int)',
        'void onDraw(android.graphics.Canvas)'
    ], 'class_name_b$1': [
        'org.chromium.chrome.browser.widget.FloatLabelLayout this$0',
        'boolean <init>()',
        'void full_function_name(class_name_a,int,int,int)'
    ]}
    self.assertDeepEqual(actual, expected)

  def testParseMappingLine(self):
    orig_name = "abc.q12$1"
    new_name = "a$1"
    line = orig_name + " -> " + new_name
    actual_orig_name, actual_new_name = dexdiffer._ParseMappingLine(line)
    self.assertEqual(orig_name, actual_orig_name)
    self.assertEqual(new_name, actual_new_name)

  def testDiffDexDicts(self):
    base_dict = { 'class_name_a': [
        'class_name_b$1 full_member_variable',
        'int mPosition',
        'void <init>(android.content.Context,android.util.AttributeSet)',
        'void init(int,int)',
        'void onDraw(android.graphics.Canvas)'
    ], 'class_name_b$1': [
        'org.chromium.chrome.browser.widget.FloatLabelLayout this$0',
        'void <init>()',
        'void full_function_name(class_name_a,int,int)'
    ], 'class_name_deleted': []}
    new_dict = { 'class_name_a': [
        'class_name_b$1 full_member_variable',
        'int mPosition',
        'void <init>(android.content.Context,android.util.AttributeSet)',
        'void init(int,int)',
        'void onDraw(android.graphics.Canvas)'
    ], 'class_name_b$1': [
        'org.chromium.chrome.browser.widget.FloatLabelLayout this$0',
        'void <init>()',
        'void full_function_name(int,int,int)'
    ], 'class_name_new': []}
    actual = dexdiffer._DiffDexDicts(base_dict, new_dict)
    expected = [('class_name_b$1\n'
                 '-  void full_function_name(class_name_a,int,int)\n'
                 '+  void full_function_name(int,int,int)'),
                '-class class_name_deleted',
                '+class class_name_new']

    self.assertDeepEqual(actual, expected)

  def assertDeepEqual(self, actual, expected):
    # Only designed to work for json-able types
    a = json.dumps(actual, sort_keys=True)
    e = json.dumps(expected, sort_keys=True)
    self.assertEqual(a, e)

if __name__ == '__main__':
  unittest.main()