Beispiel #1
0
def test_get_basic_information(test_data, select, a_id, a_ip, a_status):
    """Test get_basic_information function."""
    with patch('sqlite3.connect') as mock_db:
        mock_db.return_value = test_data.global_db
        agent_info = Agent(a_id).get_basic_information(select=select)
        if select is not None:
            assert agent_info.keys() == set(select['fields'])

        assert agent_info['id'] == a_id
        assert agent_info['ip'] == a_ip
        assert agent_info['status'] == a_status