def test_constructor(self): """ DataStoragePlugin.__init__() Test plan: -make sure self.mongo is a mongo client -make sure self.db is the hpit_data_storage collection -make sure logger set correctly """ ds = DataStoragePlugin(1234, 5678, None) ds.logger.should.equal(None) isinstance(ds.mongo, MongoClient).should.equal(True) isinstance(ds.db, Collection).should.equal(True) ds.db.full_name.should.equal("hpit_unit_test_db.data_storage")
def setUp(self): self.test_subject = DataStoragePlugin(1234, 5678, None) self.test_subject.send_response = MagicMock()