text: # -*- coding: utf-8 -*-
  #    Copyright 2014 Mirantis, Inc.
  #
  #    Licensed under the Apache License, Version 2.0 (the "License"); you may
  #    not use this file except in compliance with the License. You may obtain
  #    a copy of the License at
  #
  #         http://www.apache.org/licenses/LICENSE-2.0
  #
  #    Unless required by applicable law or agreed to in writing, software
  #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  #    License for the specific language governing permissions and limitations
  #    under the License.
  
  import base64
  import mock
  import string
  import struct
  
  from nailgun.test.base import BaseTestCase
  from nailgun.utils import AttributesGenerator
  
  
  def fake_urandom(length):
      return 'f' * length
  
  
  class TestAttributesGenerators(BaseTestCase):
  
      def is_hex(self, val):
          return all(c in string.hexdigits for c in val)
  
      def test_hexstring_generator(self):
          hex_str = AttributesGenerator.hexstring()
          self.assertEqual(len(hex_str), 8)
          self.assertTrue(all(hex_str))
  
          hex_str = AttributesGenerator.hexstring(32)
          self.assertEqual(len(hex_str), 32)
          self.assertTrue(all(hex_str))
  
      @mock.patch('nailgun.utils.os.urandom', side_effect=fake_urandom)
      @mock.patch('nailgun.utils.time.time',
                  side_effect=lambda: 1437172408.238165)
      def test_cephx_key_generator(self, m_time, m_urandom):
          coded_key = AttributesGenerator.cephx_key()
          uncoded_key = base64.b64decode(coded_key)
          package = uncoded_key[:-16]
          key = uncoded_key[-16:]
  
          key_type, sec, usec, length = struct.unpack('<hiih', package)
  
          self.assertEqual(key, 'f' * 16)
          self.assertEqual(key_type, 1)
          self.assertEqual(sec, 1437172408)
          self.assertEqual(usec, 0)
          self.assertEqual(length, 16)
  
      def test_password_generate(self):
          password_str = AttributesGenerator.password()
          self.assertEqual(len(password_str), 24)
          self.assertTrue(all(password_str))
  
          password_str = AttributesGenerator.password(8)
          self.assertEqual(len(password_str), 8)
          self.assertTrue(all(password_str))
  