class BackupTestCase(unittest.TestCase):
def __init__(self, method_name='runTest'):
super().__init__(method_name)
init_test()
self.cli = HttpClient(f'/api/v2')
@staticmethod
def _subscribe():
HttpClient(f'/api/v2').put('/subscription/vault')
@classmethod
def setUpClass(cls):
cls._subscribe()
def test01_get_state(self):
r = self.cli.get('/vault-deprecated/content')
self.assertEqual(r.status_code, 200)
@unittest.skip
def test02_backup(self):
self.backup(self.cli.get_backup_credential())
@unittest.skip
def test03_restore(self):
self.restore(self.cli.get_backup_credential())
def backup(self, credential):
r = self.cli.post('/vault-deprecated/content?to=hive_node', body={'credential': credential})
self.assertEqual(r.status_code, 201)
def restore(self, credential):
r = self.cli.post('/vault-deprecated/content?from=hive_node', body={'credential': credential})
self.assertEqual(r.status_code, 201)
class IpfsBackupTestCase(unittest.TestCase):
def __init__(self, method_name='runTest'):
super().__init__(method_name)
init_test()
self.cli = HttpClient(f'/api/v2')
self.backup_cli = HttpClient(f'/api/v2', is_backup_node=True)
@staticmethod
def _subscribe_vault():
HttpClient(f'/api/v2').put('/subscription/vault')
@staticmethod
def _unsubscribe_vault_on_backup_node():
HttpClient(f'/api/v2', is_backup_node=True).delete('/subscription/vault')
@classmethod
def setUpClass(cls):
cls._subscribe_vault()
def test01_subscribe(self):
response = self.backup_cli.put('/subscription/backup')
self.assertTrue(response.status_code in [200, 455])
def test02_get_info(self):
response = self.backup_cli.get('/subscription/backup')
self.assertEqual(response.status_code, 200)
def test03_backup(self):
r = self.cli.post('/vault/content?to=hive_node',
body={'credential': self.cli.get_backup_credential()})
self.assertEqual(r.status_code, 201)
time.sleep(10)
def test03_backup_invalid_parameter(self):
r = self.cli.post('/vault/content?to=hive_node')
self.assertEqual(r.status_code, 400)
@unittest.skip
def test03_backup_force(self):
r = self.cli.post('/vault/content?to=hive_node&is_force=true',
body={'credential': self.cli.get_backup_credential()})
self.assertEqual(r.status_code, 201)
def test04_state(self):
r = self.cli.get('/vault/content')
self.assertEqual(r.status_code, 200)
def test05_restore(self):
r = self.cli.post('/vault/content?from=hive_node',
body={'credential': self.cli.get_backup_credential()})
self.assertEqual(r.status_code, 201)
time.sleep(10)
def test05_restore_invalid_parameter(self):
r = self.cli.post('/vault/content?from=hive_node')
self.assertEqual(r.status_code, 400)
@unittest.skip
def test05_restore_force(self):
r = self.cli.post('/vault/content?from=hive_node&is_force=true',
body={'credential': self.cli.get_backup_credential()})
self.assertEqual(r.status_code, 201)
@unittest.skip
def test06_promotion(self):
# PREPARE: backup and remove the vault for local test.
self.__class__._unsubscribe_vault_on_backup_node()
# do promotion.
r = self.backup_cli.post('/backup/promotion')
self.assertEqual(r.status_code, 201)
# check the vault state.
response = self.backup_cli.get('/subscription/vault')
self.assertEqual(response.status_code, 200)
@unittest.skip
def test07_unsubscribe(self):
response = self.backup_cli.delete('/subscription/backup')
self.assertEqual(response.status_code, 204)
self.__class__._unsubscribe_vault_on_backup_node()