def test_with_empty_args(self, mock_request): mock_request.return_value = FakeGen() cmd = Command() with capture_sys_output() as (stdout, stderr): cmd.init_args([]) cmd.run() expected_content = [ "", "Id Type Description Status", "================================================================================", str(b"1 ping Description 1 Ongoing"), str(b"2 ping Description 2 Ongoing"), str(b"3 ping Description 3 Ongoing"), str(b"4 ping Description 4 Ongoing"), str(b"5 ping Description 5 Ongoing"), "================================================================================", " Showing 5 of 5 total measurements", "", ] self.maxDiff = None self.assertEqual( set(stdout.getvalue().split("\n")), set(expected_content) ) self.assertEqual( cmd.arguments.field, ("id", "type", "description", "status"))
def test_with_empty_args(self, mock_request): mock_request.return_value = FakeGen() cmd = Command() with capture_sys_output() as (stdout, stderr): cmd.init_args([]) cmd.run() expected_content = ( "\n" "Id Type Description Status\n" "================================================================================\n" "1 ping Description 1 Ongoing\n" "2 ping Description 2 Ongoing\n" "3 ping Description 3 Ongoing\n" "4 ping Description 4 Ongoing\n" "5 ping Description 5 Ongoing\n" "================================================================================\n" " Showing 5 of 5 total measurements\n" "\n" ) self.assertEqual( set(stdout.getvalue().split("\n")), set(expected_content.split("\n")) ) self.assertEqual( cmd.arguments.field, ("id", "type", "description", "status"))
def test_get_line_items(self, mock_request): mock_request.return_value = FakeGen() cmd = Command() cmd.init_args([]) cmd.run() self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), [1, "ping", "Description 1", "Ongoing"] ) cmd = Command() cmd.init_args([ "--field", "id", "--field", "status" ]) self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), [1, "Ongoing"] ) cmd = Command() cmd.init_args([ "--field", "url", ]) self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), ["https://atlas.ripe.net/measurements/1/"] )
def test_get_line_items(self, mock_request): mock_request.return_value = FakeGen() cmd = Command() cmd.init_args([]) cmd.run() self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), [1, "ping", "Description 1", "Ongoing"] ) cmd = Command() cmd.init_args([ "--field", "id", "--field", "status" ]) self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), [1, "Ongoing"] ) cmd = Command() cmd.init_args([ "--field", "url", ]) self.assertEqual( cmd._get_line_items(FakeGen.Measurement( id=1, type="ping", status="Ongoing", status_id=2, meta_data={"status": {"name": "Ongoing", "id": 2}}, target="Name 1", description="Description 1", )), ["https://atlas.ripe.net/measurements/1/"] )