def test_check_quota_Zero2():
m = Mock(spec=ArcticLibraryBinding)
m.quota = None
m.get_library_metadata.return_value = 0
ArcticLibraryBinding.check_quota(m)
m.get_library_metadata.assert_called_once_with('QUOTA')
assert m.quota == 0
def test_check_quota_Zero2():
self = create_autospec(ArcticLibraryBinding)
self.quota = None
self.get_library_metadata.return_value = 0
ArcticLibraryBinding.check_quota(self)
self.get_library_metadata.assert_called_once_with('QUOTA')
assert self.quota == 0
def test_check_quota_Zero2():
m = Mock(spec=ArcticLibraryBinding)
m.quota = None
m.get_library_metadata.return_value = 0
ArcticLibraryBinding.check_quota(m)
m.get_library_metadata.assert_called_once_with('QUOTA')
assert m.quota == 0
def test_check_quota_Zero2():
self = create_autospec(ArcticLibraryBinding)
self.quota = None
self.get_library_metadata.return_value = 0
ArcticLibraryBinding.check_quota(self)
self.get_library_metadata.assert_called_once_with('QUOTA')
assert self.quota == 0
def test_check_quota_exceeded():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.quota = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(
stats=Mock(return_value={"totals": {"size": 1024 * 1024 * 1024, "count": 100}})
)
with pytest.raises(QuotaExceededException) as e:
ArcticLibraryBinding.check_quota(self)
assert "Quota Exceeded: 1.000 / 1 GB used" in str(e)
def test_check_quota_info():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.quota = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(
stats=Mock(return_value={"totals": {"size": 1 * 1024 * 1024, "count": 100}})
)
with patch("arctic.arctic.logger.info") as info:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
info.assert_called_once_with("Mongo Quota: 0.001 / 1 GB used")
assert self.quota_countdown == 51153
def test_check_quota_exceeded():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 1024 * 1024 * 1024,
'count': 100,
}
}))
with pytest.raises(QuotaExceededException) as e:
ArcticLibraryBinding.check_quota(self)
assert "Quota Exceeded: 1.000 / 1 GB used" in str(e)
def test_check_quota_exceeded():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.quota = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(
return_value={'totals': {
'size': 1024 * 1024 * 1024,
'count': 100,
}}))
with pytest.raises(QuotaExceededException) as e:
ArcticLibraryBinding.check_quota(self)
assert "Quota Exceeded: 1.000 / 1 GB used" in str(e)
def test_check_quota_exceeded():
self = create_autospec(ArcticLibraryBinding,
database_name='arctic_db',
library='lib')
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(
return_value={'totals': {
'size': 1024 * 1024 * 1024,
'count': 100,
}}))
with pytest.raises(QuotaExceededException) as e:
ArcticLibraryBinding.check_quota(self)
assert "Quota Exceeded: arctic_db.lib 1.000 / 1 GB used" in str(e.value)
def test_check_quota_info():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 1 * 1024 * 1024,
'count': 100,
}
}))
with patch('arctic.arctic.logger.info') as info:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
info.assert_called_once_with('Mongo Quota: 0.001 / 1 GB used')
assert self.quota_countdown == 51153
def test_check_quota():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 900 * 1024 * 1024,
'count': 100,
}
}))
with patch('arctic.arctic.logger.warning') as warn:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
warn.assert_called_once_with('Mongo Quota: 0.879 / 1 GB used')
assert self.quota_countdown == 6
def test_check_quota_90_percent():
self = create_autospec(ArcticLibraryBinding, database_name='arctic_db',
library='lib')
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 0.91 * 1024 * 1024 * 1024,
'count': 1000000,
}
}))
with patch('arctic.arctic.logger.warning') as warn:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
warn.assert_called_once_with('Mongo Quota: arctic_db.lib 0.910 / 1 GB used')
def test_check_quota_info():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.quota = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(
return_value={'totals': {
'size': 1 * 1024 * 1024,
'count': 100,
}}))
with patch('arctic.arctic.logger.info') as info:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
info.assert_called_once_with('Mongo Quota: 0.001 / 1 GB used')
assert self.quota_countdown == 51153
def test_check_quota():
self = create_autospec(ArcticLibraryBinding)
self.arctic = create_autospec(Arctic)
self.quota = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 900 * 1024 * 1024,
'count': 100,
}
}))
with patch('arctic.arctic.logger.warn') as warn:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
warn.assert_called_once_with('Mongo Quota: 0.879 / 1 GB used')
assert self.quota_countdown == 6
def test_check_quota_90_percent():
self = create_autospec(ArcticLibraryBinding, database_name='arctic_db',
library='lib')
self.arctic = create_autospec(Arctic)
self.get_library_metadata.return_value = 1024 * 1024 * 1024
self.quota_countdown = 0
self.arctic.__getitem__.return_value = Mock(stats=Mock(return_value={'totals':
{'size': 0.91 * 1024 * 1024 * 1024,
'count': 1000000,
}
}))
with patch('arctic.arctic.logger.warning') as warn:
ArcticLibraryBinding.check_quota(self)
self.arctic.__getitem__.assert_called_once_with(self.get_name.return_value)
warn.assert_called_once_with('Mongo Quota: arctic_db.lib 0.910 / 1 GB used')
def test_database_library_specifier(library, expected_library, expected_database):
mongo = MagicMock()
with patch('arctic.arctic.ArcticLibraryBinding._auth'):
ml = ArcticLibraryBinding(mongo, library)
assert ml.library == expected_library
mongo._conn.__getitem__.assert_called_with(expected_database)
def test_ArcticLibraryBinding_db():
arctic = create_autospec(Arctic)
arctic._conn = create_autospec(MongoClient)
alb = ArcticLibraryBinding(arctic, "sentinel.library")
with patch.object(alb, '_auth') as _auth:
# connection is cached during __init__
alb._db
assert _auth.call_count == 0
# Change the arctic connection
arctic._conn = create_autospec(MongoClient)
alb._db
assert _auth.call_count == 1
# connection is still cached
alb._db
assert _auth.call_count == 1
def test_set_quota():
m = Mock(spec=ArcticLibraryBinding)
ArcticLibraryBinding.set_quota(m, 10000)
m.set_library_metadata.assert_called_once_with('QUOTA', 10000)
assert m.quota_countdown == 0
assert m.quota == 10000
def test_get_quota():
m = Mock(spec=ArcticLibraryBinding)
m.get_library_metadata.return_value = 42
assert ArcticLibraryBinding.get_quota(m) == 42
m.get_library_metadata.assert_called_once_with('QUOTA')
def test_check_quota_countdown():
self = create_autospec(ArcticLibraryBinding)
self.quota = 10
self.quota_countdown = 10
ArcticLibraryBinding.check_quota(self)
assert self.quota_countdown == 9
def test_check_quota_countdown():
self = create_autospec(ArcticLibraryBinding)
self.quota = 10
self.quota_countdown = 10
ArcticLibraryBinding.check_quota(self)
assert self.quota_countdown == 9
def test_check_quota_Zero():
self = create_autospec(ArcticLibraryBinding)
self.quota = 0
ArcticLibraryBinding.check_quota(self)
def test_set_quota():
self = create_autospec(ArcticLibraryBinding)
ArcticLibraryBinding.set_quota(self, 10000)
self.set_library_metadata.assert_called_once_with('QUOTA', 10000)
assert self.quota_countdown == 0
assert self.quota == 10000
def test_get_quota():
self = create_autospec(ArcticLibraryBinding)
self.get_library_metadata.return_value = 42
assert ArcticLibraryBinding.get_quota(self) == 42
self.get_library_metadata.assert_called_once_with('QUOTA')
def test_get_quota():
self = create_autospec(ArcticLibraryBinding)
self.get_library_metadata.return_value = 42
assert ArcticLibraryBinding.get_quota(self) == 42
self.get_library_metadata.assert_called_once_with('QUOTA')
def test_check_quota_countdown():
self = create_autospec(ArcticLibraryBinding)
self.get_library_metadata.return_value = 10
self.quota_countdown = 10
ArcticLibraryBinding.check_quota(self)
assert self.quota_countdown == 9
def test_check_quota_countdown():
self = create_autospec(ArcticLibraryBinding)
self.get_library_metadata.return_value = 10
self.quota_countdown = 10
ArcticLibraryBinding.check_quota(self)
assert self.quota_countdown == 9
def test_get_quota():
m = Mock(spec=ArcticLibraryBinding)
m.get_library_metadata.return_value = 42
assert ArcticLibraryBinding.get_quota(m) == 42
m.get_library_metadata.assert_called_once_with('QUOTA')
def test_set_quota():
m = Mock(spec=ArcticLibraryBinding)
ArcticLibraryBinding.set_quota(m, 10000)
m.set_library_metadata.assert_called_once_with('QUOTA', 10000)
assert m.quota_countdown == 0
assert m.quota == 10000
def test_lib_repr():
mongo = MagicMock()
with patch('arctic.arctic.ArcticLibraryBinding._auth'):
ml = ArcticLibraryBinding(mongo, 'asdf')
assert str(ml) == repr(ml)
def test_set_quota():
self = create_autospec(ArcticLibraryBinding)
ArcticLibraryBinding.set_quota(self, 10000)
self.set_library_metadata.assert_called_once_with('QUOTA', 10000)
assert self.quota_countdown == 0
assert self.quota == 10000
def test_check_quota_Zero():
self = create_autospec(ArcticLibraryBinding)
self.quota = 0
ArcticLibraryBinding.check_quota(self)
