py/hwid/encoder_unittest.py

#!/usr/bin/python -u
# -*- coding: utf-8 -*-
#
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import factory_common # pylint: disable=W0611
import os
import unittest

from cros.factory.hwid import Database
from cros.factory.hwid.encoder import BOMToBinaryString
from cros.factory.hwid.encoder import BinaryStringToEncodedString, Encode

_TEST_DATA_PATH = os.path.join(os.path.dirname(__file__), 'testdata')


class EncoderTest(unittest.TestCase):
  def setUp(self):
    self.database = Database.LoadFile(os.path.join(_TEST_DATA_PATH,
                                                   'test_db.yaml'))

  def testBOMToBinaryString(self):
    result = open(os.path.join(_TEST_DATA_PATH,
                               'test_probe_result.yaml'), 'r').read()
    bom = self.database.ProbeResultToBOM(result)
    # Manually set unprobeable components.
    bom = self.database.UpdateComponentsOfBOM(
        bom, {'camera': 'camera_0', 'display_panel': 'display_panel_0'})
    self.assertEquals(
        '00000111010000010100', BOMToBinaryString(self.database, bom))
    bom.image_id = 5
    self.assertEquals(
        '00101111010000010100', BOMToBinaryString(self.database, bom))
    bom.encoding_pattern_index = 1
    self.assertEquals(
        '10101111010000010100', BOMToBinaryString(self.database, bom))

  def testBinaryStringToEncodedString(self):
    self.assertEquals('CHROMEBOOK A5AU-LU',
                      BinaryStringToEncodedString(
                          self.database, '00000111010000010100'))

  def testEncode(self):
    result = open(os.path.join(_TEST_DATA_PATH,
                               'test_probe_result.yaml'), 'r').read()
    bom = self.database.ProbeResultToBOM(result)
    # Manually set unprobeable components.
    bom = self.database.UpdateComponentsOfBOM(
        bom, {'camera': 'camera_0', 'display_panel': 'display_panel_0'})
    hwid = Encode(self.database, bom)
    self.assertEquals('00000111010000010100', hwid.binary_string)
    self.assertEquals('CHROMEBOOK A5AU-LU', hwid.encoded_string)


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