/*
 * Copyright 2018, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Augustin Cavalier <waddlesplash>
 */
#include "../common.h"


#include <Application.h>
#include <String.h>
#include <Menu.h>
#include <MenuItem.h>
#include <PopUpMenu.h>


class MenuTestcase : public TestCase {
public:
        void
        ClassSizeTest()
        {
#ifdef B_HAIKU_32_BIT
                CPPUNIT_ASSERT_EQUAL(312, sizeof(BMenu));
                CPPUNIT_ASSERT_EQUAL(128, sizeof(BMenuItem));
#else
                CPPUNIT_ASSERT(true);
#endif
        }

        void
        ConcurrencyAbuseTest()
        {
                BApplication app("application/x-vnd.Haiku-interfacekit-menutest");
                BPopUpMenu* menu = new BPopUpMenu("Test");
                menu->AddItem(new BMenuItem("One", NULL));
                menu->AddItem(new BMenuItem("Two", NULL));
                menu->AddSeparatorItem();

                BMenuItem* items[10];
                for (int i = 0; i < 10; i++) {
                        BString str;
                        str.SetToFormat("%d", i);
                        items[i] = new BMenuItem(str.String(), NULL);
                }

                // Now for the actual abuse.
                menu->Go(BPoint(), false, true, true);
                snooze(50 * 1000 /* 50 ms */);
                for (int i = 0; i < 100; i++) {
                        for (int j = 0; j < (i % 5); j++) {
                                BMenuItem* item = items[(i + j) % 10];
                                if (item->Menu() != NULL)
                                        continue;
                                menu->AddItem(item);
                        }
                        if ((i % 3) == 0) {
                                for (int j = 0; j < (i % 5); j++)
                                        menu->RemoveItem((int32)0);
                        }
                }

                CPPUNIT_ASSERT_EQUAL(6, menu->CountItems());

                // Cleanup.
                for (int i = 0; i < 10; i++)
                        delete items[i];

                // Close the menu.
                char bytes[] = {B_ESCAPE};
                menu->KeyDown(bytes, 1);
                delete menu;
        }
};


Test*
MenuTestSuite()
{
        TestSuite* testSuite = new TestSuite();

        testSuite->addTest(new CppUnit::TestCaller<MenuTestcase>(
                "BMenu_ClassSize", &MenuTestcase::ClassSizeTest));
        testSuite->addTest(new CppUnit::TestCaller<MenuTestcase>(
                "BMenu_ConcurrencyAbuse", &MenuTestcase::ConcurrencyAbuseTest));

        return testSuite;
}