admc/tests/admc_test_members_tab.cpp

/*
 * ADMC - AD Management Center
 *
 * Copyright (C) 2020-2022 BaseALT Ltd.
 * Copyright (C) 2020-2022 Dmitry Degtyarev
 *
 * 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 <http://www.gnu.org/licenses/>.
 */

#include "admc_test_members_tab.h"

#include "select_dialogs/select_object_dialog.h"
#include "tabs/membership_tab.h"
#include "tabs/ui_membership_tab.h"

#include <QPushButton>
#include <QStandardItemModel>
#include <QTreeView>
#include <QVBoxLayout>

void ADMCTestMembersTab::init() {
    ADMCTest::init();

    view = new QTreeView(parent_widget);
    auto primary_button = new QPushButton(parent_widget);
    add_button = new QPushButton(parent_widget);
    remove_button = new QPushButton(parent_widget);
    auto properties_button = new QPushButton(parent_widget);
    auto primary_group_label = new QLabel(parent_widget);

    edit = new MembershipTabEdit(view, primary_button, add_button, remove_button, properties_button, primary_group_label, MembershipTabType_Members, parent_widget);

    model = edit->findChild<QStandardItemModel *>();
    QVERIFY(model);

    // Create test user
    const QString user_name = TEST_USER;
    user_dn = test_object_dn(user_name, CLASS_USER);
    const bool create_user_success = ad.object_add(user_dn, CLASS_USER);
    QVERIFY(create_user_success);

    // Create test group
    const QString group_name = TEST_GROUP;
    group_dn = test_object_dn(group_name, CLASS_GROUP);
    const bool create_group_success = ad.object_add(group_dn, CLASS_GROUP);
    QVERIFY(create_group_success);

    // Load it into the tab
    const AdObject object = ad.search_object(group_dn);
    edit->load(ad, object);
}

// Loading a group without members should result in empty
// model
void ADMCTestMembersTab::load_empty() {
    QCOMPARE(model->rowCount(), 0);
}

// Loading a group with members should put members in the
// model
void ADMCTestMembersTab::load() {
    const bool add_success = ad.group_add_member(group_dn, user_dn);
    QVERIFY(add_success);

    const AdObject object = ad.search_object(group_dn);
    edit->load(ad, object);

    QCOMPARE(model->rowCount(), 1);

    auto item = model->item(0, 0);
    QCOMPARE(item->text(), dn_get_name(user_dn));
}

// Removing members should remove members from model and group
void ADMCTestMembersTab::remove() {
    load();

    const QModelIndex index = model->index(0, 0);
    view->setCurrentIndex(index);

    remove_button->click();

    edit->apply(ad, group_dn);

    const AdObject updated_object = ad.search_object(group_dn);
    const QList<QString> member_list = updated_object.get_strings(ATTRIBUTE_MEMBER);

    QCOMPARE(model->rowCount(), 0);
    QVERIFY(member_list.isEmpty());
}

void ADMCTestMembersTab::add() {
    add_button->click();

    select_object_dialog_select(user_dn);

    // Check ui state before applying
    QCOMPARE(model->rowCount(), 1);
    QCOMPARE(model->item(0, 0)->text(), dn_get_name(user_dn));

    // Apply and check object state
    edit->apply(ad, group_dn);
    const AdObject object = ad.search_object(group_dn);
    const QList<QString> member_list = object.get_strings(ATTRIBUTE_MEMBER);
    QCOMPARE(member_list, QList<QString>({user_dn}));

    // Check ui state after applying (just in case)
    QCOMPARE(model->rowCount(), 1);
    QCOMPARE(model->item(0, 0)->text(), dn_get_name(user_dn));
}

QTEST_MAIN(ADMCTestMembersTab)