ilogue / niprov

    def test_Obtains_optional_snapshot_data_from_cache_when_serializing(self):

        self.pictureCache.getBytes.return_value = sentinel.snapbytes

        with patch('niprov.mongo.bson') as bson:

            bson.Binary.return_value = sentinel.snapbson

            self.setupRepo()

            img = Mock()

            img.provenance = {'a':1}

            self.repo.add(img)

            self.pictureCache.getBytes.assert_called_with(for_=img)

            bson.Binary.assert_called_with(sentinel.snapbytes)

            self.db.provenance.insert_one.assert_called_with({'a':1, 

                '_snapshot-data':sentinel.snapbson})

    def test_If_no_snapshot_doesnt_add_data_field(self):

        self.repo.add(img)

        self.db.provenance.insert_one.assert_called_with(

            {'a':1, 'duration':67.89})

    def test_Converts_duration_to_timedelta_when_deserializing(self):

        self.setupRepo()

        self.db.provenance.find_one.return_value = {'a':3, 'duration':89.01}

        out = self.repo.byLocation('/p/f1')

        self.fileFactory.fromProvenance.assert_called_with(

            {'a':3, 'duration':timedelta(seconds=89.01)})

    def test_Obtains_optional_snapshot_data_from_cache_when_serializing(self):

        self.pictureCache.getBytes.return_value = sentinel.snapbytes

        with patch('niprov.mongo.bson') as bson:

            bson.Binary.return_value = sentinel.snapbson

    def test_updateApproval(self):

        self.setupRepo()

        img = Mock()

        p = '/p/f1'

        newStatus = 'oh-oh'

        self.repo.updateApproval(p, newStatus)

        self.db.provenance.update.assert_called_with(

            {'location':p}, {'$set': {'approval': newStatus}})

    def test_latest(self):

        self.fileFactory.fromProvenance.side_effect = lambda p: 'img_'+p

        self.db.provenance.find.return_value = Mock()

        self.db.provenance.find.return_value.sort.return_value.limit.return_value = ['px','py']

        self.setupRepo()