Line data    Source code

       1              : /*
       2              :  ___________________________________________
       3              : |    _     ___                        _     |
       4              : |   | |   |__ \                      | |    |
       5              : |   | |__    ) |__ _  __ _  ___ _ __ | |_   |
       6              : |   | '_ \  / // _` |/ _` |/ _ \ '_ \| __|  |  HTTP/2 AGENT FOR MOCK TESTING
       7              : |   | | | |/ /| (_| | (_| |  __/ | | | |_   |  Version 0.0.z
       8              : |   |_| |_|____\__,_|\__, |\___|_| |_|\__|  |  https://github.com/testillano/h2agent
       9              : |                     __/ |                 |
      10              : |                    |___/                  |
      11              : |___________________________________________|
      12              : 
      13              : Licensed under the MIT License <http://opensource.org/licenses/MIT>.
      14              : SPDX-License-Identifier: MIT
      15              : Copyright (c) 2021 Eduardo Ramos
      16              : 
      17              : Permission is hereby  granted, free of charge, to any  person obtaining a copy
      18              : of this software and associated  documentation files (the "Software"), to deal
      19              : in the Software  without restriction, including without  limitation the rights
      20              : to  use, copy,  modify, merge,  publish, distribute,  sublicense, and/or  sell
      21              : copies  of  the Software,  and  to  permit persons  to  whom  the Software  is
      22              : furnished to do so, subject to the following conditions:
      23              : 
      24              : The above copyright notice and this permission notice shall be included in all
      25              : copies or substantial portions of the Software.
      26              : 
      27              : THE SOFTWARE  IS PROVIDED "AS  IS", WITHOUT WARRANTY  OF ANY KIND,  EXPRESS OR
      28              : IMPLIED,  INCLUDING BUT  NOT  LIMITED TO  THE  WARRANTIES OF  MERCHANTABILITY,
      29              : FITNESS FOR  A PARTICULAR PURPOSE AND  NONINFRINGEMENT. IN NO EVENT  SHALL THE
      30              : AUTHORS  OR COPYRIGHT  HOLDERS  BE  LIABLE FOR  ANY  CLAIM,  DAMAGES OR  OTHER
      31              : LIABILITY, WHETHER IN AN ACTION OF  CONTRACT, TORT OR OTHERWISE, ARISING FROM,
      32              : OUT OF OR IN CONNECTION WITH THE SOFTWARE  OR THE USE OR OTHER DEALINGS IN THE
      33              : SOFTWARE.
      34              : */
      35              : 
      36              : #include <ert/tracing/Logger.hpp>
      37              : 
      38              : #include <MockEventsHistory.hpp>
      39              : 
      40              : 
      41              : namespace h2agent
      42              : {
      43              : namespace model
      44              : {
      45              : 
      46          256 : void MockEventsHistory::loadEvent(std::shared_ptr<MockEvent> event, bool historyEnabled) {
      47              : 
      48          256 :     write_guard_t guard(rw_mutex_);
      49              : 
      50          256 :     if (!historyEnabled && events_.size() != 0) {
      51            2 :         events_[0] = event; // overwrite with this latest reception
      52              :     }
      53              :     else {
      54          254 :         events_.push_back(event);
      55              :     }
      56          256 : }
      57              : 
      58            6 : bool MockEventsHistory::removeEvent(std::uint64_t eventNumber, bool reverse) {
      59              : 
      60            6 :     write_guard_t guard(rw_mutex_);
      61              : 
      62            6 :     if (events_.size() == 0 || eventNumber == 0) return false;
      63            6 :     if (eventNumber > events_.size()) return false;
      64              : 
      65            6 :     events_.erase(events_.begin() + (reverse ? (events_.size() - eventNumber):(eventNumber - 1)));
      66              : 
      67            6 :     return true;
      68            6 : }
      69              : 
      70           10 : std::shared_ptr<MockEvent> MockEventsHistory::getEvent(std::uint64_t eventNumber, bool reverse) const {
      71              : 
      72           10 :     read_guard_t guard(rw_mutex_);
      73              : 
      74           10 :     if (events_.size() == 0) return nullptr;
      75           10 :     if (eventNumber == 0) return nullptr;
      76              : 
      77           10 :     if (eventNumber > events_.size()) return nullptr;
      78              : 
      79            8 :     if (reverse) {
      80            4 :         return *(events_.begin() + (events_.size() - eventNumber));
      81              :         //return *(std::prev(events_.end())); // this would be the last
      82              :     }
      83              : 
      84            4 :     return *(events_.begin() + (eventNumber - 1));
      85           10 : }
      86              : 
      87           13 : nlohmann::json MockEventsHistory::getJson() const {
      88           13 :     nlohmann::json result;
      89              : 
      90           13 :     data_key_.keyToJson(result);
      91              : 
      92           13 :     read_guard_t guard(rw_mutex_);
      93           33 :     for (auto it = events_.begin(); it != events_.end(); it ++) {
      94           60 :         result["events"].push_back((*it)->getJson());
      95              :     }
      96              : 
      97           26 :     return result;
      98           13 : }
      99              : 
     100           13 : const std::string &MockEventsHistory::getLastRegisteredRequestState() const {
     101           13 :     read_guard_t guard(rw_mutex_);
     102              :     // By design, there are no keys without history (at least 1 exists and the key is removed if it is finally deleted).
     103              :     // Then, back() is a valid iterator (https://github.com/testillano/h2agent/issues/53).
     104           26 :     return events_.back()->getState();
     105           13 : }
     106              : 
     107              : }
     108              : }
     109              :