def remove_db():
db.session.remove()
db.drop_all()
engine = db.get_engine()
if engine.url.drivername == 'sqlite':
try:
os.remove(engine.url.database)
except:
pass
def test_from_to_json_with_gate(self, mock_get_now, mock_uuid):
mock_get_now.return_value = get_now()
mock_uuid.return_value = '22cd859d-ee91-4079-a112-000000000002'
s = Server('server2', id='22cd859d-ee91-4079-a112-000000000001')
gate = mock.MagicMock()
gate.to_json.return_value = {'server_id': 1, 'server': 'n1'}
type(s).gates = mock.PropertyMock(return_value=[gate])
self.assertDictEqual(
{
'id':
'22cd859d-ee91-4079-a112-000000000001',
'name':
'server2',
'granules': [],
'gates': [{}],
'created_on':
mock_get_now.return_value.strftime(defaults.DATETIME_FORMAT),
'_old_name':
None,
'deleted':
False
}, s.to_json(add_gates=True))
db.session.add(s)
db.session.commit()
s_json = s.to_json(add_gates=True)
smashed = Server.from_json(s_json)
self.assertIs(s, smashed)
self.assertEqual(s.id, smashed.id)
self.assertEqual(s.name, smashed.name)
self.assertEqual(s.granules, smashed.granules)
self.assertEqual(s.last_modified_at, smashed.last_modified_at)
self.assertListEqual(s.gates, smashed.gates)
# from new Server
db.session.remove()
db.drop_all()
db.create_all()
with patch('dimensigon.domain.entities.server.Gate.from_json'
) as mock_gate_from_json:
smashed = Server.from_json(s_json)
self.assertEqual(s.id, smashed.id)
self.assertEqual(s.name, smashed.name)
self.assertEqual(s.granules, smashed.granules)
self.assertEqual(s.last_modified_at, smashed.last_modified_at)
self.assertEqual(1, len(smashed.gates))
mock_gate_from_json.assert_called_once()
def tearDown(self) -> None:
db.session.remove()
db.drop_all()
self.app_context.pop()
def tearDown(self) -> None:
db.session.remove()
db.drop_all()
