From fc1ce6d56d6f1a95cd10ebff8a87d566644abe27 Mon Sep 17 00:00:00 2001
From: Greg Back <gback@mitre.org>
Date: Thu, 25 May 2017 12:31:45 -0500
Subject: [PATCH] Add some tests

---
 stix2/sources/__init__.py       |  4 +---
 stix2/test/test_data_sources.py | 22 ++++++++++++++++++++++
 2 files changed, 23 insertions(+), 3 deletions(-)
 create mode 100644 stix2/test/test_data_sources.py

diff --git a/stix2/sources/__init__.py b/stix2/sources/__init__.py
index d13fbd9..4c6a36e 100644
--- a/stix2/sources/__init__.py
+++ b/stix2/sources/__init__.py
@@ -401,9 +401,7 @@ class DataSource(object):
             stix_obj (dictionary): the STIX object to be returned
 
         '''
-        stix_obj = None
-
-        return stix_obj
+        raise NotImplementedError
 
     @abc.abstractmethod
     def all_versions(self, id_, _composite_filters=None):
diff --git a/stix2/test/test_data_sources.py b/stix2/test/test_data_sources.py
new file mode 100644
index 0000000..ab95d20
--- /dev/null
+++ b/stix2/test/test_data_sources.py
@@ -0,0 +1,22 @@
+import pytest
+
+import stix2.sources
+
+
+def test_data_source():
+    ds = stix2.sources.DataSource()
+
+    assert ds.name == "DataSource"
+
+
+def test_set_data_source_name():
+    ds = stix2.sources.DataSource(name="My Data Source")
+
+    assert ds.name == "My Data Source"
+
+
+def test_data_source_get():
+    ds = stix2.sources.DataSource(name="My Data Source")
+
+    with pytest.raises(NotImplementedError):
+        ds.get("foo")