synchronous_event_unittest.cpp 1.34 KB
// Copyright (c) 2014 Baidu, Inc.
//
// Author: Ge,Jun (gejun@baidu.com)
// Date: Sat Aug 30 17:13:19 CST 2014

#include <gtest/gtest.h>
#include "base/synchronous_event.h"

namespace {
class SynchronousEventTest : public ::testing::Test{
protected:
    SynchronousEventTest(){
    };
    virtual ~SynchronousEventTest(){};
    virtual void SetUp() {
        srand(time(0));
    };
    virtual void TearDown() {
    };
};

struct Foo {};

typedef base::SynchronousEvent<int, int*> FooEvent;

FooEvent foo_event;
std::vector<std::pair<int, int> > result;

class FooObserver : public FooEvent::Observer {
public:
    FooObserver() : another_ob(NULL) {}
    
    void on_event(int x, int* p) {
        ++*p;
        result.push_back(std::make_pair(x, *p));
        if (another_ob) {
            foo_event.subscribe(another_ob);
        }
    }
    FooObserver* another_ob;
};


TEST_F(SynchronousEventTest, sanity) {
    const size_t N = 10;
    FooObserver foo_observer;
    FooObserver foo_observer2;
    foo_observer.another_ob = &foo_observer2;
    foo_event.subscribe(&foo_observer);
    int v = 0;
    result.clear();
    for (size_t i = 0; i < N; ++i) {
        foo_event.notify(i, &v);
    }
    ASSERT_EQ(2*N, result.size());
    for (size_t i = 0; i < 2*N; ++i) {
        ASSERT_EQ((int)i/2, result[i].first);
        ASSERT_EQ((int)i+1, result[i].second);
    }
}

}