text: # -*- coding: utf-8 -*-
  """Tests for the attribute containers."""
  
  from __future__ import unicode_literals
  
  import unittest
  
  from dftimewolf.lib.containers import containers
  
  class ReportDataTest(unittest.TestCase):
    """Tests for the Report data attribute container."""
  
    def testGetAttributeNames(self):
      """Tests the GetAttributeNames function."""
      attribute_container = containers.Report(module_name='name', text='text')
  
      expected_attribute_names = [
          'attributes', 'module_name', 'text', 'text_format']
  
      attribute_names = sorted(attribute_container.GetAttributeNames())
  
      self.assertEqual(attribute_names, expected_attribute_names)
  
  
  class ThreatIntelligenceDataTest(unittest.TestCase):
    """Tests for the Report data attribute container."""
  
    def testGetAttributeNames(self):
      """Tests the GetAttributeNames function."""
      attribute_container = containers.ThreatIntelligence(
          name='name',
          indicator='.*')
  
      expected_attribute_names = ['indicator', 'name']
  
      attribute_names = sorted(attribute_container.GetAttributeNames())
  
      self.assertEqual(attribute_names, expected_attribute_names)
  
  
  if __name__ == '__main__':
    unittest.main()
  