#!/usr/bin/env python
#
# Copyright (C) 2016 GNS3 Technologies Inc.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see .
from unittest.mock import MagicMock
import pytest
import uuid
from tests.utils import AsyncioMagicMock
from gns3server.controller.shape import Shape
from gns3server.controller.project import Project
@pytest.fixture
def project(controller, async_run):
return async_run(controller.add_project())
@pytest.fixture
def shape(project):
return Shape(project, None, svg="")
def test_init_without_uuid(project):
shape = Shape(project, None, svg="")
assert shape.id is not None
def test_init_with_uuid(project):
id = str(uuid.uuid4())
shape = Shape(project, id, svg="")
assert shape.id == id
def test_json(project):
i = Shape(project, None, svg="")
assert i.__json__() == {
"shape_id": i.id,
"project_id": project.id,
"x": i.x,
"y": i.y,
"z": i.z
}
assert i.__json__(topology_dump=True) == {
"shape_id": i.id,
"x": i.x,
"y": i.y,
"z": i.z
}
def test_update(shape, project, async_run, controller):
controller._notification = AsyncioMagicMock()
project.dump = MagicMock()
async_run(shape.update(x=42))
assert shape.x == 42
controller._notification.emit.assert_called_with("shape.updated", shape.__json__())
assert project.dump.called