[Feature] Access to component state data

src/__tests__/e2e.test.js

@@ -8,6 +8,7 @@ jest.setMock('../dispatchTrackingEvent', dispatchTrackingEvent);
 const testDataContext = { testDataContext: true };
 const testData = { testData: true };
 const dispatch = jest.fn();
+const testState = { booleanState: true };
 
 describe('e2e', () => {
   // eslint-disable-next-line global-require
@@ -362,6 +363,35 @@ describe('e2e', () => {
     expect(dispatch).toHaveBeenCalledTimes(2); // pageview event and simulated button click
   });
 
+  it('dispatches state data when components contain state', () => {
+    @track(testDataContext, { dispatch })
+    class TestOptions extends React.Component {
+      constructor() {
+        super();
+        this.state = {
+          booleanState: true,
+        };
+      }
+
+      @track((props, state) => ({ booleanState: state.booleanState }))
+      exampleMethod = () => {}
+
+      render() {
+        this.exampleMethod();
+        return <div />;
+      }
+    }
+
+    mount(<TestOptions />);
+
+
+    expect(dispatchTrackingEvent).not.toHaveBeenCalled();
+    expect(dispatch).toHaveBeenCalledWith({
+      ...testDataContext,
+      ...testState,
+    });
+  });
+
   it('logs a console error when there is already a process defined on context', () => {
     global.console.error = jest.fn();
     const process = () => {};

src/__tests__/trackEventMethodDecorator.test.js

@@ -63,4 +63,44 @@ describe('trackEventMethodDecorator', () => {
     expect(trackEvent).toHaveBeenCalledWith(dummyData);
     expect(spyTestEvent).toHaveBeenCalledWith('x');
   });
+
+  it('properly passes through the correct arguments when trackingData is a function', () => {
+    const dummyData = {};
+    const trackingData = jest.fn(() => dummyData);
+    const trackEvent = jest.fn();
+    const spyTestEvent = jest.fn();
+    const dummyArgument = 'x';
+
+    class TestClass {
+      constructor() {
+        this.props = {
+          tracking: {
+            trackEvent,
+          },
+        };
+        this.state = {
+          myState: "someState",
+        };
+      }
+
+      @trackEventMethodDecorator(trackingData)
+      handleTestEvent = spyTestEvent
+    }
+
+    const myTC = new TestClass();
+    myTC.handleTestEvent(dummyArgument);
+
+    // Access the trackingData arguments
+    const trackingDataArguments = trackingData.mock.calls[0];
+
+    expect(trackingData).toHaveBeenCalledTimes(1);
+    expect(trackingDataArguments[0]).toEqual(myTC.props);
+    expect(trackingDataArguments[1]).toEqual(myTC.state);
+    // Here we have access to the raw `arguments` object, which is not an actual Array,
+    // so in order to compare, we convert the arguments to an array.
+    expect(Array.from(trackingDataArguments[2])).toEqual([dummyArgument]);
+
+    expect(trackEvent).toHaveBeenCalledWith(dummyData);
+    expect(spyTestEvent).toHaveBeenCalledWith(dummyArgument);
+  });
 });

src/trackEventMethodDecorator.js

@@ -5,7 +5,7 @@ export default function trackEventMethodDecorator(trackingData = {}) {
   return makeClassMemberDecorator(decoratedFn => function decorateClassMember() {
     if (this.props && this.props.tracking && typeof this.props.tracking.trackEvent === 'function') {
       const thisTrackingData = typeof trackingData === 'function'
-                ? trackingData(this.props, arguments)
+                ? trackingData(this.props, this.state, arguments)
                 : trackingData;
       this.props.tracking.trackEvent(thisTrackingData);
     }