Merge pull request #307 from STIXProject/datetime-format-test

Add test for datetime format

stix/test/indicator_test.py

@@ -2,9 +2,11 @@
 # See LICENSE.txt for complete terms.
 
 import unittest
+from datetime import datetime
+from mixbox.vendor.six import text_type
 
 from stix.core import STIXPackage
-from stix.indicator import Indicator, RelatedCampaignRefs
+from stix.indicator import Indicator, RelatedCampaignRefs, ValidTime
 
 from stix.test import EntityTestCase, assert_warnings
 from stix.test.common import related_test
@@ -446,6 +448,15 @@ def test_deprecated_related_packages(self):
         i.related_packages.append(STIXPackage())
         self.assertEqual(len(i.related_packages), 1)
 
+    def test_datetime_format(self):
+        indicator = Indicator(title="title")
+        valid_time = ValidTime(start_time=datetime.strptime("2010-03-05",
+                                                            "%Y-%m-%d"))
+        indicator.add_valid_time_position(valid_time)
+
+        ixml = indicator.to_xml()
+        self.assertTrue("2010-03-05T" in text_type(ixml))
+
 
 class RelatedCampaignReferencesTests(unittest.TestCase, EntityTestCase):
     klass = RelatedCampaignRefs